mindstudio-probe 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml

@@ -1,925 +1,922 @@
-# Copyright 2024 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-# List of ops that register hooks
-
-  
-ops:
-  - adaptive_avg_pool1d
-  - adaptive_avg_pool2d
-  - adaptive_avg_pool3d
-  - adaptive_max_pool1d
-  - adaptive_max_pool2d
-  - avg_pool1d
-  - avg_pool2d
-  - avg_pool3d
-  - batch_norm
-  - bias_add
-  - ctc_greedy_decoder
-  - conv1d
-  - conv2d
-  - conv3d
-  - deformable_conv2d
-  - dense
-  - dropout
-  - dropout1d
-  - dropout2d
-  - dropout3d
-  - flatten
-  - fold
-  - fractional_max_pool3d
-  - lp_pool1d
-  - lp_pool2d
-  - lrn
-  - max_pool2d
-  - max_pool3d
-  - max_unpool1d
-  - max_unpool2d
-  - max_unpool3d
-  - unfold
-  - binary_cross_entropy
-  - binary_cross_entropy_with_logits
-  - cosine_embedding_loss
-  - cross_entropy
-  - ctc_loss
-  - gaussian_nll_loss
-  - hinge_embedding_loss
-  - huber_loss
-  - kl_div
-  - l1_loss
-  - margin_ranking_loss
-  - mse_loss
-  - multi_margin_loss
-  - multilabel_margin_loss
-  - multilabel_soft_margin_loss
-  - nll_loss
-  - smooth_l1_loss
-  - triplet_margin_loss
-  - elu
-  - fast_gelu
-  - gelu
-  - glu
-  - gumbel_softmax
-  - hardshrink
-  - hardsigmoid
-  - hardswish
-  - hardtanh
-  - leaky_relu
-  - log_softmax
-  - logsigmoid
-  - mish
-  - prelu
-  - relu
-  - relu6
-  - celu
-  - rrelu
-  - selu
-  - sigmoid
-  - silu
-  - softmax
-  - softmin
-  - softshrink
-  - softsign
-  - tanh
-  - threshold
-  - cdist
-  - dist
-  - pdist
-  - choice_with_mask
-  - random_categorical
-  - log_uniform_candidate_sampler
-  - uniform_candidate_sampler
-  - affine_grid
-  - bounding_box_decode
-  - bounding_box_encode
-  - col2im
-  - check_valid
-  - crop_and_resize
-  - grid_sample
-  - interpolate
-  - iou
-  - pad
-  - padding
-  - pixel_shuffle
-  - pixel_unshuffle
-  - upsample
-  - abs
-  - absolute
-  - accumulate_n
-  - acos
-  - arccos
-  - acosh
-  - add
-  - addcdiv
-  - addcmul
-  - addmv
-  - addn
-  - angle
-  - arccosh
-  - arcsin
-  - arcsinh
-  - arctan
-  - arctanh
-  - arctan2
-  - asin
-  - asinh
-  - atan
-  - atan2
-  - atanh
-  - atleast_1d
-  - atleast_2d
-  - atleast_3d
-  - bessel_i0
-  - bessel_i0e
-  - bessel_i1
-  - bessel_i1e
-  - bessel_j0
-  - bessel_j1
-  - bessel_k0
-  - bessel_k0e
-  - bessel_k1
-  - bessel_k1e
-  - bessel_y0
-  - bessel_y1
-  - bitwise_and
-  - bitwise_left_shift
-  - bitwise_or
-  - bitwise_right_shift
-  - bitwise_xor
-  - ceil
-  - clamp
-  - clip
-  - combinations
-  - copysign
-  - cos
-  - cosh
-  - cosine_similarity
-  - cov
-  - diag_embed
-  - diff
-  - deg2rad
-  - digamma
-  - div
-  - divide
-  - erf
-  - erfc
-  - erfinv
-  - exp
-  - exp2
-  - expm1
-  - floor
-  - floor_div
-  - floor_mod
-  - float_power
-  - fmod
-  - frac
-  - gcd
-  - hypot
-  - igamma
-  - igammac
-  - imag
-  - i0
-  - inv
-  - invert
-  - lcm
-  - ldexp
-  - lerp
-  - log
-  - log2
-  - log10
-  - log1p
-  - logaddexp
-  - logaddexp2
-  - logical_and
-  - logical_not
-  - logical_or
-  - logical_xor
-  - logit
-  - mul
-  - multiply
-  - mvlgamma
-  - neg
-  - negative
-  - nextafter
-  - polar
-  - polygamma
-  - positive
-  - pow
-  - rad2deg
-  - ravel
-  - real
-  - reciprocal
-  - remainder
-  - rot90
-  - round
-  - rsqrt
-  - sgn
-  - sign
-  - signbit
-  - sin
-  - sinc
-  - sinh
-  - sqrt
-  - square
-  - sub
-  - subtract
-  - t
-  - tan
-  - tanhshrink
-  - trapz
-  - tril_indices
-  - triu_indices
-  - true_divide
-  - trunc
-  - truncate_div
-  - truncate_mod
-  - xdivy
-  - xlogy
-  - zeta
-  - all
-  - amax
-  - amin
-  - aminmax
-  - any
-  - argmax
-  - argmin
-  - cummax
-  - cummin
-  - cumprod
-  - cumsum
-  - fmax
-  - histc
-  - logsumexp
-  - max
-  - mean
-  - median
-  - min
-  - norm
-  - prod
-  - std
-  - std_mean
-  - var
-  - var_mean
-  - argsort
-  - approximate_equal
-  - equal
-  - ge
-  - greater
-  - greater_equal
-  - gt
-  - intopk
-  - isclose
-  - isfinite
-  - isinf
-  - isnan
-  - isneginf
-  - isposinf
-  - isreal
-  - is_complex
-  - le
-  - less
-  - less_equal
-  - lt
-  - maximum
-  - minimum
-  - msort
-  - ne
-  - not_equal
-  - searchsorted
-  - topk
-  - bmm
-  - addbmm
-  - addmm
-  - baddbmm
-  - addr
-  - adjoint
-  - cholesky
-  - cholesky_solve
-  - batch_dot
-  - dot
-  - eig
-  - inner
-  - inverse
-  - geqrf
-  - ger
-  - kron
-  - lu_solve
-  - lu_unpack
-  - matmul
-  - matrix_solve
-  - matrix_band_part
-  - matrix_diag
-  - matrix_diag_part
-  - matrix_set_diag
-  - mm
-  - mv
-  - outer
-  - orgqr
-  - ormqr
-  - pinv
-  - svd
-  - tensor_dot
-  - logdet
-  - slogdet
-  - qr
-  - trace
-  - bartlett_window
-  - blackman_window
-  - hamming_window
-  - hann_window
-  - kaiser_window
-  - eye
-  - fill
-  - full
-  - full_like
-  - linspace
-  - logspace
-  - one_hot
-  - arange
-  - range
-  - heaviside
-  - bernoulli
-  - gamma
-  - laplace
-  - multinomial
-  - multinomial_with_replacement
-  - rand
-  - rand_like
-  - randint
-  - randint_like
-  - randn
-  - randn_like
-  - random_gamma
-  - random_poisson
-  - randperm
-  - standard_laplace
-  - standard_normal
-  - uniform
-  - argwhere
-  - batch_to_space_nd
-  - bincount
-  - block_diag
-  - broadcast_to
-  - cat
-  - channel_shuffle
-  - chunk
-  - column_stack
-  - concat
-  - conj
-  - count_nonzero
-  - deepcopy
-  - diag
-  - diagflat
-  - diagonal
-  - dyn_shape
-  - dsplit
-  - dstack
-  - einsum
-  - expand
-  - expand_dims
-  - flip
-  - fliplr
-  - flipud
-  - gather_d
-  - gather_elements
-  - gather_nd
-  - hsplit
-  - hstack
-  - index_add
-  - index_fill
-  - index_select
-  - inplace_add
-  - inplace_index_add
-  - inplace_sub
-  - inplace_update
-  - masked_fill
-  - masked_select
-  - meshgrid
-  - moveaxis
-  - movedim
-  - narrow
-  - nan_to_num
-  - nansum
-  - normal
-  - nonzero
-  - population_count
-  - rank
-  - repeat_elements
-  - repeat_interleave
-  - reshape
-  - reverse
-  - reverse_sequence
-  - roll
-  - scatter
-  - scatter_nd
-  - select
-  - sequence_mask
-  - shuffle
-  - size
-  - slice
-  - sort
-  - space_to_batch_nd
-  - sparse_segment_mean
-  - split
-  - squeeze
-  - stack
-  - strided_slice
-  - sum
-  - swapaxes
-  - swapdims
-  - tensor_scatter_add
-  - tensor_scatter_div
-  - tensor_scatter_max
-  - tensor_scatter_min
-  - tensor_scatter_mul
-  - tensor_scatter_sub
-  - tensor_scatter_elements
-  - tensor_split
-  - tile
-  - tril
-  - triu
-  - transpose
-  - unbind
-  - unique
-  - unique_consecutive
-  - unique_with_pad
-  - unsorted_segment_max
-  - unsorted_segment_min
-  - unsorted_segment_prod
-  - unsorted_segment_sum
-  - unsqueeze
-  - unstack
-  - view_as_real
-  - vsplit
-  - vstack
-  - where
-  - cross
-  - renorm
-  - is_tensor
-  - scalar_cast
-  - scalar_to_tensor
-  - tuple_to_array
-  - clip_by_global_norm
-  - clip_by_value
-  - assign
-  - assign_add
-  - assign_sub
-  - scatter_add
-  - scatter_div
-  - scatter_max
-  - scatter_min
-  - scatter_mul
-  - scatter_nd_add
-  - scatter_nd_div
-  - scatter_nd_max
-  - scatter_nd_min
-  - scatter_nd_mul
-  - scatter_nd_sub
-  - scatter_update
-  - derivative
-  - jet
-
-tensor:
-  - __abs__
-  - __add__
-  - __and__
-  - __bool__
-  - __eq__
-  - __ge__
-  - __gt__
-  - __iadd__
-  - __ifloordiv__
-  - __imatmul__
-  - __imod__
-  - __imul__
-  - __isub__
-  - __le__
-  - __lt__
-  - __matmul__
-  - __mod__
-  - __mul__
-  - __ne__
-  - __neg__
-  - __or__
-  - __pow__
-  - __radd__
-  - __rmatmul__
-  - __rmod__
-  - __rmul__
-  - __rpow__
-  - __rsub__
-  - __sub__
-  - __truediv__
-  - __xor__
-  - abs
-  - absolute
-  - acos
-  - acosh
-  - add
-  - addbmm
-  - addcdiv
-  - addcmul
-  - addmm
-  - addmv
-  - addr
-  - all
-  - amax
-  - amin
-  - any
-  - arccos
-  - arccosh
-  - argmax
-  - angle
-  - arcsin
-  - arcsinh
-  - arctan
-  - arctanh
-  - argmin
-  - argsort
-  - asin
-  - asinh
-  - atan
-  - atan2
-  - atanh
-  - baddbmm
-  - bernoulli
-  - bincount
-  - bitwise_and
-  - bitwise_or
-  - bitwise_xor
-  - bmm
-  - bool
-  - broadcast_to
-  - ceil
-  - cholesky_solve
-  - cholesky
-  - clamp
-  - clip
-  - conj
-  - copysign
-  - cos
-  - cosh
-  - cross
-  - cummax
-  - cummin
-  - cumprod
-  - cumsum
-  - deg2rad
-  - diag
-  - diagflat
-  - diff
-  - digamma
-  - div
-  - divide
-  - equal
-  - erf
-  - erfc
-  - erfinv
-  - exp
-  - expand_as
-  - expm1
-  - flip
-  - fliplr
-  - flipud
-  - float_power
-  - floor
-  - fmod
-  - frac
-  - gather_elements
-  - ge
-  - geqrf
-  - ger
-  - greater
-  - greater_equal
-  - gt
-  - half
-  - hardshrink
-  - heaviside
-  - histc
-  - hypot
-  - i0
-  - igamma
-  - igammac
-  - imag
-  - index_add
-  - index_fill
-  - index_put
-  - index_select
-  - inner
-  - int
-  - inverse
-  - isclose
-  - isfinite
-  - isinf
-  - isnan
-  - is_complex
-  - is_signed
-  - isneginf
-  - isposinf
-  - isreal
-  - lcm
-  - ldexp
-  - le
-  - lerp
-  - less
-  - less_equal
-  - log
-  - log10
-  - log1p
-  - log2
-  - logaddexp
-  - logaddexp2
-  - logdet
-  - logical_and
-  - logical_not
-  - logical_or
-  - logical_xor
-  - logit
-  - logsumexp
-  - long
-  - lt
-  - masked_fill
-  - masked_scatter
-  - masked_select
-  - matmul
-  - max
-  - maximum
-  - mean
-  - median
-  - min
-  - minimum
-  - moveaxis
-  - movedim
-  - msort
-  - multinomial
-  - multiply
-  - mvlgamma
-  - nan_to_num
-  - nansum
-  - narrow
-  - ne
-  - neg
-  - negative
-  - nelement
-  - new_ones
-  - new_zeros
-  - nextafter
-  - norm
-  - nonzero
-  - not_equal
-  - ormqr
-  - permute
-  - pow
-  - prod
-  - qr
-  - ravel
-  - real
-  - reciprocal
-  - remainder
-  - renorm
-  - rad2deg
-  - tile
-  - repeat_interleave
-  - reshape
-  - reshape
-  - round
-  - rot90
-  - rsqrt
-  - sum_to_size
-  - scatter
-  - sgn
-  - short
-  - sigmoid
-  - sign
-  - signbit
-  - sin
-  - sinc
-  - sinh
-  - slogdet
-  - sort
-  - split
-  - sqrt
-  - square
-  - squeeze
-  - std
-  - subtract
-  - subtract
-  - svd
-  - swapaxes
-  - swapdims
-  - t
-  - take
-  - tan
-  - tanh
-  - trace
-  - swapaxes
-  - tile
-  - to
-  - topk
-  - tril
-  - tensor_split
-  - transpose
-  - true_divide
-  - trunc
-  - unbind
-  - unique_consecutive
-  - unsqueeze
-  - var
-  - view
-  - where
-  - xlogy
-  - from_numpy
-  - std
-  - take
-  - var
-  - all
-  - any
-  - copy
-  - diagonal
-  - flatten
-  - resize
-  - sum
-
-mint.ops:
-  - abs
-  - absolute_import
-  - add
-  - add_ex
-  - all
-  - any
-  - any_ex
-  - arange
-  - argmax
-  - avg_pool2d
-  - baddbmm
-  - baddbmm_ex
-  - batch_norm
-  - binary_cross_entropy_with_logits
-  - bitwise_and
-  - bitwise_or
-  - bitwise_xor
-  - bmm
-  - broadcast_to
-  - cat
-  - cat_ex
-  - ceil
-  - chunk
-  - clamp
-  - conv2d
-  - conv_transpose2d
-  - cos
-  - cross
-  - cummax
-  - cummin
-  - cumsum
-  - div
-  - divide
-  - dropout
-  - embedding
-  - eq
-  - erf
-  - erfinv
-  - exp
-  - flatten
-  - flip
-  - flip_ex
-  - fold
-  - full
-  - functional
-  - gather
-  - gelu
-  - greater
-  - grid_sample
-  - group_norm
-  - gt
-  - index_select
-  - interpolate
-  - isclose
-  - isfinite
-  - layer_norm
-  - le
-  - leaky_relu
-  - less
-  - less_equal
-  - linear
-  - linspace
-  - log
-  - logical_and
-  - logical_not
-  - logical_or
-  - lt
-  - masked_select
-  - matmul
-  - max
-  - max_pool2d
-  - maximum
-  - mean
-  - mean_ex
-  - min
-  - minimum
-  - mul
-  - ne
-  - neg
-  - negative
-  - nn
-  - nonzero
-  - normal
-  - one_hot
-  - ones
-  - ones_ex
-  - ones_like
-  - pad
-  - permute
-  - permute_ex
-  - pow
-  - prod
-  - reciprocal
-  - relu
-  - remainder
-  - repeat_interleave
-  - rsqrt
-  - scatter
-  - scatter_add
-  - searchsorted
-  - sigmoid
-  - silu
-  - sin
-  - softmax
-  - softplus
-  - sort
-  - split
-  - sqrt
-  - sqrt_ex
-  - square
-  - stack
-  - sub
-  - sub_ex
-  - sum
-  - tanh
-  - tile
-  - topk
-  - tril
-  - triu
-  - unfold
-  - unique
-  - where
-  - xlogy
-  - zeros
-  - zeros_ex
-  - zeros_like
-
-mint.nn:
-  - Dropout
-  - Embedding
-  - Fold
-  - LayerNorm
-  - Linear
-  - MaxPool2d
-  - Unfold
-  - Upsample
-
-mint.nn.functional:
-  - absolute_import
-  - avg_pool2d
-  - batch_norm
-  - batch_norm_ex
-  - bce_with_logits
-  - binary_cross_entropy_with_logits
-  - conv_transpose2d
-  - dense
-  - dropout
-  - embedding
-  - fold
-  - gelu
-  - grid_sample
-  - group_norm
-  - interpolate
-  - layer_norm
-  - leaky_relu
-  - linear
-  - max_pool2d
-  - max_pool2d_ex
-  - normal
-  - one_hot
-  - one_hot_ext
-  - pad
-  - relu
-  - sigmoid
-  - silu
-  - softmax
-  - softmax_ex
-  - softplus
-  - tanh
-  - unfold
+# Copyright 2024 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+# List of ops that register hooks
+
+  
+ops:
+  - adaptive_avg_pool1d
+  - adaptive_avg_pool2d
+  - adaptive_avg_pool3d
+  - adaptive_max_pool1d
+  - adaptive_max_pool2d
+  - avg_pool1d
+  - avg_pool2d
+  - avg_pool3d
+  - batch_norm
+  - bias_add
+  - ctc_greedy_decoder
+  - conv1d
+  - conv2d
+  - conv3d
+  - deformable_conv2d
+  - dense
+  - dropout
+  - dropout1d
+  - dropout2d
+  - dropout3d
+  - flatten
+  - fold
+  - fractional_max_pool3d
+  - lp_pool1d
+  - lp_pool2d
+  - lrn
+  - max_pool2d
+  - max_pool3d
+  - max_unpool1d
+  - max_unpool2d
+  - max_unpool3d
+  - unfold
+  - binary_cross_entropy
+  - binary_cross_entropy_with_logits
+  - cosine_embedding_loss
+  - cross_entropy
+  - ctc_loss
+  - gaussian_nll_loss
+  - hinge_embedding_loss
+  - huber_loss
+  - kl_div
+  - l1_loss
+  - margin_ranking_loss
+  - mse_loss
+  - multi_margin_loss
+  - multilabel_margin_loss
+  - multilabel_soft_margin_loss
+  - nll_loss
+  - smooth_l1_loss
+  - triplet_margin_loss
+  - elu
+  - fast_gelu
+  - gelu
+  - glu
+  - gumbel_softmax
+  - hardshrink
+  - hardsigmoid
+  - hardswish
+  - hardtanh
+  - leaky_relu
+  - log_softmax
+  - logsigmoid
+  - mish
+  - prelu
+  - relu
+  - relu6
+  - celu
+  - rrelu
+  - selu
+  - sigmoid
+  - silu
+  - softmax
+  - softmin
+  - softshrink
+  - softsign
+  - tanh
+  - threshold
+  - cdist
+  - dist
+  - pdist
+  - choice_with_mask
+  - random_categorical
+  - log_uniform_candidate_sampler
+  - uniform_candidate_sampler
+  - affine_grid
+  - bounding_box_decode
+  - bounding_box_encode
+  - col2im
+  - check_valid
+  - crop_and_resize
+  - grid_sample
+  - interpolate
+  - iou
+  - pad
+  - padding
+  - pixel_shuffle
+  - pixel_unshuffle
+  - upsample
+  - abs
+  - absolute
+  - accumulate_n
+  - acos
+  - arccos
+  - acosh
+  - add
+  - addcdiv
+  - addcmul
+  - addmv
+  - addn
+  - angle
+  - arccosh
+  - arcsin
+  - arcsinh
+  - arctan
+  - arctanh
+  - arctan2
+  - asin
+  - asinh
+  - atan
+  - atan2
+  - atanh
+  - atleast_1d
+  - atleast_2d
+  - atleast_3d
+  - bessel_i0
+  - bessel_i0e
+  - bessel_i1
+  - bessel_i1e
+  - bessel_j0
+  - bessel_j1
+  - bessel_k0
+  - bessel_k0e
+  - bessel_k1
+  - bessel_k1e
+  - bessel_y0
+  - bessel_y1
+  - bitwise_and
+  - bitwise_left_shift
+  - bitwise_or
+  - bitwise_right_shift
+  - bitwise_xor
+  - ceil
+  - clamp
+  - clip
+  - combinations
+  - copysign
+  - cos
+  - cosh
+  - cosine_similarity
+  - cov
+  - diag_embed
+  - diff
+  - deg2rad
+  - digamma
+  - div
+  - divide
+  - erf
+  - erfc
+  - erfinv
+  - exp
+  - exp2
+  - expm1
+  - floor
+  - floor_div
+  - floor_mod
+  - float_power
+  - fmod
+  - frac
+  - gcd
+  - hypot
+  - igamma
+  - igammac
+  - imag
+  - i0
+  - inv
+  - invert
+  - lcm
+  - ldexp
+  - lerp
+  - log
+  - log2
+  - log10
+  - log1p
+  - logaddexp
+  - logaddexp2
+  - logical_and
+  - logical_not
+  - logical_or
+  - logical_xor
+  - logit
+  - mul
+  - multiply
+  - mvlgamma
+  - neg
+  - negative
+  - nextafter
+  - polar
+  - polygamma
+  - positive
+  - pow
+  - rad2deg
+  - ravel
+  - real
+  - reciprocal
+  - remainder
+  - rot90
+  - round
+  - rsqrt
+  - sgn
+  - sign
+  - signbit
+  - sin
+  - sinc
+  - sinh
+  - sqrt
+  - square
+  - sub
+  - subtract
+  - t
+  - tan
+  - tanhshrink
+  - trapz
+  - tril_indices
+  - triu_indices
+  - true_divide
+  - trunc
+  - truncate_div
+  - truncate_mod
+  - xdivy
+  - xlogy
+  - zeta
+  - all
+  - amax
+  - amin
+  - aminmax
+  - any
+  - argmax
+  - argmin
+  - cummax
+  - cummin
+  - cumprod
+  - cumsum
+  - fmax
+  - histc
+  - logsumexp
+  - max
+  - mean
+  - median
+  - min
+  - norm
+  - prod
+  - std
+  - std_mean
+  - var
+  - var_mean
+  - argsort
+  - approximate_equal
+  - equal
+  - ge
+  - greater
+  - greater_equal
+  - gt
+  - intopk
+  - isclose
+  - isfinite
+  - isinf
+  - isnan
+  - isneginf
+  - isposinf
+  - isreal
+  - is_complex
+  - le
+  - less
+  - less_equal
+  - lt
+  - maximum
+  - minimum
+  - msort
+  - ne
+  - not_equal
+  - searchsorted
+  - topk
+  - bmm
+  - addbmm
+  - addmm
+  - baddbmm
+  - addr
+  - adjoint
+  - cholesky
+  - cholesky_solve
+  - batch_dot
+  - dot
+  - eig
+  - inner
+  - inverse
+  - geqrf
+  - ger
+  - kron
+  - lu_solve
+  - lu_unpack
+  - matmul
+  - matrix_solve
+  - matrix_band_part
+  - matrix_diag
+  - matrix_diag_part
+  - matrix_set_diag
+  - mm
+  - mv
+  - outer
+  - orgqr
+  - ormqr
+  - pinv
+  - svd
+  - tensor_dot
+  - logdet
+  - slogdet
+  - qr
+  - trace
+  - bartlett_window
+  - blackman_window
+  - hamming_window
+  - hann_window
+  - kaiser_window
+  - eye
+  - fill
+  - full
+  - full_like
+  - linspace
+  - logspace
+  - one_hot
+  - range
+  - heaviside
+  - bernoulli
+  - gamma
+  - laplace
+  - multinomial
+  - multinomial_with_replacement
+  - rand
+  - rand_like
+  - randint
+  - randint_like
+  - randn
+  - randn_like
+  - random_gamma
+  - random_poisson
+  - randperm
+  - standard_laplace
+  - standard_normal
+  - uniform
+  - argwhere
+  - batch_to_space_nd
+  - bincount
+  - block_diag
+  - broadcast_to
+  - cat
+  - channel_shuffle
+  - chunk
+  - column_stack
+  - concat
+  - conj
+  - count_nonzero
+  - deepcopy
+  - diag
+  - diagflat
+  - diagonal
+  - dyn_shape
+  - dsplit
+  - dstack
+  - einsum
+  - expand
+  - expand_dims
+  - flip
+  - fliplr
+  - flipud
+  - gather_d
+  - gather_elements
+  - gather_nd
+  - hsplit
+  - hstack
+  - index_add
+  - index_fill
+  - index_select
+  - inplace_add
+  - inplace_index_add
+  - inplace_sub
+  - inplace_update
+  - masked_fill
+  - masked_select
+  - meshgrid
+  - moveaxis
+  - movedim
+  - narrow
+  - nan_to_num
+  - nansum
+  - normal
+  - nonzero
+  - population_count
+  - rank
+  - repeat_elements
+  - repeat_interleave
+  - reshape
+  - reverse
+  - reverse_sequence
+  - roll
+  - scatter
+  - scatter_nd
+  - select
+  - sequence_mask
+  - shuffle
+  - size
+  - slice
+  - sort
+  - space_to_batch_nd
+  - sparse_segment_mean
+  - split
+  - squeeze
+  - stack
+  - strided_slice
+  - sum
+  - swapaxes
+  - swapdims
+  - tensor_scatter_add
+  - tensor_scatter_div
+  - tensor_scatter_max
+  - tensor_scatter_min
+  - tensor_scatter_mul
+  - tensor_scatter_sub
+  - tensor_scatter_elements
+  - tensor_split
+  - tile
+  - tril
+  - triu
+  - transpose
+  - unbind
+  - unique
+  - unique_consecutive
+  - unique_with_pad
+  - unsorted_segment_max
+  - unsorted_segment_min
+  - unsorted_segment_prod
+  - unsorted_segment_sum
+  - unsqueeze
+  - unstack
+  - view_as_real
+  - vsplit
+  - vstack
+  - where
+  - cross
+  - renorm
+  - is_tensor
+  - scalar_cast
+  - scalar_to_tensor
+  - tuple_to_array
+  - clip_by_global_norm
+  - clip_by_value
+  - assign
+  - assign_add
+  - assign_sub
+  - scatter_add
+  - scatter_div
+  - scatter_max
+  - scatter_min
+  - scatter_mul
+  - scatter_nd_add
+  - scatter_nd_div
+  - scatter_nd_max
+  - scatter_nd_min
+  - scatter_nd_mul
+  - scatter_nd_sub
+  - scatter_update
+  - derivative
+  - jet
+
+tensor:
+  - __abs__
+  - __add__
+  - __and__
+  - __bool__
+  - __eq__
+  - __ge__
+  - __gt__
+  - __iadd__
+  - __ifloordiv__
+  - __imatmul__
+  - __imod__
+  - __imul__
+  - __isub__
+  - __le__
+  - __lt__
+  - __matmul__
+  - __mod__
+  - __mul__
+  - __ne__
+  - __neg__
+  - __or__
+  - __pow__
+  - __radd__
+  - __rmatmul__
+  - __rmod__
+  - __rmul__
+  - __rpow__
+  - __rsub__
+  - __sub__
+  - __truediv__
+  - __xor__
+  - abs
+  - absolute
+  - acos
+  - acosh
+  - add
+  - addbmm
+  - addcdiv
+  - addcmul
+  - addmm
+  - addmv
+  - addr
+  - all
+  - amax
+  - amin
+  - any
+  - arccos
+  - arccosh
+  - argmax
+  - angle
+  - arcsin
+  - arcsinh
+  - arctan
+  - arctanh
+  - argmin
+  - argsort
+  - asin
+  - asinh
+  - atan
+  - atan2
+  - atanh
+  - baddbmm
+  - bernoulli
+  - bincount
+  - bitwise_and
+  - bitwise_or
+  - bitwise_xor
+  - bmm
+  - bool
+  - broadcast_to
+  - ceil
+  - cholesky_solve
+  - cholesky
+  - clamp
+  - clip
+  - conj
+  - copysign
+  - cos
+  - cosh
+  - cross
+  - cummax
+  - cummin
+  - cumprod
+  - cumsum
+  - deg2rad
+  - diag
+  - diagflat
+  - diff
+  - digamma
+  - div
+  - divide
+  - equal
+  - erf
+  - erfc
+  - erfinv
+  - exp
+  - expand_as
+  - expm1
+  - flip
+  - fliplr
+  - flipud
+  - float_power
+  - floor
+  - fmod
+  - frac
+  - gather_elements
+  - ge
+  - geqrf
+  - ger
+  - greater
+  - greater_equal
+  - gt
+  - half
+  - hardshrink
+  - heaviside
+  - histc
+  - hypot
+  - i0
+  - igamma
+  - igammac
+  - imag
+  - index_add
+  - index_fill
+  - index_put
+  - index_select
+  - inner
+  - int
+  - inverse
+  - isclose
+  - isfinite
+  - isinf
+  - isnan
+  - is_complex
+  - is_signed
+  - isneginf
+  - isposinf
+  - isreal
+  - lcm
+  - ldexp
+  - le
+  - lerp
+  - less
+  - less_equal
+  - log
+  - log10
+  - log1p
+  - log2
+  - logaddexp
+  - logaddexp2
+  - logdet
+  - logical_and
+  - logical_not
+  - logical_or
+  - logical_xor
+  - logit
+  - logsumexp
+  - long
+  - lt
+  - masked_fill
+  - masked_scatter
+  - masked_select
+  - matmul
+  - max
+  - maximum
+  - mean
+  - median
+  - min
+  - minimum
+  - moveaxis
+  - movedim
+  - msort
+  - multinomial
+  - multiply
+  - mvlgamma
+  - nan_to_num
+  - nansum
+  - narrow
+  - ne
+  - neg
+  - negative
+  - nelement
+  - new_ones
+  - new_zeros
+  - nextafter
+  - norm
+  - nonzero
+  - not_equal
+  - ormqr
+  - permute
+  - pow
+  - prod
+  - qr
+  - ravel
+  - real
+  - reciprocal
+  - remainder
+  - renorm
+  - rad2deg
+  - tile
+  - repeat_interleave
+  - reshape
+  - reshape
+  - round
+  - rot90
+  - rsqrt
+  - sum_to_size
+  - scatter
+  - sgn
+  - short
+  - sigmoid
+  - sign
+  - signbit
+  - sin
+  - sinc
+  - sinh
+  - slogdet
+  - sort
+  - split
+  - sqrt
+  - square
+  - squeeze
+  - std
+  - subtract
+  - subtract
+  - svd
+  - swapaxes
+  - swapdims
+  - t
+  - take
+  - tan
+  - tanh
+  - trace
+  - swapaxes
+  - tile
+  - topk
+  - tril
+  - tensor_split
+  - transpose
+  - true_divide
+  - trunc
+  - unbind
+  - unique_consecutive
+  - unsqueeze
+  - var
+  - view
+  - where
+  - xlogy
+  - from_numpy
+  - std
+  - take
+  - var
+  - all
+  - any
+  - copy
+  - diagonal
+  - flatten
+  - resize
+  - sum
+
+mint.ops:
+  - abs
+  - absolute_import
+  - add
+  - add_ex
+  - all
+  - any
+  - any_ex
+  - argmax
+  - avg_pool2d
+  - baddbmm
+  - baddbmm_ex
+  - batch_norm
+  - binary_cross_entropy_with_logits
+  - bitwise_and
+  - bitwise_or
+  - bitwise_xor
+  - bmm
+  - broadcast_to
+  - cat
+  - cat_ex
+  - ceil
+  - chunk
+  - clamp
+  - conv2d
+  - conv_transpose2d
+  - cos
+  - cross
+  - cummax
+  - cummin
+  - cumsum
+  - div
+  - divide
+  - dropout
+  - embedding
+  - eq
+  - erf
+  - erfinv
+  - exp
+  - flatten
+  - flip
+  - flip_ex
+  - fold
+  - full
+  - functional
+  - gather
+  - gelu
+  - greater
+  - grid_sample
+  - group_norm
+  - gt
+  - index_select
+  - interpolate
+  - isclose
+  - isfinite
+  - layer_norm
+  - le
+  - leaky_relu
+  - less
+  - less_equal
+  - linear
+  - linspace
+  - log
+  - logical_and
+  - logical_not
+  - logical_or
+  - lt
+  - masked_select
+  - matmul
+  - max
+  - max_pool2d
+  - maximum
+  - mean
+  - mean_ex
+  - min
+  - minimum
+  - mul
+  - ne
+  - neg
+  - negative
+  - nn
+  - nonzero
+  - normal
+  - one_hot
+  - ones
+  - ones_ex
+  - ones_like
+  - pad
+  - permute
+  - permute_ex
+  - pow
+  - prod
+  - reciprocal
+  - relu
+  - remainder
+  - repeat_interleave
+  - rsqrt
+  - scatter
+  - scatter_add
+  - searchsorted
+  - sigmoid
+  - silu
+  - sin
+  - softmax
+  - softplus
+  - sort
+  - split
+  - sqrt
+  - sqrt_ex
+  - square
+  - stack
+  - sub
+  - sub_ex
+  - sum
+  - tanh
+  - tile
+  - topk
+  - tril
+  - triu
+  - unfold
+  - unique
+  - where
+  - xlogy
+  - zeros
+  - zeros_ex
+  - zeros_like
+
+mint.nn:
+  - Dropout
+  - Embedding
+  - Fold
+  - LayerNorm
+  - Linear
+  - MaxPool2d
+  - Unfold
+  - Upsample
+
+mint.nn.functional:
+  - absolute_import
+  - avg_pool2d
+  - batch_norm
+  - batch_norm_ex
+  - bce_with_logits
+  - binary_cross_entropy_with_logits
+  - conv_transpose2d
+  - dense
+  - dropout
+  - embedding
+  - fold
+  - gelu
+  - grid_sample
+  - group_norm
+  - interpolate
+  - layer_norm
+  - leaky_relu
+  - linear
+  - max_pool2d
+  - max_pool2d_ex
+  - normal
+  - one_hot
+  - one_hot_ext
+  - pad
+  - relu
+  - sigmoid
+  - silu
+  - softmax
+  - softmax_ex
+  - softplus
+  - tanh
+  - unfold