Skip to content

Commit d0950c8

Browse files
narendasannarendasan
committed
docs: Update op list
Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
1 parent fe08950 commit d0950c8

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

‎docsrc/indices/supported_ops.rst‎

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,10 @@ Operators Currently Supported Through Converters
1616
- aten::acosh(Tensor self) -> (Tensor)
1717
- aten::adaptive_avg_pool1d(Tensor self, int[1] output_size) -> (Tensor)
1818
- aten::adaptive_avg_pool2d(Tensor self, int[2] output_size) -> (Tensor)
19+
- aten::adaptive_avg_pool3d(Tensor self, int[3] output_size) -> (Tensor)
20+
- aten::adaptive_max_pool1d(Tensor self, int[2] output_size) -> (Tensor, Tensor)
1921
- aten::adaptive_max_pool2d(Tensor self, int[2] output_size) -> (Tensor, Tensor)
22+
- aten::adaptive_max_pool3d(Tensor self, int[3] output_size) -> (Tensor, Tensor)
2023
- aten::add.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> (Tensor)
2124
- aten::add.Tensor(Tensor self, Tensor other, Scalar alpha=1) -> (Tensor)
2225
- aten::add_.Tensor(Tensor(a!) self, Tensor other, *, Scalar alpha=1) -> (Tensor(a!))
@@ -40,6 +43,7 @@ Operators Currently Supported Through Converters
4043
- aten::cumsum(Tensor self, int dim, *, int? dtype=None) -> (Tensor)
4144
- aten::div.Scalar(Tensor self, Scalar other) -> (Tensor)
4245
- aten::div.Tensor(Tensor self, Tensor other) -> (Tensor)
46+
- aten::div.Tensor_mode(Tensor self, Tensor other, *, str? rounding_mode) -> (Tensor)
4347
- aten::div_.Scalar(Tensor(a!) self, Scalar other) -> (Tensor(a!))
4448
- aten::div_.Tensor(Tensor(a!) self, Tensor other) -> (Tensor(a!))
4549
- aten::elu(Tensor self, Scalar alpha=1, Scalar scale=1, Scalar input_scale=1) -> (Tensor)
@@ -58,12 +62,12 @@ Operators Currently Supported Through Converters
5862
- aten::floor_divide.Scalar(Tensor self, Scalar other) -> (Tensor)
5963
- aten::ge.Scalar(Tensor self, Scalar other) -> (Tensor)
6064
- aten::ge.Tensor(Tensor self, Tensor other) -> (Tensor)
61-
- aten::gelu(Tensor self) -> (Tensor)
6265
- aten::gru_cell(Tensor input, Tensor hx, Tensor w_ih, Tensor w_hh, Tensor? b_ih=None, Tensor? b_hh=None) -> (Tensor)
6366
- aten::gt.Scalar(Tensor self, Scalar other) -> (Tensor)
6467
- aten::gt.Tensor(Tensor self, Tensor other) -> (Tensor)
6568
- aten::hardtanh(Tensor self, Scalar min_val=-1, Scalar max_val=1) -> (Tensor)
6669
- aten::hardtanh_(Tensor(a!) self, Scalar min_val=-1, Scalar max_val=1) -> (Tensor(a!))
70+
- aten::index.Tensor(Tensor self, Tensor?[] indices) -> (Tensor)
6771
- aten::instance_norm(Tensor input, Tensor? weight, Tensor? bias, Tensor? running_mean, Tensor? running_var, bool use_input_stats, float momentum, float eps, bool cudnn_enabled) -> (Tensor)
6872
- aten::layer_norm(Tensor input, int[] normalized_shape, Tensor? gamma, Tensor? beta, float eps, bool cudnn_enabled) -> (Tensor)
6973
- aten::le.Scalar(Tensor self, Scalar other) -> (Tensor)
@@ -78,6 +82,7 @@ Operators Currently Supported Through Converters
7882
- aten::masked_fill.Scalar(Tensor self, Tensor mask, Scalar value) -> (Tensor)
7983
- aten::matmul(Tensor self, Tensor other) -> (Tensor)
8084
- aten::max(Tensor self) -> (Tensor)
85+
- aten::max.dim(Tensor self, int dim, bool keepdim=False) -> (Tensor values, Tensor indices)
8186
- aten::max.other(Tensor self, Tensor other) -> (Tensor)
8287
- aten::max_pool1d(Tensor self, int[1] kernel_size, int[1] stride=[], int[1] padding=[], int[1] dilation=[], bool ceil_mode=False) -> (Tensor)
8388
- aten::max_pool2d(Tensor self, int[2] kernel_size, int[2] stride=[], int[2] padding=[0, 0], int[2] dilation=[1, 1], bool ceil_mode=False) -> (Tensor)
@@ -103,13 +108,16 @@ Operators Currently Supported Through Converters
103108
- aten::prod(Tensor self, *, int? dtype=None) -> (Tensor)
104109
- aten::prod.dim_int(Tensor self, int dim, bool keepdim=False, *, int? dtype=None) -> (Tensor)
105110
- aten::reciprocal(Tensor self) -> (Tensor)
111+
- aten::reflection_pad1d(Tensor self, int[2] padding) -> (Tensor)
112+
- aten::reflection_pad2d(Tensor self, int[4] padding) -> (Tensor)
106113
- aten::relu(Tensor input) -> (Tensor)
107114
- aten::relu_(Tensor(a!) self) -> (Tensor(a!))
108115
- aten::repeat(Tensor self, int[] repeats) -> (Tensor)
109116
- aten::replication_pad1d(Tensor self, int[2] padding) -> (Tensor)
110117
- aten::replication_pad2d(Tensor self, int[4] padding) -> (Tensor)
111118
- aten::replication_pad3d(Tensor self, int[6] padding) -> (Tensor)
112119
- aten::reshape(Tensor self, int[] shape) -> (Tensor)
120+
- aten::roll(Tensor self, int[1] shifts, int[1] dims=[]) -> (Tensor)
113121
- aten::rsub.Scalar(Tensor self, Scalar other, Scalar alpha=1) -> (Tensor)
114122
- aten::rsub.Tensor(Tensor self, Tensor other, Scalar alpha=1) -> (Tensor)
115123
- aten::select.int(Tensor(a) self, int dim, int index) -> (Tensor(a))
@@ -134,11 +142,13 @@ Operators Currently Supported Through Converters
134142
- aten::tan(Tensor self) -> (Tensor)
135143
- aten::tanh(Tensor input) -> (Tensor)
136144
- aten::tanh_(Tensor(a!) self) -> (Tensor(a!))
145+
- aten::to.device(Tensor(a) self, Device device, int dtype, bool non_blocking=False, bool copy=False, int? memory_format=None) -> (Tensor(a))
137146
- aten::to.dtype(Tensor self, int dtype, bool non_blocking=False, bool copy=False, int? memory_format=None) -> (Tensor)
138147
- aten::to.other(Tensor self, Tensor other, bool non_blocking=False, bool copy=False, int? memory_format=None) -> (Tensor)
139148
- aten::to.prim_Device(Tensor(a) self, Device? device, int? dtype=None, bool non_blocking=False, bool copy=False) -> (Tensor(a|b))
140149
- aten::topk(Tensor self, int k, int dim=-1, bool largest=True, bool sorted=True) -> (Tensor values, Tensor indices)
141150
- aten::transpose.int(Tensor(a) self, int dim0, int dim1) -> (Tensor(a))
151+
- aten::unbind.int(Tensor(a -> *) self, int dim=0) -> (Tensor[])
142152
- aten::unsqueeze(Tensor(a) self, int dim) -> (Tensor(a))
143153
- aten::upsample_bilinear2d(Tensor self, int[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> (Tensor)
144154
- aten::upsample_bilinear2d.vec(Tensor input, int[]? output_size, bool align_corners, float[]? scale_factors) -> (Tensor)
@@ -174,10 +184,12 @@ Operators Currently Supported Through Evaluators
174184
- aten::__isnot__(t1 self, t2 obj) -> bool
175185
- aten::__not__(bool self) -> bool
176186
- aten::__or__(int a, int b) -> (bool)
187+
- aten::__range_length(int lo, int hi, int step) -> int
177188
- aten::__round_to_zero_floordiv(int a, int b) -> (int)
178189
- aten::__xor__(int a, int b) -> (bool)
179190
- aten::add.float(float a, float b) -> (float)
180191
- aten::add.int(int a, int b) -> (int)
192+
- aten::add.str(str a, str b) -> (str)
181193
- aten::add_.t(t[](a!) self, t[] b) -> (t[])
182194
- aten::append.t(t[](a!) self, t(c -> *) el) -> (t[](a!))
183195
- aten::arange(Scalar end, *, int? dtype=None, int? layout=None,
@@ -197,10 +209,12 @@ Operators Currently Supported Through Evaluators
197209
- aten::eq.int(int a, int b) -> (bool)
198210
- aten::eq.int_float(int a, float b) -> (bool)
199211
- aten::eq.str(str a, str b) -> (bool)
212+
- aten::extend.t(t[](a!) self, t[] other) -> ()
200213
- aten::floor.float(float a) -> (int)
201214
- aten::floor.int(int a) -> (int)
202215
- aten::floordiv.float(float a, float b) -> (int)
203216
- aten::floordiv.int(int a, int b) -> (int)
217+
- aten::format(str self, ...) -> (str)
204218
- aten::ge.bool(bool a, bool b) -> (bool)
205219
- aten::ge.float(float a, float b) -> (bool)
206220
- aten::ge.float_int(float a, int b) -> (bool)
@@ -232,6 +246,10 @@ Operators Currently Supported Through Evaluators
232246
- aten::ne.int_float(int a, float b) -> (bool)
233247
- aten::neg.int(int a) -> (int)
234248
- aten::numel(Tensor self) -> int
249+
- aten::pow.float(float a, float b) -> (float)
250+
- aten::pow.float_int(float a, int b) -> (float)
251+
- aten::pow.int(int a, int b) -> (float)
252+
- aten::pow.int_float(int a, float b) -> (float)
235253
- aten::size(Tensor self) -> (int[])
236254
- aten::size.int(Tensor self, int dim) -> (int)
237255
- aten::slice.t(t[] l, int start, int end=9223372036854775807, int step=1) -> (t[])

0 commit comments

Comments
 (0)