From 7835368489444fe7a73cfab1992ea1cfe2d31e0d Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Mon, 8 Jan 2024 11:11:16 +0100 Subject: [PATCH] [DOCS] Update supported ops by PT FE (#21961) --- ...upported_operations_framework_frontend.rst | 599 +++++++++++------- 1 file changed, 366 insertions(+), 233 deletions(-) diff --git a/docs/articles_en/about_openvino/compatibility_and_support/supported_operations_framework_frontend.rst b/docs/articles_en/about_openvino/compatibility_and_support/supported_operations_framework_frontend.rst index e3a9a46c3d8..1c4edeac967 100644 --- a/docs/articles_en/about_openvino/compatibility_and_support/supported_operations_framework_frontend.rst +++ b/docs/articles_en/about_openvino/compatibility_and_support/supported_operations_framework_frontend.rst @@ -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