Skip to content

Latest commit

 

History

History
162 lines (157 loc) · 1.75 KB

OperatorList.md

File metadata and controls

162 lines (157 loc) · 1.75 KB

Operator List

v1.0

  • addmm
  • bmm
  • matmul
  • mm
  • cumsum
  • linear
  • dropout
  • gelu
  • layer_norm
  • native_dropout
  • native_layer_norm
  • relu
  • silu
  • softmax
  • triu
  • abs
  • add
  • div
  • exp
  • mul
  • pow
  • reciprocal
  • rsqrt
  • rsub
  • sub
  • mean

v2.0

  • all
  • any
  • bitwise_and
  • bitwise_not
  • bitwise_or
  • cos
  • clamp
  • eq
  • ge
  • gt
  • mv
  • isinf
  • isnan
  • le
  • lt
  • ne
  • neg
  • or_
  • sin
  • tanh
  • amax
  • argmax
  • max
  • min
  • outer
  • prod
  • sum
  • var_mean
  • vector_norm
  • CrossEntropyLoss
  • group_norm
  • log_softmax
  • native_group_norm
  • sigmoid

v3.0

  • _conv_depthwise2d
  • _convolution
  • conv1d
  • conv2d
  • convolution
  • cudnn_convolution (N/A)
  • multinomial
  • nonzero
  • normal
  • rand
  • rand_like
  • randn
  • topk
  • uniform_
  • _efficient_attention_forward
  • _scaled_dot_product_efficient_attention
  • embedding
  • nll_loss
  • nll_loss_forward
  • nll_loss_nd
  • scaled_dot_product_attention
  • upsample_nearest2d
  • _fft_c2r
  • _fft_r2c
  • conj_physical
  • erf
  • fft_irfft
  • fft_rfft
  • resolve_conj
  • resolve_neg
  • arange
  • cat
  • chunk (N/A)
  • concat
  • constant_pad_nd
  • contiguous (N/A)
  • copy_ (N/A)
  • fill
  • flip
  • full
  • full_like
  • gather
  • hstack
  • index_put_
  • index_select
  • masked_fill
  • narrow
  • ones
  • pad
  • permute (N/A)
  • repeat
  • repeat_interleave
  • resize_ (N/A)
  • scatter
  • index_add
  • select (N/A)
  • select_scatter
  • slice
  • slice_scatter
  • sort
  • split (N/A)
  • split_with_sizes (N/A)
  • stack
  • tile
  • transpose (N/A)
  • unfold (N/A)
  • where
  • zeros

v4.0

  • rms_norm
  • skip_rms_norm
  • skip_layer_norm
  • gelu_and_mul
  • silu_and_mul
  • apply_rotary_pos_emb

v5.0

  • unique
  • isin
  • allclose
  • isclose
  • isfinite
  • exponential_
  • vstack
  • weight_norm_interface
  • minimum
  • maximum
  • ones_like
  • zeros_like
  • randn_like
  • floor_divide
  • masked_select
  • trunc_divide
  • remainder