[DOCS] Update supported ops by PT FE (#21961)
This commit is contained in:
Maxim Vafin
GitHub
parent
cc8fadf187
commit
7835368489
|
|
@ -18,239 +18,372 @@ This page lists operations supported by OpenVINO Framework Frontend.
|
|||
========================================== ==========================================================================================
|
||||
PyTorch Supported Operations Limitations
|
||||
========================================== ==========================================================================================
|
||||
aten::__and__ Only for boolean inputs
|
||||
aten::__getitem__ Supported in limitted set of patterns
|
||||
aten::__not__
|
||||
aten::__or__
|
||||
aten::_convolution
|
||||
aten::_convolution_mode
|
||||
aten::_set_item
|
||||
aten::abs
|
||||
aten::acos
|
||||
aten::acos_
|
||||
aten::acosh
|
||||
aten::acosh_
|
||||
aten::adaptive_avg_pool2d
|
||||
aten::adaptive_avg_pool3d
|
||||
aten::adaptive_max_pool2d
|
||||
aten::add
|
||||
aten::add_
|
||||
aten::addcmul
|
||||
aten::addmm
|
||||
aten::append Supported in limitted set of patterns
|
||||
aten::arange
|
||||
aten::argsort
|
||||
aten::as_tensor
|
||||
aten::asin
|
||||
aten::asin_
|
||||
aten::asinh
|
||||
aten::asinh_
|
||||
aten::atan
|
||||
aten::atan_
|
||||
aten::atanh
|
||||
aten::atanh_
|
||||
aten::avg_pool1d
|
||||
aten::avg_pool2d
|
||||
aten::avg_pool3d
|
||||
aten::baddbmm
|
||||
aten::batch_norm
|
||||
aten::bitwise_not
|
||||
aten::bmm
|
||||
aten::Bool
|
||||
aten::cat
|
||||
aten::ceil
|
||||
aten::ceil_
|
||||
aten::chunk Supported in limitted set of patterns
|
||||
aten::clamp
|
||||
aten::clamp_max
|
||||
aten::clamp_min
|
||||
aten::clone
|
||||
aten::contiguous
|
||||
aten::conv_transpose1d
|
||||
aten::conv_transpose2d
|
||||
aten::conv_transpose3d
|
||||
aten::conv1d
|
||||
aten::conv2d
|
||||
aten::conv3d
|
||||
aten::convolution
|
||||
aten::copy
|
||||
aten::cos
|
||||
aten::cos_
|
||||
aten::cosh
|
||||
aten::cosh_
|
||||
aten::cumsum
|
||||
aten::detach
|
||||
aten::dim
|
||||
aten::div
|
||||
aten::div_
|
||||
aten::dropout
|
||||
aten::dropout_
|
||||
aten::einsum Supported in limitted set of patterns
|
||||
aten::elu
|
||||
aten::embedding
|
||||
aten::empty
|
||||
aten::eq
|
||||
aten::exp
|
||||
aten::expand Supported in limitted set of patterns
|
||||
aten::expand_as
|
||||
aten::eye
|
||||
aten::fill_
|
||||
aten::flatten
|
||||
aten::floor
|
||||
aten::floor_
|
||||
aten::floor_divide
|
||||
aten::floordiv
|
||||
aten::full
|
||||
aten::full_like
|
||||
aten::gather
|
||||
aten::ge
|
||||
aten::gelu
|
||||
aten::glu
|
||||
aten::grid_sampler
|
||||
aten::group_norm
|
||||
aten::gt
|
||||
aten::hardsigmoid
|
||||
aten::hardswish
|
||||
aten::hardswish_
|
||||
aten::hardtanh
|
||||
aten::hardtanh_
|
||||
aten::im2col
|
||||
aten::index Supported in limitted set of patterns
|
||||
aten::index_put_ Supported in limitted set of patterns
|
||||
aten::index_select
|
||||
aten::instance_norm
|
||||
aten::Int
|
||||
aten::IntImplicit
|
||||
aten::is_grad_enabled
|
||||
aten::layer_norm
|
||||
aten::le
|
||||
aten::leaky_relu
|
||||
aten::leaky_relu_
|
||||
aten::len
|
||||
aten::linalg_matrix_norm
|
||||
aten::linalg_norm
|
||||
aten::linalg_vector_norm
|
||||
aten::linear
|
||||
aten::log
|
||||
aten::log_
|
||||
aten::log2
|
||||
aten::log2_
|
||||
aten::lt
|
||||
aten::masked_fill
|
||||
aten::masked_fill_
|
||||
aten::matmul
|
||||
aten::max
|
||||
aten::max_pool1d
|
||||
aten::max_pool2d
|
||||
aten::max_pool3d
|
||||
aten::mean
|
||||
aten::meshgrid Supported in limitted set of patterns
|
||||
aten::min
|
||||
aten::mm
|
||||
aten::mul
|
||||
aten::mul_
|
||||
aten::narrow
|
||||
aten::ne
|
||||
aten::neg
|
||||
aten::new_empty
|
||||
aten::new_full
|
||||
aten::new_ones
|
||||
aten::new_zeros
|
||||
aten::nonzero
|
||||
aten::nonzero_numpy Supported in limitted set of patterns
|
||||
aten::norm
|
||||
aten::numel
|
||||
aten::ones
|
||||
aten::ones_like
|
||||
aten::pad Supported in limitted set of patterns
|
||||
aten::permute
|
||||
aten::pow
|
||||
aten::reciprocal
|
||||
aten::relu
|
||||
aten::relu_
|
||||
aten::relu6
|
||||
aten::remainder
|
||||
aten::repeat
|
||||
aten::repeat_interleave
|
||||
aten::reshape
|
||||
aten::reshape_as
|
||||
aten::roll
|
||||
aten::rsqrt
|
||||
aten::rsub
|
||||
aten::ScalarImplicit
|
||||
aten::scaled_dot_product_attention
|
||||
aten::select
|
||||
aten::selu
|
||||
aten::selu_
|
||||
aten::sigmoid
|
||||
aten::sigmoid_
|
||||
aten::sign
|
||||
aten::silu
|
||||
aten::silu_
|
||||
aten::sin
|
||||
aten::sin_
|
||||
aten::sinh
|
||||
aten::sinh_
|
||||
aten::size
|
||||
aten::slice
|
||||
aten::softmax
|
||||
aten::sort
|
||||
aten::split Supported in limitted set of patterns
|
||||
aten::split_with_sizes Supported in limitted set of patterns
|
||||
aten::sqrt
|
||||
aten::square
|
||||
aten::squeeze
|
||||
aten::stack Supported in limitted set of patterns
|
||||
aten::sub
|
||||
aten::sum
|
||||
aten::tan
|
||||
aten::tan_
|
||||
aten::tanh
|
||||
aten::tanh_
|
||||
aten::tensor
|
||||
aten::to
|
||||
aten::topk
|
||||
aten::transpose
|
||||
aten::tril
|
||||
aten::triu
|
||||
aten::type_as
|
||||
aten::unbind Supported in limitted set of patterns
|
||||
aten::unfold
|
||||
aten::unsqueeze
|
||||
aten::unsqueeze_
|
||||
aten::upsample_bicubic2d
|
||||
aten::upsample_bilinear2d
|
||||
aten::upsample_linear1d
|
||||
aten::upsample_nearest1d
|
||||
aten::upsample_nearest2d
|
||||
aten::upsample_nearest3d
|
||||
aten::upsample_trilinear3d
|
||||
aten::var
|
||||
aten::var_mean
|
||||
aten::view
|
||||
aten::where Supported in limitted set of patterns
|
||||
aten::zeros
|
||||
aten::zeros_like
|
||||
prim::Constant String and None constant is only supported when used by supported operation
|
||||
prim::device
|
||||
prim::DictConstruct Supported in limitted set of patterns
|
||||
prim::dtype Supported in limitted set of patterns
|
||||
prim::GetAttr
|
||||
prim::If
|
||||
prim::is_cuda
|
||||
prim::ListConstruct Supported in limitted set of patterns
|
||||
prim::ListUnpack Supported in limitted set of patterns
|
||||
prim::Loop
|
||||
prim::max Supported in limitted set of patterns
|
||||
prim::min Supported in limitted set of patterns
|
||||
prim::NumToTensor
|
||||
prim::PythonOp Supported only in tracing
|
||||
prim::requires_grad
|
||||
prim::TupleConstruct Supported in limitted set of patterns
|
||||
prim::type
|
||||
torchvision::deform_conv2d
|
||||
torchvision::nms
|
||||
torchvision::roi_align
|
||||
aten::__and__
|
||||
aten::__derive_index
|
||||
aten::__getitem__
|
||||
aten::__not__
|
||||
aten::__or__
|
||||
aten::__range_length
|
||||
aten::__xor__
|
||||
aten::_convolution
|
||||
aten::_convolution_mode
|
||||
aten::_native_multi_head_attention
|
||||
aten::_set_item
|
||||
aten::_shape_as_tensor
|
||||
aten::_upsample_bicubic2d_aa
|
||||
aten::_upsample_bilinear2d_aa
|
||||
aten::_weight_norm
|
||||
aten::abs
|
||||
aten::acos
|
||||
aten::acos_
|
||||
aten::acosh
|
||||
aten::acosh_
|
||||
aten::adaptive_avg_pool1d
|
||||
aten::adaptive_avg_pool2d
|
||||
aten::adaptive_avg_pool3d
|
||||
aten::adaptive_max_pool1d
|
||||
aten::adaptive_max_pool2d
|
||||
aten::adaptive_max_pool3d
|
||||
aten::add
|
||||
aten::add_
|
||||
aten::addcmul
|
||||
aten::addmm
|
||||
aten::alias
|
||||
aten::alias_copy
|
||||
aten::all
|
||||
aten::amax
|
||||
aten::amin
|
||||
aten::append Supported in limited set of patterns
|
||||
aten::arange
|
||||
aten::argmax
|
||||
aten::argmin
|
||||
aten::argsort
|
||||
aten::as_strided
|
||||
aten::as_tensor
|
||||
aten::asin
|
||||
aten::asin_
|
||||
aten::asinh
|
||||
aten::asinh_
|
||||
aten::atan
|
||||
aten::atan_
|
||||
aten::atanh
|
||||
aten::atanh_
|
||||
aten::avg_pool1d
|
||||
aten::avg_pool2d
|
||||
aten::avg_pool3d
|
||||
aten::baddbmm
|
||||
aten::batch_norm
|
||||
aten::bitwise_and
|
||||
aten::bitwise_not
|
||||
aten::bitwise_or
|
||||
aten::bitwise_xor
|
||||
aten::bmm
|
||||
aten::Bool
|
||||
aten::broadcast_tensors Supported in limited set of patterns
|
||||
aten::broadcast_to
|
||||
aten::cat
|
||||
aten::cdist
|
||||
aten::ceil
|
||||
aten::ceil_
|
||||
aten::channel_shuffle
|
||||
aten::chunk Supported in limited set of patterns
|
||||
aten::clamp
|
||||
aten::clamp_
|
||||
aten::clamp_max
|
||||
aten::clamp_min
|
||||
aten::clip
|
||||
aten::clip_
|
||||
aten::clone
|
||||
aten::complex Supported in limited set of patterns
|
||||
aten::concat
|
||||
aten::contiguous
|
||||
aten::conv1d
|
||||
aten::conv2d
|
||||
aten::conv3d
|
||||
aten::conv_transpose1d
|
||||
aten::conv_transpose2d
|
||||
aten::conv_transpose3d
|
||||
aten::convolution
|
||||
aten::copy
|
||||
aten::copy_
|
||||
aten::cos
|
||||
aten::cos_
|
||||
aten::cosh
|
||||
aten::cosh_
|
||||
aten::cross
|
||||
aten::cumsum
|
||||
aten::dequantize
|
||||
aten::detach
|
||||
aten::dim
|
||||
aten::div
|
||||
aten::div_
|
||||
aten::dropout
|
||||
aten::dropout_
|
||||
aten::einsum Supported in limited set of patterns
|
||||
aten::elu
|
||||
aten::embedding
|
||||
aten::embedding_bag
|
||||
aten::empty
|
||||
aten::empty_like
|
||||
aten::eq
|
||||
aten::erf
|
||||
aten::erf_
|
||||
aten::erfc
|
||||
aten::erfc_
|
||||
aten::exp
|
||||
aten::exp_
|
||||
aten::expand
|
||||
aten::expand_as
|
||||
aten::eye
|
||||
aten::fake_quantize_per_channel_affine
|
||||
aten::fake_quantize_per_tensor_affine
|
||||
aten::feature_dropout
|
||||
aten::fft_irfftn Supported in limited set of patterns
|
||||
aten::fft_rfftn Supported in limited set of patterns
|
||||
aten::fill
|
||||
aten::fill_
|
||||
aten::fill_diagonal_
|
||||
aten::flatten
|
||||
aten::flip
|
||||
aten::floor
|
||||
aten::floor_
|
||||
aten::floor_divide
|
||||
aten::floordiv
|
||||
aten::fmod
|
||||
aten::frobenius_norm
|
||||
aten::full
|
||||
aten::full_like
|
||||
aten::gather
|
||||
aten::ge
|
||||
aten::gelu
|
||||
aten::glu
|
||||
aten::grid_sampler
|
||||
aten::group_norm
|
||||
aten::gru
|
||||
aten::gt
|
||||
aten::hardsigmoid
|
||||
aten::hardswish
|
||||
aten::hardswish_
|
||||
aten::hardtanh
|
||||
aten::hardtanh_
|
||||
aten::im2col
|
||||
aten::imag Supported in limited set of patterns
|
||||
aten::index Supported in limited set of patterns
|
||||
aten::index_put_
|
||||
aten::index_select
|
||||
aten::instance_norm
|
||||
aten::Int
|
||||
aten::IntImplicit
|
||||
aten::is_grad_enabled
|
||||
aten::is_nonzero
|
||||
aten::item
|
||||
aten::layer_norm
|
||||
aten::le
|
||||
aten::leaky_relu
|
||||
aten::leaky_relu_
|
||||
aten::len
|
||||
aten::lift
|
||||
aten::lift_fresh
|
||||
aten::lift_fresh_copy
|
||||
aten::linalg_cross
|
||||
aten::linalg_matrix_norm
|
||||
aten::linalg_norm
|
||||
aten::linalg_vector_norm
|
||||
aten::linear
|
||||
aten::linspace
|
||||
aten::log
|
||||
aten::log10
|
||||
aten::log10_
|
||||
aten::log1p
|
||||
aten::log1p_
|
||||
aten::log2
|
||||
aten::log2_
|
||||
aten::log_
|
||||
aten::log_softmax
|
||||
aten::logical_and
|
||||
aten::logical_not
|
||||
aten::logical_or
|
||||
aten::logical_xor
|
||||
aten::lstm
|
||||
aten::lt
|
||||
aten::masked_fill
|
||||
aten::masked_fill_
|
||||
aten::masked_scatter
|
||||
aten::masked_scatter_
|
||||
aten::matmul
|
||||
aten::max
|
||||
aten::max_pool1d
|
||||
aten::max_pool1d_with_indices
|
||||
aten::max_pool2d
|
||||
aten::max_pool2d_with_indices
|
||||
aten::max_pool3d
|
||||
aten::max_pool3d_with_indices
|
||||
aten::maximum
|
||||
aten::mean
|
||||
aten::meshgrid
|
||||
aten::min
|
||||
aten::minimum
|
||||
aten::mm
|
||||
aten::mul
|
||||
aten::mul_
|
||||
aten::multinomial
|
||||
aten::multiply
|
||||
aten::multiply_
|
||||
aten::narrow
|
||||
aten::ne
|
||||
aten::neg
|
||||
aten::new_empty
|
||||
aten::new_full
|
||||
aten::new_ones
|
||||
aten::new_zeros
|
||||
aten::nonzero
|
||||
aten::nonzero_numpy Supported in limited set of patterns
|
||||
aten::norm
|
||||
aten::normal
|
||||
aten::normal_
|
||||
aten::numel
|
||||
aten::numpy_T
|
||||
aten::one_hot
|
||||
aten::ones
|
||||
aten::ones_like
|
||||
aten::outer
|
||||
aten::pad
|
||||
aten::pairwise_distance
|
||||
aten::permute
|
||||
aten::pixel_shuffle
|
||||
aten::pixel_unshuffle
|
||||
aten::pow
|
||||
aten::pow_
|
||||
aten::prelu
|
||||
aten::prod
|
||||
aten::quantize_per_channel
|
||||
aten::quantize_per_tensor
|
||||
aten::rand
|
||||
aten::rand_like
|
||||
aten::randint
|
||||
aten::randn
|
||||
aten::randn_like
|
||||
aten::real Supported in limited set of patterns
|
||||
aten::reciprocal
|
||||
aten::reflection_pad2d Supported in limited set of patterns
|
||||
aten::relu
|
||||
aten::relu6
|
||||
aten::relu_
|
||||
aten::remainder
|
||||
aten::repeat
|
||||
aten::repeat_interleave
|
||||
aten::reshape
|
||||
aten::reshape_as
|
||||
aten::resolve_conj
|
||||
aten::resolve_neg
|
||||
aten::rnn_relu
|
||||
aten::rnn_tanh
|
||||
aten::roll
|
||||
aten::round
|
||||
aten::rsqrt
|
||||
aten::rsub
|
||||
aten::ScalarImplicit
|
||||
aten::scaled_dot_product_attention
|
||||
aten::scatter
|
||||
aten::scatter_
|
||||
aten::scatter_add
|
||||
aten::scatter_add_
|
||||
aten::scatter_reduce
|
||||
aten::scatter_reduce_
|
||||
aten::select
|
||||
aten::selu
|
||||
aten::selu_
|
||||
aten::sigmoid
|
||||
aten::sigmoid_
|
||||
aten::sign
|
||||
aten::silu
|
||||
aten::silu_
|
||||
aten::sin
|
||||
aten::sin_
|
||||
aten::sinh
|
||||
aten::sinh_
|
||||
aten::size
|
||||
aten::slice
|
||||
aten::softmax
|
||||
aten::softplus
|
||||
aten::sort
|
||||
aten::split Supported in limited set of patterns
|
||||
aten::split_with_sizes Supported in limited set of patterns
|
||||
aten::sqrt
|
||||
aten::square
|
||||
aten::squeeze
|
||||
aten::stack Supported in limited set of patterns
|
||||
aten::std
|
||||
aten::std_mean
|
||||
aten::sub
|
||||
aten::sub_
|
||||
aten::sum
|
||||
aten::swapaxes
|
||||
aten::t
|
||||
aten::t_
|
||||
aten::take_along_dim
|
||||
aten::tan
|
||||
aten::tan_
|
||||
aten::tanh
|
||||
aten::tanh_
|
||||
aten::tensor
|
||||
aten::tensor_split Supported in limited set of patterns
|
||||
aten::tile
|
||||
aten::to
|
||||
aten::topk
|
||||
aten::transpose
|
||||
aten::tril
|
||||
aten::tril_
|
||||
aten::triu
|
||||
aten::triu_
|
||||
aten::type_as
|
||||
aten::unbind Supported in limited set of patterns
|
||||
aten::unflatten
|
||||
aten::unfold
|
||||
aten::unsqueeze
|
||||
aten::unsqueeze_
|
||||
aten::upsample_bicubic2d
|
||||
aten::upsample_bilinear2d
|
||||
aten::upsample_linear1d
|
||||
aten::upsample_nearest1d
|
||||
aten::upsample_nearest2d
|
||||
aten::upsample_nearest3d
|
||||
aten::upsample_trilinear3d
|
||||
aten::var
|
||||
aten::var_mean
|
||||
aten::view
|
||||
aten::view_as
|
||||
aten::where
|
||||
aten::zero_
|
||||
aten::zeros
|
||||
aten::zeros_like
|
||||
prim::Constant
|
||||
prim::device
|
||||
prim::DictConstruct Supported in limited set of patterns
|
||||
prim::GetAttr
|
||||
prim::If
|
||||
prim::is_cuda
|
||||
prim::ListConstruct
|
||||
prim::ListUnpack
|
||||
prim::Loop
|
||||
prim::NumToTensor
|
||||
prim::PythonOp
|
||||
prim::requires_grad
|
||||
prim::TupleConstruct Supported in limited set of patterns
|
||||
prim::TupleIndex
|
||||
prim::TupleUnpack Supported in limited set of patterns
|
||||
prim::type
|
||||
quantized::add
|
||||
quantized::add_relu
|
||||
quantized::cat
|
||||
quantized::conv2d
|
||||
quantized::conv2d_relu
|
||||
quantized::hardswish
|
||||
quantized::linear
|
||||
quantized::mul
|
||||
torchvision::deform_conv2d
|
||||
torchvision::nms
|
||||
torchvision::roi_align
|
||||
========================================== ==========================================================================================
|
||||
|
||||
.. tab-item:: ONNX
|
||||
|
|
|
|||
Loading…
Reference in New Issue