.. _pytorch-neuron-supported-operators:

PyTorch Neuron (``torch-neuronx``) - Supported Operators
========================================================


.. contents:: Table of Contents
   :local:
   :depth: 2

Operator support
~~~~~~~~~~~~~~~~

The following list the aten operators supported by torch-neuronx.

+----------------------------------+
| aten::_s_where                   |
+----------------------------------+
| aten::_softmax                   |
+----------------------------------+
| aten::_softmax_backward_data     |
+----------------------------------+
| aten::_unsafe_view               |
+----------------------------------+
| aten::add                        |
+----------------------------------+
| aten::addcdiv\_                  |
+----------------------------------+
| aten::addcmul                    |
+----------------------------------+
| aten::addmm                      |
+----------------------------------+
| aten::bernoulli\_                |
+----------------------------------+
| aten::bmm                        |
+----------------------------------+
| aten::constant_pad_nd            |
+----------------------------------+
| aten::div                        |
+----------------------------------+
| aten::embedding                  |
+----------------------------------+
| aten::embedding_dense_backward   |
+----------------------------------+
| aten::empty                      |
+----------------------------------+
| aten::expand                     |
+----------------------------------+
| aten::fill\_                     |
+----------------------------------+
| aten::index_select               |
+----------------------------------+
| aten::_log_softmax               |
+----------------------------------+
| aten::_log_softmax_backward_data |
+----------------------------------+
| aten::lt                         |
+----------------------------------+
| aten::mm                         |
+----------------------------------+
| aten::mul                        |
+----------------------------------+
| aten::native_batch_norm          |
+----------------------------------+
| aten::native_batch_norm_backward |
+----------------------------------+
| aten::neg                        |
+----------------------------------+
| aten::permute                    |
+----------------------------------+
| aten::relu                       |
+----------------------------------+
| aten::rsub                       |
+----------------------------------+
| aten::select                     |
+----------------------------------+
| aten::slice                      |
+----------------------------------+
| aten::sqrt                       |
+----------------------------------+
| aten::sum                        |
+----------------------------------+
| aten::t                          |
+----------------------------------+
| aten::tanh                       |
+----------------------------------+
| aten::tanh_backward              |
+----------------------------------+
| aten::threshold_backward         |
+----------------------------------+
| aten::transpose                  |
+----------------------------------+
| aten::unsqueeze                  |
+----------------------------------+
| aten::view                       |
+----------------------------------+
| aten::zero\_                     |
+----------------------------------+