PyTorch OperatorsĀ¶

OverviewĀ¶

This document provides an overview of PyTorch-supported operators for the IntelĀ® GaudiĀ® AI accelerator. Note that the operators listed below support only selected variants and limited optional parameters for Gaudi.

For details on Fused Ops, see Fused Optimizers and Custom Ops for Intel Gaudi.

PyTorch Operators Support SummaryĀ¶

PyTorch Operator

FP32

BF16

FP16

FP8

Operator Type

adaptive_avg_pool1d

Yes

Yes

Yes

No

torch.nn.functional

adaptive_avg_pool2d

Yes

Yes

Yes

No

torch.nn.functional

adaptive_avg_pool3d

Yes

Yes

Yes

No

torch.nn.functional

avg_pool1d

Yes

Yes

Yes

No

torch.nn.functional

avg_pool2d

Yes

Yes

Yes

No

torch.nn.functional

avg_pool3d

Yes

Yes

Yes

No

torch.nn.functional

batch_norm

Yes

Yes

Yes

No

torch.nn.functional

binary_cross_entropy

Yes

Yes

Yes

No

torch.nn.functional

binary_cross_entropy_with_logits

Yes

Yes

Yes

No

torch.nn.functional

channel_shuffle

Yes

Yes

Yes

No

torch.nn.functional

conv1d

Yes

Yes

Yes

Yes

torch.nn.functional

conv2d

Yes

Yes

Yes

Yes

torch.nn.functional

conv3d

Yes

Yes

Yes

Yes

torch.nn.functional

conv_transpose1d

Yes

Yes

Yes

Yes

torch.nn.functional

conv_transpose2d

Yes

Yes

Yes

Yes

torch.nn.functional

conv_transpose3d

Yes

Yes

Yes

Yes

torch.nn.functional

dropout

Yes

Yes

Yes

No

torch.nn.functional

elu

Yes

Yes

Yes

No

torch.nn.functional

elu_

Yes

Yes

Yes

No

torch.nn.functional

embedding

Yes

Yes

Yes

Yes

torch.nn.functional

embedding_bag

Yes

Yes

Yes

No

torch.nn.functional

gelu

Yes

Yes

Yes

No

torch.nn.functional

gelu_

Yes

Yes

Yes

No

torch.nn.functional

glu

Yes

Yes

Yes

No

torch.nn.functional

grid_sample

Yes

Yes

Yes

No

torch.nn.functional

hardshrink

Yes

Yes

Yes

No

torch.nn.functional

hardsigmoid

Yes

Yes

Yes

No

torch.nn.functional

hardsigmoid_

Yes

Yes

Yes

No

torch.nn.functional

hardtanh

Yes

Yes

Yes

Yes

torch.nn.functional

hardtanh_

Yes

Yes

Yes

Yes

torch.nn.functional

huber_loss

Yes

Yes

Yes

No

torch.nn.functional

instance_norm

Yes

Yes

No

No

torch.nn.functional

kl_div

Yes

Yes

No

No

torch.nn.functional

l1_loss

Yes

Yes

No

No

torch.nn.functional

layer_norm

Yes

Yes

Yes

No

torch.nn.functional

leaky_relu

Yes

Yes

Yes

No

torch.nn.functional

leaky_relu_

Yes

Yes

Yes

No

torch.nn.functional

linear

Yes

Yes

Yes

No

torch.nn.functional

log_softmax

Yes

Yes

Yes

No

torch.nn.functional

logsigmoid

Yes

Yes

Yes

No

torch.nn.functional

max_pool2d

Yes

Yes

Yes

No

torch.nn.functional

max_pool2d_with_indices

Yes

Yes

Yes

No

torch.nn.functional

max_pool3d

Yes

Yes

Yes

No

torch.nn.functional

max_pool3d_with_indices

Yes

Yes

Yes

No

torch.nn.functional

mish

Yes

Yes

Yes

No

torch.nn.functional

mish_

Yes

Yes

Yes

No

torch.nn.functional

mse_loss

Yes

Yes

Yes

No

torch.nn.functional

multi_margin_loss

Yes

Yes

Yes

No

torch.nn.functional

multilabel_margin_loss

Yes

Yes

Yes

No

torch.nn.functional

nll_loss

Yes

Yes

Yes

No

torch.nn.functional

one_hot

No

No

No

No

torch.nn.functional

pad

Yes

Yes

No

Yes

torch.nn.functional

pixel_shuffle

Yes

Yes

Yes

No

torch.nn.functional

prelu

Yes

Yes

No

No

torch.nn.functional

relu

Yes

Yes

Yes

No

torch.nn.functional

relu_

Yes

Yes

Yes

No

torch.nn.functional

rrelu

Yes

Yes

Yes

No

torch.nn.functional

rrelu_

Yes

Yes

Yes

No

torch.nn.functional

sigmoid

Yes

Yes

Yes

No

torch.nn.functional

sigmoid_

Yes

Yes

Yes

No

torch.nn.functional

silu

Yes

Yes

No

No

torch.nn.functional

silu_

Yes

Yes

No

No

torch.nn.functional

smooth_l1_loss

Yes

Yes

Yes

No

torch.nn.functional

softmax

Yes

Yes

Yes

No

torch.nn.functional

softplus

Yes

Yes

Yes

No

torch.nn.functional

softshrink

Yes

Yes

Yes

No

torch.nn.functional

tanh

Yes

Yes

Yes

No

torch.nn.functional

threshold

Yes

Yes

Yes

No

torch.nn.functional

threshold_

Yes

Yes

Yes

No

torch.nn.functional

upsample

Yes

Yes

Yes

No

torch.nn.functional

upsample_bilinear

Yes

Yes

Yes

No

torch.nn.functional

upsample_nearest

Yes

Yes

Yes

No

torch.nn.functional

cross

Yes

Yes

Yes

Yes

torch.linalg

vector_norm

Yes

Yes

Yes

No

torch.linalg

_adaptive_avg_pool2d

Yes

Yes

Yes

No

torch

_adaptive_avg_pool3d

Yes

Yes

Yes

No

torch

_add_relu

Yes

Yes

Yes

No

torch

_add_relu_

Yes

Yes

Yes

No

torch

_addmm_activation

Yes

Yes

Yes

No

torch

_aminmax

Yes

Yes

Yes

Yes

torch

_ctc_loss

Yes

No

No

No

torch

_efficientzerotensor

Yes

Yes

Yes

Yes

torch

_foreach_abs

Yes

Yes

Yes

No

torch

_foreach_abs_

Yes

Yes

Yes

No

torch

_foreach_acos

Yes

Yes

Yes

No

torch

_foreach_acos_

Yes

Yes

Yes

No

torch

_foreach_add

Yes

Yes

Yes

Yes

torch

_foreach_add_

Yes

Yes

Yes

Yes

torch

_foreach_addcdiv

Yes

Yes

Yes

No

torch

_foreach_addcdiv_

Yes

Yes

Yes

No

torch

_foreach_addcmul

Yes

Yes

Yes

No

torch

_foreach_addcmul_

Yes

Yes

Yes

No

torch

_foreach_asin

Yes

Yes

Yes

No

torch

_foreach_asin_

Yes

Yes

Yes

No

torch

_foreach_atan

Yes

Yes

Yes

No

torch

_foreach_atan_

Yes

Yes

Yes

No

torch

_foreach_ceil

Yes

Yes

Yes

No

torch

_foreach_ceil_

Yes

Yes

Yes

No

torch

_foreach_clamp_max

Yes

Yes

Yes

Yes

torch

_foreach_clamp_max_

Yes

Yes

Yes

Yes

torch

_foreach_clamp_min

Yes

Yes

Yes

Yes

torch

_foreach_clamp_min_

Yes

Yes

Yes

Yes

torch

_foreach_cos

Yes

Yes

Yes

No

torch

_foreach_cos_

Yes

Yes

Yes

No

torch

_foreach_cosh

Yes

Yes

Yes

No

torch

_foreach_cosh_

Yes

Yes

Yes

No

torch

_foreach_div

Yes

Yes

Yes

No

torch

_foreach_div_

Yes

Yes

Yes

No

torch

_foreach_erf

Yes

Yes

Yes

No

torch

_foreach_erf_

Yes

Yes

Yes

No

torch

_foreach_erfc

Yes

Yes

Yes

Yes

torch

_foreach_erfc_

Yes

Yes

Yes

Yes

torch

_foreach_exp

Yes

Yes

Yes

No

torch

_foreach_exp_

Yes

Yes

Yes

No

torch

_foreach_expm1

Yes

Yes

Yes

No

torch

_foreach_expm1_

Yes

Yes

Yes

No

torch

_foreach_floor

Yes

Yes

Yes

No

torch

_foreach_floor_

Yes

Yes

Yes

No

torch

_foreach_frac

Yes

Yes

Yes

No

torch

_foreach_frac_

Yes

Yes

Yes

No

torch

_foreach_lerp

Yes

Yes

Yes

Yes

torch

_foreach_lerp_

Yes

Yes

Yes

Yes

torch

_foreach_lgamma

Yes

Yes

Yes

No

torch

_foreach_lgamma_

Yes

Yes

Yes

No

torch

_foreach_log

Yes

Yes

Yes

No

torch

_foreach_log10

Yes

Yes

Yes

No

torch

_foreach_log10_

Yes

Yes

Yes

No

torch

_foreach_log1p

Yes

Yes

Yes

No

torch

_foreach_log1p_

Yes

Yes

Yes

No

torch

_foreach_log2

Yes

Yes

Yes

No

torch

_foreach_log2_

Yes

Yes

Yes

No

torch

_foreach_log_

Yes

Yes

Yes

No

torch

_foreach_maximum

Yes

Yes

Yes

Yes

torch

_foreach_maximum_

Yes

Yes

Yes

Yes

torch

_foreach_minimum

Yes

Yes

Yes

Yes

torch

_foreach_minimum_

Yes

Yes

Yes

Yes

torch

_foreach_mul

Yes

Yes

Yes

Yes

torch

_foreach_mul_

Yes

Yes

Yes

Yes

torch

_foreach_neg

Yes

Yes

Yes

No

torch

_foreach_neg_

Yes

Yes

Yes

No

torch

_foreach_pow

Yes

Yes

Yes

No

torch

_foreach_pow_

Yes

Yes

Yes

No

torch

_foreach_reciprocal

Yes

Yes

Yes

Yes

torch

_foreach_reciprocal_

Yes

Yes

Yes

Yes

torch

_foreach_round

Yes

Yes

Yes

No

torch

_foreach_round_

Yes

Yes

Yes

No

torch

_foreach_sigmoid

Yes

Yes

Yes

No

torch

_foreach_sigmoid_

Yes

Yes

Yes

No

torch

_foreach_sign

Yes

Yes

Yes

No

torch

_foreach_sign_

Yes

Yes

Yes

No

torch

_foreach_sin

Yes

Yes

Yes

No

torch

_foreach_sin_

Yes

Yes

Yes

No

torch

_foreach_sinh

Yes

Yes

Yes

No

torch

_foreach_sinh_

Yes

Yes

Yes

No

torch

_foreach_sqrt

Yes

Yes

Yes

No

torch

_foreach_sqrt_

Yes

Yes

Yes

No

torch

_foreach_sub

Yes

Yes

Yes

Yes

torch

_foreach_sub_

Yes

Yes

Yes

Yes

torch

_foreach_tan

Yes

Yes

Yes

No

torch

_foreach_tan_

Yes

Yes

Yes

No

torch

_foreach_tanh

Yes

Yes

Yes

No

torch

_foreach_tanh_

Yes

Yes

Yes

No

torch

_foreach_trunc

Yes

Yes

Yes

No

torch

_foreach_trunc_

Yes

Yes

Yes

No

torch

_foreach_zero_

Yes

Yes

Yes

Yes

torch

_fused_dropout

Yes

Yes

Yes

No

torch

_logcumsumexp

Yes

Yes

Yes

No

torch

_masked_scale

Yes

Yes

Yes

Yes

torch

_native_batch_norm_legit

Yes

Yes

Yes

No

torch

_native_batch_norm_legit_no_training

Yes

Yes

Yes

No

torch

_resize_output_

Yes

Yes

Yes

Yes

torch

_unique

Yes

No

No

No

torch

_unique2

Yes

No

No

No

torch

_weight_norm_interface

Yes

Yes

Yes

No

torch

abs

Yes

Yes

Yes

No

torch

abs_

Yes

Yes

Yes

No

torch

absolute

Yes

Yes

Yes

No

torch

acos

Yes

Yes

No

No

torch

acos_

Yes

Yes

No

No

torch

acosh

Yes

Yes

No

No

torch

acosh_

Yes

Yes

No

No

torch

add

Yes

Yes

Yes

Yes

torch

addbmm

Yes

Yes

Yes

Yes

torch

addcdiv

Yes

Yes

Yes

No

torch

addcmul

Yes

Yes

Yes

No

torch

addmm

Yes

Yes

Yes

Yes

torch

addmv

Yes

Yes

Yes

No

torch

addmv_

Yes

Yes

Yes

No

torch

addr

Yes

Yes

Yes

No

torch

all

Yes

Yes

Yes

No

torch

amax

Yes

Yes

Yes

Yes

torch

amin

Yes

Yes

Yes

Yes

torch

aminmax

Yes

Yes

Yes

Yes

torch

any

Yes

Yes

Yes

Yes

torch

arange

Yes

Yes

Yes

No

torch

arccos

Yes

Yes

No

No

torch

arccos_

Yes

Yes

No

No

torch

arccosh

Yes

Yes

No

No

torch

arccosh_

Yes

Yes

No

No

torch

arcsin

Yes

Yes

No

No

torch

arcsin_

Yes

Yes

No

No

torch

arcsinh

Yes

Yes

No

No

torch

arcsinh_

Yes

Yes

No

No

torch

arctan

Yes

Yes

No

No

torch

arctan2

Yes

Yes

Yes

No

torch

arctan_

Yes

Yes

No

No

torch

arctanh

Yes

Yes

No

No

torch

arctanh_

Yes

Yes

No

No

torch

argmax

Yes

Yes

Yes

No

torch

argmin

Yes

Yes

Yes

No

torch

argsort

Yes

Yes

Yes

No

torch

as_strided

Yes

Yes

Yes

No

torch

as_strided_

Yes

Yes

Yes

No

torch

asin

Yes

Yes

No

No

torch

asin_

Yes

Yes

No

No

torch

asinh

Yes

Yes

No

No

torch

asinh_

Yes

Yes

No

No

torch

atan

Yes

Yes

No

No

torch

atan2

Yes

Yes

Yes

No

torch

atan_

Yes

Yes

No

No

torch

atanh

Yes

Yes

No

No

torch

atanh_

Yes

Yes

No

No

torch

baddbmm

Yes

Yes

Yes

Yes

torch

batch_norm

Yes

Yes

Yes

No

torch

bernoulli

Yes

Yes

Yes

No

torch

binary_cross_entropy_with_logits

Yes

Yes

Yes

No

torch

bincount

No

No

No

No

torch

bitwise_and

No

No

No

No

torch

bitwise_left_shift

No

No

No

No

torch

bitwise_not

No

No

No

No

torch

bitwise_or

No

No

No

No

torch

bitwise_right_shift

No

No

No

No

torch

bitwise_xor

No

No

No

No

torch

bmm

Yes

Yes

Yes

Yes

torch

broadcast_tensors

Yes

Yes

No

No

torch

bucketize

Yes

Yes

Yes

No

torch

cat

Yes

Yes

Yes

Yes

torch

ceil

Yes

Yes

Yes

No

torch

ceil_

Yes

Yes

Yes

No

torch

channel_shuffle

Yes

Yes

Yes

No

torch

cholesky

Yes

No

No

No

torch

chunk

Yes

Yes

No

No

torch

clamp

Yes

Yes

Yes

Yes

torch

clamp_

Yes

Yes

Yes

Yes

torch

clamp_max

Yes

Yes

Yes

Yes

torch

clamp_max_

Yes

Yes

Yes

Yes

torch

clamp_min

Yes

Yes

Yes

Yes

torch

clamp_min_

Yes

Yes

Yes

Yes

torch

clip

Yes

Yes

No

No

torch

clip_

Yes

Yes

No

No

torch

clone

Yes

Yes

Yes

Yes

torch

conj

Yes

Yes

No

No

torch

constant_pad_nd

Yes

Yes

Yes

Yes

torch

conv1d

Yes

Yes

Yes

Yes

torch

conv2d

Yes

Yes

Yes

Yes

torch

conv3d

Yes

Yes

Yes

Yes

torch

conv_transpose1d

Yes

Yes

Yes

Yes

torch

conv_transpose2d

Yes

Yes

Yes

Yes

torch

conv_transpose3d

Yes

Yes

Yes

Yes

torch

copysign

Yes

Yes

Yes

No

torch

cos

Yes

Yes

Yes

No

torch

cos_

Yes

Yes

Yes

No

torch

cosh

Yes

Yes

Yes

No

torch

cosh_

Yes

Yes

Yes

No

torch

count_nonzero

Yes

Yes

Yes

No

torch

cross

Yes

Yes

Yes

Yes

torch

cumprod

Yes

Yes

Yes

No

torch

cumsum

Yes

Yes

Yes

No

torch

diag

Yes

Yes

Yes

No

torch

div

Yes

Yes

Yes

No

torch

divide

Yes

Yes

Yes

No

torch

dot

Yes

Yes

Yes

Yes

torch

dropout

Yes

Yes

Yes

No

torch

embedding

Yes

Yes

Yes

Yes

torch

embedding_bag

Yes

Yes

Yes

No

torch

embedding_renorm_

Yes

Yes

Yes

No

torch

empty

Yes

Yes

Yes

Yes

torch

empty_like

Yes

Yes

Yes

Yes

torch

empty_strided

Yes

Yes

Yes

Yes

torch

eq

Yes

Yes

Yes

No

torch

equal

Yes

Yes

Yes

No

torch

erf

Yes

Yes

Yes

No

torch

erf_

Yes

Yes

Yes

No

torch

erfc

Yes

Yes

No

No

torch

erfc_

Yes

Yes

No

No

torch

erfinv

Yes

No

No

No

torch

exp

Yes

Yes

Yes

No

torch

exp2

Yes

Yes

Yes

No

torch

exp2_

Yes

Yes

Yes

No

torch

exp_

Yes

Yes

Yes

No

torch

expm1

Yes

Yes

No

No

torch

expm1_

Yes

Yes

No

No

torch

eye

Yes

Yes

Yes

No

torch

fill

Yes

Yes

Yes

Yes

torch

fill_

Yes

Yes

Yes

Yes

torch

flatten

Yes

Yes

No

No

torch

flip

Yes

Yes

Yes

Yes

torch

floor

Yes

Yes

Yes

No

torch

floor_

Yes

Yes

Yes

No

torch

floor_divide

Yes

Yes

Yes

No

torch

fmax

Yes

Yes

Yes

Yes

torch

fmin

Yes

Yes

Yes

Yes

torch

fmod

Yes

Yes

Yes

No

torch

frac

Yes

Yes

Yes

No

torch

frac_

Yes

Yes

Yes

No

torch

frexp

Yes

Yes

Yes

No

torch

full

Yes

Yes

Yes

Yes

torch

full_like

Yes

Yes

Yes

Yes

torch

gather

Yes

Yes

Yes

No

torch

ge

Yes

Yes

Yes

Yes

torch

greater

Yes

Yes

Yes

Yes

torch

greater_equal

Yes

Yes

Yes

Yes

torch

grid_sampler_2d

Yes

Yes

Yes

No

torch

grid_sampler_3d

Yes

Yes

Yes

No

torch

gt

Yes

Yes

Yes

Yes

torch

hardshrink

Yes

Yes

Yes

No

torch

heaviside

Yes

Yes

Yes

No

torch

hypot

Yes

Yes

Yes

No

torch

index_copy

Yes

Yes

Yes

Yes

torch

index_fill

Yes

Yes

Yes

Yes

torch

index_put

Yes

Yes

No

No

torch

index_put_

Yes

Yes

No

No

torch

index_reduce

Yes

Yes

No

No

torch

index_select

Yes

Yes

Yes

Yes

torch

instance_norm

Yes

Yes

No

No

torch

is_complex

Yes

Yes

No

No

torch

is_floating_point

Yes

Yes

No

No

torch

is_nonzero

Yes

Yes

No

No

torch

isfinite

Yes

Yes

Yes

No

torch

isin

Yes

Yes

Yes

No

torch

isinf

Yes

Yes

Yes

No

torch

isnan

Yes

Yes

Yes

No

torch

isneginf

Yes

Yes

Yes

No

torch

isposinf

Yes

Yes

Yes

No

torch

kl_div

Yes

Yes

No

No

torch

kthvalue

Yes

Yes

Yes

No

torch

layer_norm

Yes

Yes

Yes

No

torch

le

Yes

Yes

Yes

Yes

torch

lerp

Yes

Yes

Yes

Yes

torch

less

Yes

Yes

Yes

Yes

torch

less_equal

Yes

Yes

Yes

Yes

torch

lgamma

Yes

Yes

Yes

No

torch

linspace

Yes

Yes

Yes

Yes

torch

log

Yes

Yes

Yes

No

torch

log10

Yes

Yes

Yes

No

torch

log10_

Yes

Yes

Yes

No

torch

log1p

Yes

Yes

Yes

No

torch

log1p_

Yes

Yes

Yes

No

torch

log2

Yes

Yes

Yes

No

torch

log2_

Yes

Yes

Yes

No

torch

log_

Yes

Yes

Yes

No

torch

log_softmax

Yes

Yes

Yes

No

torch

logaddexp

Yes

Yes

No

No

torch

logaddexp2

Yes

Yes

Yes

No

torch

logcumsumexp

Yes

Yes

Yes

No

torch

logical_and

Yes

Yes

Yes

No

torch

logical_not

Yes

Yes

Yes

No

torch

logical_or

Yes

Yes

Yes

No

torch

logical_xor

Yes

Yes

Yes

No

torch

logit

Yes

Yes

Yes

No

torch

logit_

Yes

Yes

Yes

No

torch

logspace

Yes

Yes

Yes

Yes

torch

logsumexp

Yes

Yes

No

No

torch

lt

Yes

Yes

Yes

Yes

torch

masked_fill

Yes

Yes

Yes

Yes

torch

masked_scatter

Yes

Yes

Yes

No

torch

masked_select

Yes

Yes

No

No

torch

matmul

Yes

Yes

No

No

torch

max

Yes

Yes

Yes

Yes

torch

max_pool2d

Yes

Yes

Yes

No

torch

max_pool3d

Yes

Yes

Yes

No

torch

maximum

Yes

Yes

Yes

Yes

torch

mean

Yes

Yes

Yes

No

torch

median

Yes

Yes

Yes

No

torch

meshgrid

Yes

Yes

No

No

torch

min

Yes

Yes

Yes

Yes

torch

minimum

Yes

Yes

Yes

Yes

torch

mm

Yes

Yes

Yes

Yes

torch

mul

Yes

Yes

Yes

Yes

torch

multinomial

Yes

Yes

Yes

No

torch

mv

Yes

Yes

Yes

Yes

torch

nan_to_num

Yes

Yes

Yes

No

torch

nan_to_num_

Yes

Yes

Yes

No

torch

nansum

Yes

Yes

Yes

No

torch

narrow

Yes

No

No

No

torch

native_dropout

Yes

Yes

Yes

No

torch

native_group_norm

Yes

Yes

Yes

No

torch

native_layer_norm

Yes

Yes

Yes

No

torch

ne

Yes

Yes

Yes

Yes

torch

neg

Yes

Yes

Yes

No

torch

neg_

Yes

Yes

Yes

No

torch

nextafter

Yes

Yes

Yes

No

torch

nonzero

Yes

Yes

No

No

torch

norm

Yes

Yes

Yes

No

torch

normal

Yes

Yes

Yes

No

torch

not_equal

Yes

Yes

Yes

Yes

torch

ones

Yes

Yes

No

Yes

torch

ones_like

Yes

Yes

No

Yes

torch

permute

Yes

Yes

Yes

Yes

torch

pixel_shuffle

Yes

Yes

Yes

No

torch

poisson

Yes

Yes

Yes

No

torch

pow

Yes

Yes

Yes

No

torch

prelu

Yes

Yes

No

No

torch

prod

Yes

Yes

Yes

No

torch

put

Yes

Yes

Yes

No

torch

randperm

Yes

Yes

Yes

Yes

torch

reciprocal

Yes

Yes

Yes

Yes

torch

reciprocal_

Yes

Yes

Yes

Yes

torch

relu

Yes

Yes

Yes

No

torch

relu_

Yes

Yes

Yes

No

torch

remainder

Yes

Yes

Yes

No

torch

reshape

Yes

Yes

Yes

Yes

torch

resolve_conj

Yes

Yes

No

No

torch

resolve_neg

Yes

Yes

No

No

torch

result_type

Yes

Yes

No

No

torch

roll

Yes

Yes

No

Yes

torch

round

Yes

Yes

Yes

No

torch

round_

Yes

Yes

Yes

No

torch

rsqrt

Yes

Yes

Yes

No

torch

rsqrt_

Yes

Yes

Yes

No

torch

rsub

Yes

Yes

Yes

Yes

torch

scalar_tensor

Yes

Yes

Yes

Yes

torch

scatter

Yes

Yes

Yes

Yes

torch

scatter_add

Yes

Yes

Yes

No

torch

scatter_reduce

Yes

Yes

Yes

No

torch

searchsorted

Yes

Yes

Yes

No

torch

select

Yes

Yes

Yes

Yes

torch

sgn

Yes

Yes

Yes

No

torch

sigmoid

Yes

Yes

Yes

No

torch

sigmoid_

Yes

Yes

Yes

No

torch

sign

Yes

Yes

Yes

No

torch

signbit

Yes

Yes

Yes

No

torch

sin

Yes

Yes

Yes

No

torch

sin_

Yes

Yes

Yes

No

torch

sinc

Yes

Yes

Yes

No

torch

sinc_

Yes

Yes

Yes

No

torch

sinh

Yes

Yes

Yes

No

torch

sinh_

Yes

Yes

Yes

No

torch

softmax

Yes

Yes

Yes

No

torch

sort

Yes

Yes

Yes

No

torch

split_with_sizes

Yes

Yes

No

Yes

torch

sqrt

Yes

Yes

Yes

No

torch

sqrt_

Yes

Yes

Yes

No

torch

square

Yes

Yes

No

No

torch

square_

Yes

Yes

No

No

torch

squeeze

Yes

Yes

Yes

Yes

torch

stack

Yes

Yes

No

No

torch

std

Yes

Yes

Yes

No

torch

std_mean

Yes

Yes

Yes

No

torch

sub

Yes

Yes

Yes

Yes

torch

sum

Yes

Yes

Yes

Yes

torch

t

Yes

Yes

Yes

Yes

torch

take

Yes

Yes

Yes

No

torch

tan

Yes

Yes

No

No

torch

tan_

Yes

Yes

No

No

torch

tanh

Yes

Yes

Yes

No

torch

tanh_

Yes

Yes

Yes

No

torch

threshold

Yes

Yes

Yes

No

torch

threshold_

Yes

Yes

Yes

No

torch

topk

Yes

Yes

Yes

No

torch

trace

Yes

Yes

Yes

No

torch

transpose

Yes

Yes

Yes

Yes

torch

tril

Yes

Yes

Yes

No

torch

tril_indices

No

No

No

No

torch

triu

Yes

Yes

Yes

No

torch

triu_indices

No

No

No

No

torch

trunc

Yes

Yes

Yes

No

torch

trunc_

Yes

Yes

Yes

No

torch

unbind

Yes

Yes

No

No

torch

unique

Yes

No

No

No

torch

unsqueeze

Yes

Yes

Yes

Yes

torch

var

Yes

Yes

Yes

No

torch

var_mean

Yes

Yes

Yes

No

torch

vdot

Yes

Yes

Yes

Yes

torch

where

Yes

Yes

Yes

Yes

torch

xlogy

Yes

Yes

Yes

No

torch

xlogy_

Yes

Yes

Yes

No

torch

zero_

Yes

Yes

Yes

Yes

torch

zeros

Yes

Yes

Yes

Yes

torch

zeros_like

Yes

Yes

Yes

Yes

torch

glu_jvp

Yes

Yes

Yes

No

torch.ops.aten

AdaptiveAvgPool1d

Yes

Yes

Yes

No

torch.nn

AdaptiveAvgPool2d

Yes

Yes

Yes

No

torch.nn

AdaptiveAvgPool3d

Yes

Yes

Yes

No

torch.nn

AvgPool1d

Yes

Yes

Yes

No

torch.nn

AvgPool2d

Yes

Yes

Yes

No

torch.nn

AvgPool3d

Yes

Yes

Yes

No

torch.nn

BCELoss

Yes

Yes

Yes

No

torch.nn

BCEWithLogitsLoss

Yes

Yes

Yes

No

torch.nn

BatchNorm1d

Yes

Yes

Yes

No

torch.nn

BatchNorm2d

Yes

Yes

Yes

No

torch.nn

ChannelShuffle

Yes

Yes

Yes

No

torch.nn

ConstantPad1d

Yes

Yes

Yes

No

torch.nn

Conv1d

Yes

Yes

Yes

Yes

torch.nn

Conv2d

Yes

Yes

Yes

Yes

torch.nn

Conv3d

Yes

Yes

Yes

Yes

torch.nn

ConvTranspose1d

Yes

Yes

Yes

Yes

torch.nn

ConvTranspose2d

Yes

Yes

Yes

Yes

torch.nn

ConvTranspose3d

Yes

Yes

Yes

Yes

torch.nn

CrossEntropyLoss

Yes

Yes

No

No

torch.nn

Dropout

Yes

Yes

Yes

No

torch.nn

ELU

Yes

Yes

Yes

No

torch.nn

EmbeddingBag

Yes

Yes

Yes

No

torch.nn

GELU

Yes

Yes

Yes

No

torch.nn

GLU

Yes

Yes

Yes

No

torch.nn

Hardshrink

Yes

Yes

Yes

No

torch.nn

Hardsigmoid

Yes

Yes

Yes

No

torch.nn

Hardswish

Yes

Yes

No

No

torch.nn

Hardtanh

Yes

Yes

Yes

Yes

torch.nn

HuberLoss

Yes

Yes

Yes

No

torch.nn

InstanceNorm2d

Yes

Yes

No

No

torch.nn

KLDivLoss

Yes

Yes

No

No

torch.nn

LayerNorm

Yes

Yes

Yes

No

torch.nn

LeakyReLU

Yes

Yes

Yes

No

torch.nn

Linear

Yes

Yes

Yes

No

torch.nn

LogSigmoid

Yes

Yes

Yes

No

torch.nn

LogSoftmax

Yes

Yes

Yes

No

torch.nn

MSELoss

Yes

Yes

Yes

No

torch.nn

MaxPool2d

Yes

Yes

Yes

No

torch.nn

MaxPool3d

Yes

Yes

Yes

No

torch.nn

Mish

Yes

Yes

Yes

No

torch.nn

MultiLabelMarginLoss

Yes

Yes

Yes

No

torch.nn

MultiMarginLoss

Yes

Yes

Yes

No

torch.nn

NLLLoss

Yes

Yes

Yes

No

torch.nn

PReLU

Yes

Yes

No

No

torch.nn

PixelShuffle

Yes

Yes

Yes

No

torch.nn

RReLU

Yes

Yes

Yes

No

torch.nn

ReLU

Yes

Yes

Yes

No

torch.nn

ReflectionPad1d

Yes

Yes

Yes

Yes

torch.nn

ReflectionPad2d

Yes

Yes

Yes

Yes

torch.nn

ReflectionPad3d

Yes

Yes

Yes

Yes

torch.nn

ReplicationPad1d

Yes

Yes

Yes

Yes

torch.nn

ReplicationPad2d

Yes

Yes

Yes

Yes

torch.nn

ReplicationPad3d

Yes

Yes

Yes

Yes

torch.nn

SiLU

Yes

Yes

No

No

torch.nn

Sigmoid

Yes

Yes

Yes

No

torch.nn

SmoothL1Loss

Yes

Yes

Yes

No

torch.nn

Softmax

Yes

Yes

Yes

No

torch.nn

Softplus

Yes

Yes

Yes

No

torch.nn

Softshrink

Yes

Yes

Yes

No

torch.nn

Tanh

Yes

Yes

Yes

No

torch.nn

Threshold

Yes

Yes

Yes

No

torch.nn

Upsample

Yes

Yes

Yes

No

torch.nn

UpsamplingNearest2d

Yes

Yes

Yes

No

torch.nn

weight_norm

Yes

Yes

No

No

torch.nn.utils

T

Yes

Yes

Yes

Yes

torch.Tensor

__and__

No

No

No

No

torch.Tensor

__iand__

No

No

No

No

torch.Tensor

__ilshift__

No

No

No

No

torch.Tensor

__ior__

No

No

No

No

torch.Tensor

__irshift__

No

No

No

No

torch.Tensor

__ixor__

No

No

No

No

torch.Tensor

__lshift__

No

No

No

No

torch.Tensor

__or__

No

No

No

No

torch.Tensor

__rshift__

No

No

No

No

torch.Tensor

__xor__

No

No

No

No

torch.Tensor

abs

Yes

Yes

Yes

No

torch.Tensor

abs_

Yes

Yes

Yes

No

torch.Tensor

absolute

Yes

Yes

Yes

No

torch.Tensor

absolute_

Yes

Yes

Yes

No

torch.Tensor

acos

Yes

Yes

No

No

torch.Tensor

acos_

Yes

Yes

No

No

torch.Tensor

acosh

Yes

Yes

No

No

torch.Tensor

acosh_

Yes

Yes

No

No

torch.Tensor

add

Yes

Yes

Yes

Yes

torch.Tensor

add_

Yes

Yes

Yes

Yes

torch.Tensor

addbmm

Yes

Yes

Yes

Yes

torch.Tensor

addbmm_

Yes

Yes

Yes

Yes

torch.Tensor

addcdiv

Yes

Yes

Yes

No

torch.Tensor

addcdiv_

Yes

Yes

Yes

No

torch.Tensor

addcmul

Yes

Yes

Yes

No

torch.Tensor

addcmul_

Yes

Yes

Yes

No

torch.Tensor

addmm

Yes

Yes

Yes

Yes

torch.Tensor

addmm_

Yes

Yes

Yes

Yes

torch.Tensor

addmv

Yes

Yes

Yes

No

torch.Tensor

addmv_

Yes

Yes

Yes

No

torch.Tensor

addr

Yes

Yes

Yes

No

torch.Tensor

addr_

Yes

Yes

Yes

No

torch.Tensor

all

Yes

Yes

Yes

No

torch.Tensor

amax

Yes

Yes

Yes

Yes

torch.Tensor

amin

Yes

Yes

Yes

Yes

torch.Tensor

aminmax

Yes

Yes

Yes

Yes

torch.Tensor

any

Yes

Yes

Yes

Yes

torch.Tensor

arccos

Yes

Yes

No

No

torch.Tensor

arccos_

Yes

Yes

No

No

torch.Tensor

arccosh

Yes

Yes

No

No

torch.Tensor

arccosh_

Yes

Yes

No

No

torch.Tensor

arcsin

Yes

Yes

No

No

torch.Tensor

arcsin_

Yes

Yes

No

No

torch.Tensor

arcsinh

Yes

Yes

No

No

torch.Tensor

arcsinh_

Yes

Yes

No

No

torch.Tensor

arctan

Yes

Yes

No

No

torch.Tensor

arctan2

Yes

Yes

Yes

No

torch.Tensor

arctan2_

Yes

Yes

Yes

No

torch.Tensor

arctan_

Yes

Yes

No

No

torch.Tensor

arctanh

Yes

Yes

No

No

torch.Tensor

arctanh_

Yes

Yes

No

No

torch.Tensor

argmax

Yes

Yes

Yes

No

torch.Tensor

argmin

Yes

Yes

Yes

No

torch.Tensor

argsort

Yes

Yes

Yes

No

torch.Tensor

as_strided

Yes

Yes

Yes

No

torch.Tensor

as_strided_

Yes

Yes

Yes

No

torch.Tensor

asin

Yes

Yes

No

No

torch.Tensor

asin_

Yes

Yes

No

No

torch.Tensor

asinh

Yes

Yes

No

No

torch.Tensor

asinh_

Yes

Yes

No

No

torch.Tensor

atan

Yes

Yes

No

No

torch.Tensor

atan2

Yes

Yes

Yes

No

torch.Tensor

atan2_

Yes

Yes

Yes

No

torch.Tensor

atan_

Yes

Yes

No

No

torch.Tensor

atanh

Yes

Yes

No

No

torch.Tensor

atanh_

Yes

Yes

No

No

torch.Tensor

baddbmm

Yes

Yes

Yes

Yes

torch.Tensor

baddbmm_

Yes

Yes

Yes

Yes

torch.Tensor

bernoulli

Yes

Yes

Yes

No

torch.Tensor

bernoulli_

Yes

Yes

Yes

No

torch.Tensor

bincount

No

No

No

No

torch.Tensor

bitwise_and

No

No

No

No

torch.Tensor

bitwise_and_

No

No

No

No

torch.Tensor

bitwise_left_shift

No

No

No

No

torch.Tensor

bitwise_left_shift_

No

No

No

No

torch.Tensor

bitwise_not

No

No

No

No

torch.Tensor

bitwise_not_

No

No

No

No

torch.Tensor

bitwise_or

No

No

No

No

torch.Tensor

bitwise_or_

No

No

No

No

torch.Tensor

bitwise_right_shift

No

No

No

No

torch.Tensor

bitwise_right_shift_

No

No

No

No

torch.Tensor

bitwise_xor

No

No

No

No

torch.Tensor

bitwise_xor_

No

No

No

No

torch.Tensor

bmm

Yes

Yes

Yes

Yes

torch.Tensor

ceil

Yes

Yes

Yes

No

torch.Tensor

ceil_

Yes

Yes

Yes

No

torch.Tensor

cholesky

Yes

No

No

No

torch.Tensor

chunk

Yes

Yes

No

No

torch.Tensor

clamp

Yes

Yes

Yes

Yes

torch.Tensor

clamp_

Yes

Yes

Yes

Yes

torch.Tensor

clamp_max

Yes

Yes

Yes

Yes

torch.Tensor

clamp_max_

Yes

Yes

Yes

Yes

torch.Tensor

clamp_min

Yes

Yes

Yes

Yes

torch.Tensor

clamp_min_

Yes

Yes

Yes

Yes

torch.Tensor

clip

Yes

Yes

No

No

torch.Tensor

clip_

Yes

Yes

No

No

torch.Tensor

clone

Yes

Yes

Yes

Yes

torch.Tensor

conj

Yes

Yes

No

No

torch.Tensor

copy_

Yes

Yes

No

Yes

torch.Tensor

copysign

Yes

Yes

Yes

No

torch.Tensor

copysign_

Yes

Yes

Yes

No

torch.Tensor

cos

Yes

Yes

Yes

No

torch.Tensor

cos_

Yes

Yes

Yes

No

torch.Tensor

cosh

Yes

Yes

Yes

No

torch.Tensor

cosh_

Yes

Yes

Yes

No

torch.Tensor

count_nonzero

Yes

Yes

Yes

No

torch.Tensor

cross

Yes

Yes

Yes

Yes

torch.Tensor

cumprod

Yes

Yes

Yes

No

torch.Tensor

cumprod_

Yes

Yes

Yes

No

torch.Tensor

cumsum

Yes

Yes

Yes

No

torch.Tensor

cumsum_

Yes

Yes

Yes

No

torch.Tensor

diag

Yes

Yes

Yes

No

torch.Tensor

div

Yes

Yes

Yes

No

torch.Tensor

div_

Yes

Yes

Yes

No

torch.Tensor

divide

Yes

Yes

Yes

No

torch.Tensor

divide_

Yes

Yes

Yes

No

torch.Tensor

dot

Yes

Yes

Yes

Yes

torch.Tensor

eq

Yes

Yes

Yes

No

torch.Tensor

eq_

Yes

Yes

Yes

No

torch.Tensor

equal

Yes

Yes

Yes

No

torch.Tensor

erf

Yes

Yes

Yes

No

torch.Tensor

erf_

Yes

Yes

Yes

No

torch.Tensor

erfc

Yes

Yes

No

No

torch.Tensor

erfc_

Yes

Yes

No

No

torch.Tensor

erfinv

Yes

No

No

No

torch.Tensor

erfinv_

Yes

No

No

No

torch.Tensor

exp

Yes

Yes

Yes

No

torch.Tensor

exp2

Yes

Yes

Yes

No

torch.Tensor

exp2_

Yes

Yes

Yes

No

torch.Tensor

exp_

Yes

Yes

Yes

No

torch.Tensor

expand

Yes

Yes

Yes

No

torch.Tensor

expand_as

Yes

Yes

Yes

No

torch.Tensor

expm1

Yes

Yes

No

No

torch.Tensor

expm1_

Yes

Yes

No

No

torch.Tensor

exponential_

Yes

Yes

Yes

No

torch.Tensor

fill_

Yes

Yes

Yes

Yes

torch.Tensor

flatten

Yes

Yes

No

No

torch.Tensor

flip

Yes

Yes

Yes

Yes

torch.Tensor

floor

Yes

Yes

Yes

No

torch.Tensor

floor_

Yes

Yes

Yes

No

torch.Tensor

floor_divide

Yes

Yes

Yes

No

torch.Tensor

floor_divide_

Yes

Yes

Yes

No

torch.Tensor

fmax

Yes

Yes

Yes

Yes

torch.Tensor

fmin

Yes

Yes

Yes

Yes

torch.Tensor

fmod

Yes

Yes

Yes

No

torch.Tensor

fmod_

Yes

Yes

Yes

No

torch.Tensor

frac

Yes

Yes

Yes

No

torch.Tensor

frac_

Yes

Yes

Yes

No

torch.Tensor

frexp

Yes

Yes

Yes

No

torch.Tensor

gather

Yes

Yes

Yes

No

torch.Tensor

ge

Yes

Yes

Yes

Yes

torch.Tensor

ge_

Yes

Yes

Yes

Yes

torch.Tensor

geometric_

Yes

Yes

Yes

No

torch.Tensor

greater

Yes

Yes

Yes

Yes

torch.Tensor

greater_

Yes

Yes

Yes

Yes

torch.Tensor

greater_equal

Yes

Yes

Yes

Yes

torch.Tensor

greater_equal_

Yes

Yes

Yes

Yes

torch.Tensor

gt

Yes

Yes

Yes

Yes

torch.Tensor

gt_

Yes

Yes

Yes

Yes

torch.Tensor

hardshrink

Yes

Yes

Yes

No

torch.Tensor

heaviside

Yes

Yes

Yes

No

torch.Tensor

heaviside_

Yes

Yes

Yes

No

torch.Tensor

hypot

Yes

Yes

Yes

No

torch.Tensor

hypot_

Yes

Yes

Yes

No

torch.Tensor

index_add_

Yes

Yes

Yes

No

torch.Tensor

index_copy

Yes

Yes

Yes

Yes

torch.Tensor

index_copy_

Yes

Yes

Yes

Yes

torch.Tensor

index_fill

Yes

Yes

Yes

Yes

torch.Tensor

index_fill_

Yes

Yes

Yes

Yes

torch.Tensor

index_put

Yes

Yes

No

No

torch.Tensor

index_put_

Yes

Yes

No

No

torch.Tensor

index_reduce

Yes

Yes

No

No

torch.Tensor

index_reduce_

Yes

Yes

No

No

torch.Tensor

index_select

Yes

Yes

Yes

Yes

torch.Tensor

is_complex

Yes

Yes

No

No

torch.Tensor

is_floating_point

Yes

Yes

No

No

torch.Tensor

is_nonzero

Yes

Yes

No

No

torch.Tensor

isfinite

Yes

Yes

Yes

No

torch.Tensor

isinf

Yes

Yes

Yes

No

torch.Tensor

isnan

Yes

Yes

Yes

No

torch.Tensor

isneginf

Yes

Yes

Yes

No

torch.Tensor

isposinf

Yes

Yes

Yes

No

torch.Tensor

item

Yes

No

No

No

torch.Tensor

kthvalue

Yes

Yes

Yes

No

torch.Tensor

le

Yes

Yes

Yes

Yes

torch.Tensor

le_

Yes

Yes

Yes

Yes

torch.Tensor

lerp

Yes

Yes

Yes

Yes

torch.Tensor

lerp_

Yes

Yes

Yes

Yes

torch.Tensor

less

Yes

Yes

Yes

Yes

torch.Tensor

less_

Yes

Yes

Yes

No

torch.Tensor

less_equal

Yes

Yes

Yes

Yes

torch.Tensor

less_equal_

Yes

Yes

Yes

Yes

torch.Tensor

log

Yes

Yes

Yes

No

torch.Tensor

log10

Yes

Yes

Yes

No

torch.Tensor

log10_

Yes

Yes

Yes

No

torch.Tensor

log1p

Yes

Yes

Yes

No

torch.Tensor

log1p_

Yes

Yes

Yes

No

torch.Tensor

log2

Yes

Yes

Yes

No

torch.Tensor

log2_

Yes

Yes

Yes

No

torch.Tensor

log_

Yes

Yes

Yes

No

torch.Tensor

log_normal_

Yes

Yes

Yes

Yes

torch.Tensor

log_softmax

Yes

Yes

Yes

No

torch.Tensor

logaddexp

Yes

Yes

No

No

torch.Tensor

logaddexp2

Yes

Yes

Yes

No

torch.Tensor

logcumsumexp

Yes

Yes

Yes

No

torch.Tensor

logical_and

Yes

Yes

Yes

No

torch.Tensor

logical_and_

Yes

Yes

Yes

No

torch.Tensor

logical_not

Yes

Yes

Yes

No

torch.Tensor

logical_not_

No

No

Yes

No

torch.Tensor

logical_or

Yes

Yes

Yes

No

torch.Tensor

logical_or_

Yes

Yes

Yes

No

torch.Tensor

logical_xor

Yes

Yes

Yes

No

torch.Tensor

logical_xor_

Yes

Yes

Yes

No

torch.Tensor

logit

Yes

Yes

Yes

No

torch.Tensor

logit_

Yes

Yes

Yes

No

torch.Tensor

logsumexp

Yes

Yes

No

No

torch.Tensor

lt

Yes

Yes

Yes

Yes

torch.Tensor

lt_

Yes

Yes

Yes

No

torch.Tensor

masked_fill

Yes

Yes

Yes

Yes

torch.Tensor

masked_fill_

Yes

Yes

Yes

Yes

torch.Tensor

masked_scatter

Yes

Yes

Yes

No

torch.Tensor

masked_scatter_

Yes

Yes

Yes

No

torch.Tensor

masked_select

Yes

Yes

No

No

torch.Tensor

matmul

Yes

Yes

No

No

torch.Tensor

max

Yes

Yes

Yes

Yes

torch.Tensor

maximum

Yes

Yes

Yes

Yes

torch.Tensor

mean

Yes

Yes

Yes

No

torch.Tensor

median

Yes

Yes

Yes

No

torch.Tensor

min

Yes

Yes

Yes

Yes

torch.Tensor

minimum

Yes

Yes

Yes

Yes

torch.Tensor

mm

Yes

Yes

Yes

Yes

torch.Tensor

mul

Yes

Yes

Yes

Yes

torch.Tensor

mul_

Yes

Yes

Yes

Yes

torch.Tensor

multinomial

Yes

Yes

Yes

No

torch.Tensor

mv

Yes

Yes

Yes

Yes

torch.Tensor

nan_to_num

Yes

Yes

Yes

No

torch.Tensor

nan_to_num_

Yes

Yes

Yes

No

torch.Tensor

nansum

Yes

Yes

Yes

No

torch.Tensor

narrow

Yes

No

No

No

torch.Tensor

ne

Yes

Yes

Yes

Yes

torch.Tensor

ne_

Yes

Yes

Yes

Yes

torch.Tensor

neg

Yes

Yes

Yes

No

torch.Tensor

neg_

Yes

Yes

Yes

No

torch.Tensor

new_empty

Yes

Yes

No

No

torch.Tensor

new_empty_strided

Yes

Yes

No

No

torch.Tensor

new_full

Yes

Yes

No

No

torch.Tensor

new_ones

Yes

Yes

No

No

torch.Tensor

new_zeros

Yes

Yes

Yes

Yes

torch.Tensor

nextafter

Yes

Yes

Yes

No

torch.Tensor

nextafter_

Yes

Yes

Yes

No

torch.Tensor

nonzero

Yes

Yes

No

No

torch.Tensor

norm

Yes

Yes

Yes

No

torch.Tensor

normal_

Yes

Yes

Yes

Yes

torch.Tensor

not_equal

Yes

Yes

Yes

Yes

torch.Tensor

not_equal_

Yes

Yes

Yes

Yes

torch.Tensor

permute

Yes

Yes

Yes

Yes

torch.Tensor

pin_memory

Yes

Yes

No

No

torch.Tensor

pow

Yes

Yes

Yes

No

torch.Tensor

pow_

Yes

Yes

Yes

No

torch.Tensor

prod

Yes

Yes

Yes

No

torch.Tensor

put

Yes

Yes

Yes

No

torch.Tensor

put_

Yes

Yes

Yes

No

torch.Tensor

random_

Yes

Yes

Yes

No

torch.Tensor

reciprocal

Yes

Yes

Yes

Yes

torch.Tensor

reciprocal_

Yes

Yes

Yes

Yes

torch.Tensor

relu

Yes

Yes

Yes

No

torch.Tensor

relu_

Yes

Yes

Yes

No

torch.Tensor

remainder

Yes

Yes

Yes

No

torch.Tensor

remainder_

Yes

Yes

Yes

No

torch.Tensor

repeat

Yes

Yes

Yes

Yes

torch.Tensor

repeat_interleave

Yes

Yes

Yes

No

torch.Tensor

reshape

Yes

Yes

Yes

Yes

torch.Tensor

resize_

Yes

Yes

Yes

Yes

torch.Tensor

resolve_conj

Yes

Yes

No

No

torch.Tensor

resolve_neg

Yes

Yes

No

No

torch.Tensor

roll

Yes

Yes

No

Yes

torch.Tensor

round

Yes

Yes

Yes

No

torch.Tensor

round_

Yes

Yes

Yes

No

torch.Tensor

rsqrt

Yes

Yes

Yes

No

torch.Tensor

rsqrt_

Yes

Yes

Yes

No

torch.Tensor

scatter

Yes

Yes

Yes

Yes

torch.Tensor

scatter_

Yes

Yes

Yes

Yes

torch.Tensor

scatter_add

Yes

Yes

Yes

No

torch.Tensor

scatter_add_

Yes

Yes

Yes

No

torch.Tensor

scatter_reduce

Yes

Yes

Yes

No

torch.Tensor

scatter_reduce_

Yes

Yes

Yes

No

torch.Tensor

select

Yes

Yes

Yes

Yes

torch.Tensor

sgn

Yes

Yes

Yes

No

torch.Tensor

sgn_

Yes

Yes

Yes

No

torch.Tensor

sigmoid

Yes

Yes

Yes

No

torch.Tensor

sigmoid_

Yes

Yes

Yes

No

torch.Tensor

sign

Yes

Yes

Yes

No

torch.Tensor

sign_

Yes

Yes

Yes

No

torch.Tensor

signbit

Yes

Yes

Yes

No

torch.Tensor

sin

Yes

Yes

Yes

No

torch.Tensor

sin_

Yes

Yes

Yes

No

torch.Tensor

sinc

Yes

Yes

Yes

No

torch.Tensor

sinc_

Yes

Yes

Yes

No

torch.Tensor

sinh

Yes

Yes

Yes

No

torch.Tensor

sinh_

Yes

Yes

Yes

No

torch.Tensor

sort

Yes

Yes

Yes

No

torch.Tensor

split_with_sizes

Yes

Yes

No

Yes

torch.Tensor

sqrt

Yes

Yes

Yes

No

torch.Tensor

sqrt_

Yes

Yes

Yes

No

torch.Tensor

square

Yes

Yes

No

No

torch.Tensor

square_

Yes

Yes

No

No

torch.Tensor

squeeze

Yes

Yes

Yes

Yes

torch.Tensor

squeeze_

Yes

Yes

Yes

Yes

torch.Tensor

std

Yes

Yes

Yes

No

torch.Tensor

sub

Yes

Yes

Yes

Yes

torch.Tensor

sub_

Yes

Yes

Yes

Yes

torch.Tensor

sum

Yes

Yes

Yes

Yes

torch.Tensor

t

Yes

Yes

Yes

Yes

torch.Tensor

take

Yes

Yes

Yes

No

torch.Tensor

tan

Yes

Yes

No

No

torch.Tensor

tan_

Yes

Yes

No

No

torch.Tensor

tanh

Yes

Yes

Yes

No

torch.Tensor

tanh_

Yes

Yes

Yes

No

torch.Tensor

to

Yes

Yes

Yes

Yes

torch.Tensor

topk

Yes

Yes

Yes

No

torch.Tensor

trace

Yes

Yes

Yes

No

torch.Tensor

transpose

Yes

Yes

Yes

Yes

torch.Tensor

tril

Yes

Yes

Yes

No

torch.Tensor

tril_

Yes

Yes

Yes

No

torch.Tensor

triu

Yes

Yes

Yes

No

torch.Tensor

triu_

Yes

Yes

Yes

No

torch.Tensor

trunc

Yes

Yes

Yes

No

torch.Tensor

trunc_

Yes

Yes

Yes

No

torch.Tensor

unbind

Yes

Yes

No

No

torch.Tensor

uniform_

Yes

Yes

Yes

No

torch.Tensor

unique

Yes

No

No

No

torch.Tensor

unsqueeze

Yes

Yes

Yes

Yes

torch.Tensor

unsqueeze_

Yes

Yes

Yes

Yes

torch.Tensor

var

Yes

Yes

Yes

No

torch.Tensor

vdot

Yes

Yes

Yes

Yes

torch.Tensor

view

Yes

Yes

Yes

Yes

torch.Tensor

where

Yes

Yes

Yes

Yes

torch.Tensor

xlogy

Yes

Yes

Yes

No

torch.Tensor

xlogy_

Yes

Yes

Yes

No

torch.Tensor

zero_

Yes

Yes

Yes

Yes

torch.Tensor

entr

Yes

Yes

Yes

No

torch.special

erf

Yes

Yes

Yes

No

torch.special

erfc

Yes

Yes

No

No

torch.special

erfcx

Yes

Yes

No

No

torch.special

erfinv

Yes

No

No

No

torch.special

exp2

Yes

Yes

Yes

No

torch.special

expit

Yes

Yes

No

No

torch.special

expm1

Yes

Yes

No

No

torch.special

log1p

Yes

Yes

Yes

No

torch.special

logit

Yes

Yes

Yes

No

torch.special

logsumexp

Yes

Yes

No

No

torch.special

round

Yes

Yes

Yes

No

torch.special

sinc

Yes

Yes

Yes

No

torch.special

softmax

Yes

Yes

Yes

No

torch.special

xlog1py

Yes

Yes

Yes

No

torch.special

xlogy

Yes

Yes

Yes

No

torch.special

batched_nms

Yes

Yes

No

No

torchvision.ops

deform_conv2d

Yes

No

No

No

torchvision.ops

nms

Yes

Yes

No

No

torchvision.ops

roi_align

Yes

No

No

No

torchvision.ops

_safe_softmax

Yes

Yes

Yes

No

torch.ops

PyTorch Operator

INT64

INT32

INT16

INT8

BOOL

Operator Type

adaptive_avg_pool1d

No

No

No

No

No

torch.nn.functional

adaptive_avg_pool2d

No

No

No

No

No

torch.nn.functional

adaptive_avg_pool3d

No

No

No

No

No

torch.nn.functional

avg_pool1d

No

No

No

No

No

torch.nn.functional

avg_pool2d

No

No

No

No

No

torch.nn.functional

avg_pool3d

No

No

No

No

No

torch.nn.functional

batch_norm

No

No

No

No

No

torch.nn.functional

binary_cross_entropy

No

No

No

No

No

torch.nn.functional

binary_cross_entropy_with_logits

No

No

No

No

No

torch.nn.functional

channel_shuffle

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

conv1d

No

No

No

No

No

torch.nn.functional

conv2d

No

No

No

No

No

torch.nn.functional

conv3d

No

No

No

No

No

torch.nn.functional

conv_transpose1d

No

No

No

No

No

torch.nn.functional

conv_transpose2d

No

No

No

No

No

torch.nn.functional

conv_transpose3d

No

No

No

No

No

torch.nn.functional

dropout

No

No

No

No

No

torch.nn.functional

elu

No

No

No

No

No

torch.nn.functional

elu_

No

No

No

No

No

torch.nn.functional

embedding

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

embedding_bag

No

No

No

No

No

torch.nn.functional

gelu

No

No

No

No

No

torch.nn.functional

gelu_

No

No

No

No

No

torch.nn.functional

glu

No

No

No

No

No

torch.nn.functional

grid_sample

No

No

No

No

No

torch.nn.functional

hardshrink

No

No

No

No

No

torch.nn.functional

hardsigmoid

No

No

No

No

No

torch.nn.functional

hardsigmoid_

No

No

No

No

No

torch.nn.functional

hardtanh

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

hardtanh_

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

huber_loss

No

No

No

No

No

torch.nn.functional

instance_norm

No

No

No

No

No

torch.nn.functional

kl_div

No

No

No

No

No

torch.nn.functional

l1_loss

No

No

No

No

No

torch.nn.functional

layer_norm

No

No

No

No

No

torch.nn.functional

leaky_relu

No

No

No

No

No

torch.nn.functional

leaky_relu_

No

No

No

No

No

torch.nn.functional

linear

No

No

No

No

No

torch.nn.functional

log_softmax

No

No

No

No

No

torch.nn.functional

logsigmoid

No

No

No

No

No

torch.nn.functional

max_pool2d

No

No

No

No

No

torch.nn.functional

max_pool2d_with_indices

No

No

No

No

No

torch.nn.functional

max_pool3d

No

No

No

No

No

torch.nn.functional

max_pool3d_with_indices

No

No

No

No

No

torch.nn.functional

mish

No

No

No

No

No

torch.nn.functional

mish_

No

No

No

No

No

torch.nn.functional

mse_loss

No

No

No

No

No

torch.nn.functional

multi_margin_loss

No

No

No

No

No

torch.nn.functional

multilabel_margin_loss

No

No

No

No

No

torch.nn.functional

nll_loss

No

No

No

No

No

torch.nn.functional

one_hot

Yes

No

No

No

No

torch.nn.functional

pad

No

No

No

No

No

torch.nn.functional

pixel_shuffle

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

prelu

No

No

No

No

No

torch.nn.functional

relu

No

No

No

No

No

torch.nn.functional

relu_

No

No

No

No

No

torch.nn.functional

rrelu

No

No

No

No

No

torch.nn.functional

rrelu_

No

No

No

No

No

torch.nn.functional

sigmoid

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

sigmoid_

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

silu

No

No

No

No

No

torch.nn.functional

silu_

No

No

No

No

No

torch.nn.functional

smooth_l1_loss

No

No

No

No

No

torch.nn.functional

softmax

No

No

No

No

No

torch.nn.functional

softplus

No

No

No

No

No

torch.nn.functional

softshrink

No

No

No

No

No

torch.nn.functional

tanh

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

threshold

No

No

No

No

No

torch.nn.functional

threshold_

No

No

No

No

No

torch.nn.functional

upsample

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

upsample_bilinear

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

upsample_nearest

Yes

Yes

Yes

Yes

Yes

torch.nn.functional

cross

Yes

Yes

Yes

Yes

Yes

torch.linalg

vector_norm

No

No

No

No

No

torch.linalg

_adaptive_avg_pool2d

No

No

No

No

No

torch

_adaptive_avg_pool3d

No

No

No

No

No

torch

_add_relu

No

No

No

No

No

torch

_add_relu_

No

No

No

No

No

torch

_addmm_activation

No

No

No

No

No

torch

_aminmax

Yes

Yes

Yes

Yes

Yes

torch

_ctc_loss

No

No

No

No

No

torch

_efficientzerotensor

Yes

Yes

Yes

Yes

Yes

torch

_foreach_abs

Yes

Yes

Yes

Yes

Yes

torch

_foreach_abs_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_acos

Yes

Yes

Yes

Yes

Yes

torch

_foreach_acos_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_add

Yes

Yes

Yes

Yes

Yes

torch

_foreach_add_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_addcdiv

Yes

Yes

Yes

Yes

Yes

torch

_foreach_addcdiv_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_addcmul

Yes

Yes

Yes

Yes

Yes

torch

_foreach_addcmul_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_asin

Yes

Yes

Yes

Yes

Yes

torch

_foreach_asin_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_atan

Yes

Yes

Yes

Yes

Yes

torch

_foreach_atan_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_ceil

Yes

Yes

Yes

Yes

Yes

torch

_foreach_ceil_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_clamp_max

Yes

Yes

Yes

Yes

Yes

torch

_foreach_clamp_max_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_clamp_min

Yes

Yes

Yes

Yes

Yes

torch

_foreach_clamp_min_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_cos

Yes

Yes

Yes

Yes

Yes

torch

_foreach_cos_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_cosh

Yes

Yes

Yes

Yes

Yes

torch

_foreach_cosh_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_div

Yes

Yes

Yes

Yes

Yes

torch

_foreach_div_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_erf

Yes

Yes

Yes

Yes

Yes

torch

_foreach_erf_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_erfc

Yes

Yes

Yes

Yes

Yes

torch

_foreach_erfc_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_exp

Yes

Yes

Yes

Yes

Yes

torch

_foreach_exp_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_expm1

Yes

Yes

Yes

Yes

Yes

torch

_foreach_expm1_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_floor

Yes

Yes

Yes

Yes

Yes

torch

_foreach_floor_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_frac

No

No

No

No

No

torch

_foreach_frac_

No

No

No

No

No

torch

_foreach_lerp

Yes

Yes

Yes

Yes

Yes

torch

_foreach_lerp_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_lgamma

Yes

Yes

Yes

Yes

Yes

torch

_foreach_lgamma_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log10

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log10_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log1p

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log1p_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log2

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log2_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_log_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_maximum

Yes

Yes

Yes

Yes

Yes

torch

_foreach_maximum_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_minimum

Yes

Yes

Yes

Yes

Yes

torch

_foreach_minimum_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_mul

Yes

Yes

Yes

Yes

Yes

torch

_foreach_mul_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_neg

Yes

Yes

Yes

Yes

Yes

torch

_foreach_neg_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_pow

Yes

Yes

Yes

Yes

Yes

torch

_foreach_pow_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_reciprocal

Yes

Yes

Yes

Yes

Yes

torch

_foreach_reciprocal_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_round

Yes

Yes

Yes

Yes

Yes

torch

_foreach_round_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sigmoid

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sigmoid_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sign

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sign_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sin

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sin_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sinh

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sinh_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sqrt

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sqrt_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sub

Yes

Yes

Yes

Yes

Yes

torch

_foreach_sub_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_tan

Yes

Yes

Yes

Yes

Yes

torch

_foreach_tan_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_tanh

Yes

Yes

Yes

Yes

Yes

torch

_foreach_tanh_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_trunc

Yes

Yes

Yes

Yes

Yes

torch

_foreach_trunc_

Yes

Yes

Yes

Yes

Yes

torch

_foreach_zero_

Yes

Yes

Yes

Yes

Yes

torch

_fused_dropout

No

No

No

No

No

torch

_logcumsumexp

No

No

No

No

No

torch

_masked_scale

Yes

Yes

Yes

Yes

Yes

torch

_native_batch_norm_legit

No

No

No

No

No

torch

_native_batch_norm_legit_no_training

No

No

No

No

No

torch

_resize_output_

Yes

Yes

Yes

Yes

Yes

torch

_unique

No

Yes

No

No

No

torch

_unique2

No

Yes

No

No

No

torch

_weight_norm_interface

No

No

No

No

No

torch

abs

Yes

Yes

Yes

Yes

Yes

torch

abs_

Yes

Yes

Yes

Yes

Yes

torch

absolute

Yes

Yes

Yes

Yes

Yes

torch

acos

Yes

Yes

Yes

Yes

Yes

torch

acos_

Yes

Yes

Yes

Yes

Yes

torch

acosh

Yes

Yes

Yes

Yes

Yes

torch

acosh_

Yes

Yes

Yes

Yes

Yes

torch

add

Yes

Yes

Yes

Yes

Yes

torch

addbmm

No

No

No

No

No

torch

addcdiv

Yes

Yes

Yes

Yes

Yes

torch

addcmul

Yes

Yes

Yes

Yes

Yes

torch

addmm

No

No

No

No

No

torch

addmv

No

No

No

No

No

torch

addmv_

No

No

No

No

No

torch

addr

No

No

No

No

No

torch

all

Yes

Yes

Yes

Yes

Yes

torch

amax

Yes

Yes

Yes

Yes

Yes

torch

amin

Yes

Yes

Yes

Yes

Yes

torch

aminmax

Yes

Yes

Yes

Yes

Yes

torch

any

Yes

Yes

Yes

Yes

Yes

torch

arange

Yes

Yes

Yes

Yes

Yes

torch

arccos

Yes

Yes

Yes

Yes

Yes

torch

arccos_

Yes

Yes

Yes

Yes

Yes

torch

arccosh

Yes

Yes

Yes

Yes

Yes

torch

arccosh_

Yes

Yes

Yes

Yes

Yes

torch

arcsin

Yes

Yes

Yes

Yes

Yes

torch

arcsin_

Yes

Yes

Yes

Yes

Yes

torch

arcsinh

Yes

Yes

Yes

Yes

Yes

torch

arcsinh_

Yes

Yes

Yes

Yes

Yes

torch

arctan

Yes

Yes

Yes

Yes

Yes

torch

arctan2

Yes

Yes

Yes

Yes

Yes

torch

arctan_

Yes

Yes

Yes

Yes

Yes

torch

arctanh

No

No

No

No

No

torch

arctanh_

No

No

No

No

No

torch

argmax

Yes

Yes

No

Yes

Yes

torch

argmin

Yes

Yes

No

Yes

Yes

torch

argsort

Yes

Yes

Yes

No

No

torch

as_strided

Yes

Yes

No

Yes

Yes

torch

as_strided_

Yes

Yes

No

Yes

Yes

torch

asin

Yes

Yes

Yes

Yes

Yes

torch

asin_

Yes

Yes

Yes

Yes

Yes

torch

asinh

Yes

Yes

Yes

Yes

Yes

torch

asinh_

Yes

Yes

Yes

Yes

Yes

torch

atan

Yes

Yes

Yes

Yes

Yes

torch

atan2

Yes

Yes

Yes

Yes

Yes

torch

atan_

Yes

Yes

Yes

Yes

Yes

torch

atanh

No

No

No

No

No

torch

atanh_

No

No

No

No

No

torch

baddbmm

No

No

No

No

No

torch

batch_norm

No

No

No

No

No

torch

bernoulli

No

No

No

No

No

torch

binary_cross_entropy_with_logits

No

No

No

No

No

torch

bincount

Yes

Yes

Yes

Yes

No

torch

bitwise_and

Yes

Yes

Yes

Yes

Yes

torch

bitwise_left_shift

Yes

Yes

Yes

Yes

Yes

torch

bitwise_not

Yes

Yes

Yes

Yes

Yes

torch

bitwise_or

Yes

Yes

Yes

Yes

Yes

torch

bitwise_right_shift

Yes

Yes

Yes

Yes

Yes

torch

bitwise_xor

Yes

Yes

Yes

Yes

Yes

torch

bmm

No

No

No

No

No

torch

broadcast_tensors

No

No

No

No

No

torch

bucketize

Yes

Yes

No

No

No

torch

cat

Yes

Yes

Yes

Yes

Yes

torch

ceil

Yes

Yes

Yes

Yes

Yes

torch

ceil_

Yes

Yes

Yes

Yes

Yes

torch

channel_shuffle

Yes

Yes

Yes

Yes

Yes

torch

cholesky

No

No

No

No

No

torch

chunk

No

Yes

No

No

No

torch

clamp

Yes

Yes

Yes

Yes

Yes

torch

clamp_

Yes

Yes

Yes

Yes

Yes

torch

clamp_max

Yes

Yes

Yes

Yes

Yes

torch

clamp_max_

Yes

Yes

Yes

Yes

Yes

torch

clamp_min

Yes

Yes

Yes

Yes

Yes

torch

clamp_min_

Yes

Yes

Yes

Yes

Yes

torch

clip

No

Yes

No

No

No

torch

clip_

No

Yes

No

No

No

torch

clone

Yes

Yes

Yes

Yes

Yes

torch

conj

No

Yes

No

No

No

torch

constant_pad_nd

Yes

Yes

Yes

Yes

Yes

torch

conv1d

No

No

No

No

No

torch

conv2d

No

No

No

No

No

torch

conv3d

No

No

No

No

No

torch

conv_transpose1d

No

No

No

No

No

torch

conv_transpose2d

No

No

No

No

No

torch

conv_transpose3d

No

No

No

No

No

torch

copysign

Yes

Yes

Yes

Yes

Yes

torch

cos

Yes

Yes

Yes

Yes

Yes

torch

cos_

Yes

Yes

Yes

Yes

Yes

torch

cosh

Yes

Yes

Yes

Yes

Yes

torch

cosh_

Yes

Yes

Yes

Yes

Yes

torch

count_nonzero

Yes

Yes

Yes

Yes

Yes

torch

cross

Yes

Yes

Yes

Yes

Yes

torch

cumprod

Yes

Yes

No

Yes

Yes

torch

cumsum

Yes

Yes

No

Yes

Yes

torch

diag

No

No

No

No

No

torch

div

Yes

Yes

Yes

Yes

Yes

torch

divide

Yes

Yes

Yes

Yes

Yes

torch

dot

Yes

Yes

No

No

No

torch

dropout

No

No

No

No

No

torch

embedding

Yes

Yes

Yes

Yes

Yes

torch

embedding_bag

No

No

No

No

No

torch

embedding_renorm_

No

No

No

No

No

torch

empty

Yes

Yes

Yes

Yes

Yes

torch

empty_like

Yes

Yes

Yes

Yes

Yes

torch

empty_strided

Yes

Yes

Yes

Yes

Yes

torch

eq

Yes

Yes

Yes

Yes

Yes

torch

equal

Yes

Yes

Yes

Yes

Yes

torch

erf

Yes

Yes

Yes

Yes

Yes

torch

erf_

Yes

Yes

Yes

Yes

Yes

torch

erfc

Yes

Yes

Yes

Yes

Yes

torch

erfc_

Yes

Yes

Yes

Yes

Yes

torch

erfinv

Yes

Yes

Yes

Yes

Yes

torch

exp

Yes

Yes

Yes

Yes

Yes

torch

exp2

Yes

Yes

Yes

Yes

Yes

torch

exp2_

Yes

Yes

Yes

Yes

Yes

torch

exp_

Yes

Yes

Yes

Yes

Yes

torch

expm1

Yes

Yes

Yes

Yes

Yes

torch

expm1_

Yes

Yes

Yes

Yes

Yes

torch

eye

Yes

Yes

Yes

Yes

Yes

torch

fill

Yes

Yes

Yes

Yes

Yes

torch

fill_

Yes

Yes

Yes

Yes

Yes

torch

flatten

No

Yes

No

No

No

torch

flip

Yes

Yes

Yes

Yes

Yes

torch

floor

Yes

Yes

Yes

Yes

Yes

torch

floor_

Yes

Yes

Yes

Yes

Yes

torch

floor_divide

Yes

Yes

Yes

Yes

Yes

torch

fmax

Yes

Yes

Yes

Yes

Yes

torch

fmin

Yes

Yes

Yes

Yes

Yes

torch

fmod

Yes

Yes

Yes

Yes

Yes

torch

frac

No

No

No

No

No

torch

frac_

No

No

No

No

No

torch

frexp

No

No

No

No

No

torch

full

Yes

Yes

Yes

Yes

Yes

torch

full_like

Yes

Yes

Yes

Yes

Yes

torch

gather

Yes

Yes

Yes

Yes

Yes

torch

ge

Yes

Yes

No

Yes

Yes

torch

greater

Yes

Yes

No

Yes

Yes

torch

greater_equal

Yes

Yes

No

Yes

Yes

torch

grid_sampler_2d

No

No

No

No

No

torch

grid_sampler_3d

No

No

No

No

No

torch

gt

Yes

Yes

No

Yes

Yes

torch

hardshrink

No

No

No

No

No

torch

heaviside

Yes

Yes

No

No

No

torch

hypot

No

No

No

No

No

torch

index_copy

Yes

Yes

Yes

Yes

Yes

torch

index_fill

Yes

Yes

Yes

Yes

Yes

torch

index_put

No

Yes

No

No

No

torch

index_put_

No

Yes

No

No

No

torch

index_reduce

Yes

Yes

No

No

No

torch

index_select

Yes

Yes

Yes

Yes

Yes

torch

instance_norm

No

No

No

No

No

torch

is_complex

No

Yes

No

No

No

torch

is_floating_point

No

Yes

No

No

No

torch

is_nonzero

No

Yes

No

No

No

torch

isfinite

Yes

Yes

Yes

Yes

Yes

torch

isin

Yes

Yes

Yes

Yes

Yes

torch

isinf

Yes

Yes

Yes

Yes

Yes

torch

isnan

Yes

Yes

Yes

Yes

Yes

torch

isneginf

Yes

Yes

Yes

Yes

Yes

torch

isposinf

Yes

Yes

Yes

Yes

Yes

torch

kl_div

No

No

No

No

No

torch

kthvalue

Yes

Yes

No

No

No

torch

layer_norm

No

No

No

No

No

torch

le

Yes

Yes

No

Yes

Yes

torch

lerp

Yes

Yes

Yes

Yes

Yes

torch

less

Yes

Yes

No

Yes

Yes

torch

less_equal

Yes

Yes

No

Yes

Yes

torch

lgamma

Yes

Yes

Yes

Yes

Yes

torch

linspace

Yes

Yes

Yes

Yes

Yes

torch

log

Yes

Yes

Yes

Yes

Yes

torch

log10

Yes

Yes

Yes

Yes

Yes

torch

log10_

Yes

Yes

Yes

Yes

Yes

torch

log1p

Yes

Yes

Yes

Yes

Yes

torch

log1p_

Yes

Yes

Yes

Yes

Yes

torch

log2

Yes

Yes

Yes

Yes

Yes

torch

log2_

Yes

Yes

Yes

Yes

Yes

torch

log_

Yes

Yes

Yes

Yes

Yes

torch

log_softmax

No

No

No

No

No

torch

logaddexp

No

No

No

No

No

torch

logaddexp2

No

No

No

No

No

torch

logcumsumexp

No

No

No

No

No

torch

logical_and

Yes

Yes

No

Yes

Yes

torch

logical_not

No

Yes

Yes

Yes

Yes

torch

logical_or

No

No

No

Yes

Yes

torch

logical_xor

No

No

No

Yes

Yes

torch

logit

No

No

No

No

No

torch

logit_

No

No

No

No

No

torch

logspace

Yes

Yes

Yes

Yes

Yes

torch

logsumexp

No

No

No

No

No

torch

lt

Yes

Yes

No

Yes

Yes

torch

masked_fill

Yes

Yes

No

Yes

Yes

torch

masked_scatter

Yes

Yes

No

Yes

Yes

torch

masked_select

No

Yes

No

No

No

torch

matmul

No

No

No

No

No

torch

max

Yes

Yes

No

No

No

torch

max_pool2d

No

No

No

No

No

torch

max_pool3d

No

No

No

No

No

torch

maximum

Yes

Yes

Yes

Yes

Yes

torch

mean

Yes

Yes

Yes

Yes

Yes

torch

median

Yes

Yes

No

No

No

torch

meshgrid

No

No

No

No

No

torch

min

Yes

Yes

No

No

No

torch

minimum

Yes

Yes

Yes

Yes

Yes

torch

mm

No

No

No

No

No

torch

mul

Yes

Yes

Yes

Yes

No

torch

multinomial

No

No

No

No

No

torch

mv

No

No

No

No

No

torch

nan_to_num

Yes

Yes

Yes

Yes

Yes

torch

nan_to_num_

Yes

Yes

Yes

Yes

Yes

torch

nansum

Yes

Yes

Yes

Yes

Yes

torch

narrow

No

No

No

No

No

torch

native_dropout

No

No

No

No

No

torch

native_group_norm

No

No

No

No

No

torch

native_layer_norm

No

No

No

No

No

torch

ne

Yes

Yes

Yes

Yes

Yes

torch

neg

Yes

Yes

Yes

Yes

Yes

torch

neg_

Yes

Yes

Yes

Yes

Yes

torch

nextafter

No

No

No

No

No

torch

nonzero

No

Yes

No

No

Yes

torch

norm

No

No

No

No

No

torch

normal

No

No

No

No

No

torch

not_equal

Yes

Yes

Yes

Yes

Yes

torch

ones

No

Yes

No

Yes

Yes

torch

ones_like

No

Yes

Yes

Yes

Yes

torch

permute

Yes

Yes

Yes

Yes

Yes

torch

pixel_shuffle

Yes

Yes

Yes

Yes

Yes

torch

poisson

No

No

No

No

No

torch

pow

Yes

Yes

Yes

Yes

Yes

torch

prelu

No

No

No

No

No

torch

prod

Yes

Yes

Yes

Yes

Yes

torch

put

No

No

No

No

No

torch

randperm

Yes

Yes

Yes

Yes

Yes

torch

reciprocal

Yes

Yes

Yes

Yes

Yes

torch

reciprocal_

Yes

Yes

Yes

Yes

Yes

torch

relu

No

No

No

No

No

torch

relu_

No

No

No

No

No

torch

remainder

Yes

Yes

Yes

Yes

Yes

torch

reshape

Yes

Yes

Yes

Yes

Yes

torch

resolve_conj

No

No

No

No

No

torch

resolve_neg

No

No

No

No

No

torch

result_type

No

Yes

No

No

No

torch

roll

No

Yes

Yes

Yes

Yes

torch

round

Yes

Yes

Yes

Yes

Yes

torch

round_

Yes

Yes

Yes

Yes

Yes

torch

rsqrt

Yes

Yes

Yes

Yes

Yes

torch

rsqrt_

Yes

Yes

Yes

Yes

Yes

torch

rsub

Yes

Yes

Yes

Yes

Yes

torch

scalar_tensor

Yes

Yes

Yes

Yes

Yes

torch

scatter

Yes

Yes

Yes

Yes

Yes

torch

scatter_add

No

No

No

No

No

torch

scatter_reduce

No

No

No

No

No

torch

searchsorted

Yes

Yes

No

No

No

torch

select

Yes

Yes

Yes

Yes

Yes

torch

sgn

Yes

Yes

Yes

Yes

Yes

torch

sigmoid

Yes

Yes

Yes

Yes

Yes

torch

sigmoid_

Yes

Yes

Yes

Yes

Yes

torch

sign

Yes

Yes

Yes

Yes

Yes

torch

signbit

Yes

Yes

Yes

Yes

Yes

torch

sin

Yes

Yes

Yes

Yes

Yes

torch

sin_

Yes

Yes

Yes

Yes

Yes

torch

sinc

Yes

Yes

Yes

Yes

Yes

torch

sinc_

Yes

Yes

Yes

Yes

Yes

torch

sinh

Yes

Yes

Yes

Yes

Yes

torch

sinh_

No

No

No

No

No

torch

softmax

No

No

No

No

No

torch

sort

Yes

Yes

Yes

No

No

torch

split_with_sizes

No

Yes

Yes

No

No

torch

sqrt

Yes

Yes

Yes

Yes

Yes

torch

sqrt_

Yes

Yes

Yes

Yes

Yes

torch

square

No

Yes

No

Yes

Yes

torch

square_

No

Yes

No

Yes

Yes

torch

squeeze

Yes

Yes

Yes

Yes

Yes

torch

stack

No

Yes

Yes

No

No

torch

std

No

No

No

No

No

torch

std_mean

No

No

No

No

No

torch

sub

Yes

Yes

Yes

Yes

Yes

torch

sum

Yes

Yes

Yes

Yes

Yes

torch

t

Yes

Yes

Yes

Yes

Yes

torch

take

Yes

Yes

No

No

No

torch

tan

Yes

Yes

Yes

Yes

Yes

torch

tan_

Yes

Yes

Yes

Yes

Yes

torch

tanh

Yes

Yes

Yes

Yes

Yes

torch

tanh_

Yes

Yes

Yes

Yes

Yes

torch

threshold

No

No

No

No

No

torch

threshold_

No

No

No

No

No

torch

topk

Yes

Yes

Yes

No

No

torch

trace

Yes

Yes

No

No

No

torch

transpose

Yes

Yes

Yes

Yes

Yes

torch

tril

No

No

No

Yes

Yes

torch

tril_indices

Yes

Yes

No

No

No

torch

triu

No

No

No

Yes

Yes

torch

triu_indices

Yes

Yes

No

No

No

torch

trunc

Yes

Yes

Yes

Yes

Yes

torch

trunc_

Yes

Yes

Yes

Yes

Yes

torch

unbind

No

Yes

No

No

No

torch

unique

No

Yes

No

No

No

torch

unsqueeze

Yes

Yes

Yes

No

No

torch

var

No

No

No

No

No

torch

var_mean

No

No

No

No

No

torch

vdot

Yes

Yes

No

No

No

torch

where

Yes

Yes

Yes

Yes

Yes

torch

xlogy

Yes

Yes

Yes

Yes

Yes

torch

xlogy_

Yes

Yes

Yes

Yes

Yes

torch

zero_

Yes

Yes

Yes

Yes

Yes

torch

zeros

No

Yes

Yes

Yes

Yes

torch

zeros_like

No

Yes

Yes

Yes

Yes

torch

glu_jvp

No

No

No

No

No

torch.ops.aten

AdaptiveAvgPool1d

No

No

No

No

No

torch.nn

AdaptiveAvgPool2d

No

No

No

No

No

torch.nn

AdaptiveAvgPool3d

No

No

No

No

No

torch.nn

AvgPool1d

No

No

No

No

No

torch.nn

AvgPool2d

No

No

No

No

No

torch.nn

AvgPool3d

No

No

No

No

No

torch.nn

BCELoss

No

No

No

No

No

torch.nn

BCEWithLogitsLoss

No

No

No

No

No

torch.nn

BatchNorm1d

No

No

No

No

No

torch.nn

BatchNorm2d

No

No

No

No

No

torch.nn

ChannelShuffle

Yes

Yes

Yes

Yes

Yes

torch.nn

ConstantPad1d

No

No

No

No

No

torch.nn

Conv1d

No

No

No

No

No

torch.nn

Conv2d

No

No

No

No

No

torch.nn

Conv3d

No

No

No

No

No

torch.nn

ConvTranspose1d

No

No

No

No

No

torch.nn

ConvTranspose2d

No

No

No

No

No

torch.nn

ConvTranspose3d

No

No

No

No

No

torch.nn

CrossEntropyLoss

No

No

No

No

No

torch.nn

Dropout

No

No

No

No

No

torch.nn

ELU

No

No

No

No

No

torch.nn

EmbeddingBag

No

No

No

No

No

torch.nn

GELU

No

No

No

No

No

torch.nn

GLU

No

No

No

No

No

torch.nn

Hardshrink

No

No

No

No

No

torch.nn

Hardsigmoid

No

No

No

No

No

torch.nn

Hardswish

No

No

No

No

No

torch.nn

Hardtanh

Yes

Yes

Yes

Yes

Yes

torch.nn

HuberLoss

No

No

No

No

No

torch.nn

InstanceNorm2d

No

No

No

No

No

torch.nn

KLDivLoss

No

No

No

No

No

torch.nn

LayerNorm

No

No

No

No

No

torch.nn

LeakyReLU

No

No

No

No

No

torch.nn

Linear

No

No

No

No

No

torch.nn

LogSigmoid

No

No

No

No

No

torch.nn

LogSoftmax

No

No

No

No

No

torch.nn

MSELoss

No

No

No

No

No

torch.nn

MaxPool2d

No

No

No

No

No

torch.nn

MaxPool3d

No

No

No

No

No

torch.nn

Mish

No

No

No

No

No

torch.nn

MultiLabelMarginLoss

No

No

No

No

No

torch.nn

MultiMarginLoss

No

No

No

No

No

torch.nn

NLLLoss

No

No

No

No

No

torch.nn

PReLU

No

No

No

No

No

torch.nn

PixelShuffle

Yes

Yes

Yes

Yes

Yes

torch.nn

RReLU

No

No

No

No

No

torch.nn

ReLU

No

No

No

No

No

torch.nn

ReflectionPad1d

Yes

Yes

Yes

Yes

Yes

torch.nn

ReflectionPad2d

Yes

Yes

Yes

Yes

Yes

torch.nn

ReflectionPad3d

Yes

Yes

Yes

Yes

Yes

torch.nn

ReplicationPad1d

Yes

Yes

Yes

Yes

Yes

torch.nn

ReplicationPad2d

Yes

Yes

Yes

Yes

Yes

torch.nn

ReplicationPad3d

Yes

Yes

Yes

Yes

Yes

torch.nn

SiLU

No

No

No

No

No

torch.nn

Sigmoid

Yes

Yes

Yes

Yes

Yes

torch.nn

SmoothL1Loss

No

No

No

No

No

torch.nn

Softmax

No

No

No

No

No

torch.nn

Softplus

No

No

No

No

No

torch.nn

Softshrink

No

No

No

No

No

torch.nn

Tanh

Yes

Yes

Yes

Yes

Yes

torch.nn

Threshold

No

No

No

No

No

torch.nn

Upsample

Yes

Yes

Yes

Yes

Yes

torch.nn

UpsamplingNearest2d

Yes

Yes

Yes

Yes

Yes

torch.nn

weight_norm

No

No

No

No

No

torch.nn.utils

T

Yes

Yes

Yes

Yes

Yes

torch.Tensor

__and__

No

Yes

No

Yes

Yes

torch.Tensor

__iand__

No

Yes

No

Yes

Yes

torch.Tensor

__ilshift__

Yes

Yes

Yes

Yes

Yes

torch.Tensor

__ior__

No

Yes

No

Yes

Yes

torch.Tensor

__irshift__

Yes

Yes

Yes

Yes

Yes

torch.Tensor

__ixor__

No

Yes

No

Yes

Yes

torch.Tensor

__lshift__

Yes

Yes

Yes

Yes

Yes

torch.Tensor

__or__

No

Yes

No

Yes

Yes

torch.Tensor

__rshift__

Yes

Yes

Yes

Yes

Yes

torch.Tensor

__xor__

No

Yes

No

Yes

Yes

torch.Tensor

abs

Yes

Yes

Yes

Yes

Yes

torch.Tensor

abs_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

absolute

Yes

Yes

Yes

Yes

Yes

torch.Tensor

absolute_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

acos

Yes

Yes

Yes

Yes

Yes

torch.Tensor

acos_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

acosh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

acosh_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

add

Yes

Yes

Yes

Yes

Yes

torch.Tensor

add_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

addbmm

No

No

No

No

No

torch.Tensor

addbmm_

No

No

No

No

No

torch.Tensor

addcdiv

Yes

Yes

Yes

Yes

Yes

torch.Tensor

addcdiv_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

addcmul

Yes

Yes

Yes

Yes

Yes

torch.Tensor

addcmul_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

addmm

No

No

No

No

No

torch.Tensor

addmm_

No

No

No

No

No

torch.Tensor

addmv

No

No

No

No

No

torch.Tensor

addmv_

No

No

No

No

No

torch.Tensor

addr

No

No

No

No

No

torch.Tensor

addr_

No

No

No

No

No

torch.Tensor

all

Yes

Yes

Yes

Yes

Yes

torch.Tensor

amax

Yes

Yes

Yes

Yes

Yes

torch.Tensor

amin

Yes

Yes

Yes

Yes

Yes

torch.Tensor

aminmax

Yes

Yes

Yes

Yes

Yes

torch.Tensor

any

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arccos

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arccos_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arccosh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arccosh_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arcsin

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arcsin_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arcsinh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arcsinh_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arctan

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arctan2

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arctan2_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arctan_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

arctanh

No

No

No

No

No

torch.Tensor

arctanh_

No

No

No

No

No

torch.Tensor

argmax

Yes

Yes

No

Yes

Yes

torch.Tensor

argmin

Yes

Yes

No

Yes

Yes

torch.Tensor

argsort

Yes

Yes

Yes

No

No

torch.Tensor

as_strided

Yes

Yes

No

Yes

Yes

torch.Tensor

as_strided_

Yes

Yes

No

Yes

Yes

torch.Tensor

asin

Yes

Yes

Yes

Yes

Yes

torch.Tensor

asin_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

asinh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

asinh_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

atan

Yes

Yes

Yes

Yes

Yes

torch.Tensor

atan2

Yes

Yes

Yes

Yes

Yes

torch.Tensor

atan2_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

atan_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

atanh

No

No

No

No

No

torch.Tensor

atanh_

No

No

No

No

No

torch.Tensor

baddbmm

No

No

No

No

No

torch.Tensor

baddbmm_

No

No

No

No

No

torch.Tensor

bernoulli

No

No

No

No

No

torch.Tensor

bernoulli_

Yes

Yes

Yes

Yes

No

torch.Tensor

bincount

Yes

Yes

Yes

Yes

No

torch.Tensor

bitwise_and

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_and_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_left_shift

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_left_shift_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_not

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_not_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_or

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_or_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_right_shift

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_right_shift_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_xor

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bitwise_xor_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

bmm

No

No

No

No

No

torch.Tensor

ceil

Yes

Yes

Yes

Yes

Yes

torch.Tensor

ceil_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cholesky

No

No

No

No

No

torch.Tensor

chunk

No

Yes

No

No

No

torch.Tensor

clamp

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_max

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_max_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_min

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_min_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clip

No

Yes

No

No

No

torch.Tensor

clip_

No

Yes

No

No

No

torch.Tensor

clone

Yes

Yes

Yes

Yes

Yes

torch.Tensor

conj

No

Yes

No

No

No

torch.Tensor

copy_

No

Yes

No

Yes

Yes

torch.Tensor

copysign

Yes

Yes

Yes

Yes

Yes

torch.Tensor

copysign_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cos

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cos_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cosh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cosh_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

count_nonzero

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cross

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cumprod

Yes

Yes

No

Yes

Yes

torch.Tensor

cumprod_

Yes

Yes

No

No

No

torch.Tensor

cumsum

Yes

Yes

No

Yes

Yes

torch.Tensor

cumsum_

Yes

Yes

No

Yes

Yes

torch.Tensor

diag

No

No

No

No

No

torch.Tensor

div

Yes

Yes

Yes

Yes

Yes

torch.Tensor

div_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

divide

Yes

Yes

Yes

Yes

Yes

torch.Tensor

divide_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

dot

Yes

Yes

No

No

No

torch.Tensor

eq

Yes

Yes

Yes

Yes

Yes

torch.Tensor

eq_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

equal

Yes

Yes

Yes

Yes

Yes

torch.Tensor

erf

Yes

Yes

Yes

Yes

Yes

torch.Tensor

erf_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

erfc

Yes

Yes

Yes

Yes

Yes

torch.Tensor

erfc_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

erfinv

Yes

Yes

Yes

Yes

Yes

torch.Tensor

erfinv_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

exp

Yes

Yes

Yes

Yes

Yes

torch.Tensor

exp2

Yes

Yes

Yes

Yes

Yes

torch.Tensor

exp2_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

exp_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

expand

Yes

Yes

No

No

No

torch.Tensor

expand_as

Yes

Yes

No

Yes

Yes

torch.Tensor

expm1

Yes

Yes

Yes

Yes

Yes

torch.Tensor

expm1_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

exponential_

No

No

No

No

No

torch.Tensor

fill_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

flatten

No

Yes

No

No

No

torch.Tensor

flip

Yes

Yes

Yes

Yes

Yes

torch.Tensor

floor

Yes

Yes

Yes

Yes

Yes

torch.Tensor

floor_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

floor_divide

Yes

Yes

Yes

Yes

Yes

torch.Tensor

floor_divide_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

fmax

Yes

Yes

Yes

Yes

Yes

torch.Tensor

fmin

Yes

Yes

Yes

Yes

Yes

torch.Tensor

fmod

Yes

Yes

Yes

Yes

Yes

torch.Tensor

fmod_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

frac

No

No

No

No

No

torch.Tensor

frac_

No

No

No

No

No

torch.Tensor

frexp

No

No

No

No

No

torch.Tensor

gather

Yes

Yes

Yes

Yes

Yes

torch.Tensor

ge

Yes

Yes

No

Yes

Yes

torch.Tensor

ge_

Yes

Yes

No

Yes

Yes

torch.Tensor

geometric_

No

No

No

No

No

torch.Tensor

greater

Yes

Yes

No

Yes

Yes

torch.Tensor

greater_

Yes

Yes

No

Yes

Yes

torch.Tensor

greater_equal

Yes

Yes

No

Yes

Yes

torch.Tensor

greater_equal_

Yes

Yes

No

Yes

Yes

torch.Tensor

gt

Yes

Yes

No

Yes

Yes

torch.Tensor

gt_

Yes

Yes

No

Yes

Yes

torch.Tensor

hardshrink

No

No

No

No

No

torch.Tensor

heaviside

Yes

Yes

No

No

No

torch.Tensor

heaviside_

Yes

Yes

No

No

No

torch.Tensor

hypot

No

No

No

No

No

torch.Tensor

hypot_

No

No

No

No

No

torch.Tensor

index_add_

No

Yes

No

No

No

torch.Tensor

index_copy

Yes

Yes

Yes

Yes

Yes

torch.Tensor

index_copy_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

index_fill

Yes

Yes

Yes

Yes

Yes

torch.Tensor

index_fill_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

index_put

No

Yes

No

No

No

torch.Tensor

index_put_

No

Yes

No

No

No

torch.Tensor

index_reduce

Yes

Yes

No

No

No

torch.Tensor

index_reduce_

Yes

Yes

No

No

No

torch.Tensor

index_select

Yes

Yes

Yes

Yes

Yes

torch.Tensor

is_complex

No

Yes

No

No

No

torch.Tensor

is_floating_point

No

Yes

No

No

No

torch.Tensor

is_nonzero

No

Yes

No

No

No

torch.Tensor

isfinite

Yes

Yes

Yes

Yes

Yes

torch.Tensor

isinf

Yes

Yes

Yes

Yes

Yes

torch.Tensor

isnan

Yes

Yes

Yes

Yes

Yes

torch.Tensor

isneginf

Yes

Yes

Yes

Yes

Yes

torch.Tensor

isposinf

Yes

Yes

Yes

Yes

Yes

torch.Tensor

item

No

No

No

No

No

torch.Tensor

kthvalue

Yes

Yes

No

No

No

torch.Tensor

le

Yes

Yes

No

Yes

Yes

torch.Tensor

le_

Yes

Yes

No

Yes

Yes

torch.Tensor

lerp

Yes

Yes

Yes

Yes

Yes

torch.Tensor

lerp_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

less

Yes

Yes

No

Yes

Yes

torch.Tensor

less_

Yes

Yes

No

Yes

Yes

torch.Tensor

less_equal

Yes

Yes

No

Yes

Yes

torch.Tensor

less_equal_

Yes

Yes

No

Yes

Yes

torch.Tensor

log

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log10

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log10_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log1p

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log1p_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log2

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log2_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log_normal_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

log_softmax

No

No

No

No

No

torch.Tensor

logaddexp

No

No

No

No

No

torch.Tensor

logaddexp2

No

No

No

No

No

torch.Tensor

logcumsumexp

No

No

No

No

No

torch.Tensor

logical_and

Yes

Yes

No

Yes

Yes

torch.Tensor

logical_and_

Yes

Yes

No

Yes

Yes

torch.Tensor

logical_not

No

Yes

Yes

Yes

Yes

torch.Tensor

logical_not_

No

No

No

Yes

Yes

torch.Tensor

logical_or

No

No

No

Yes

Yes

torch.Tensor

logical_or_

No

No

No

Yes

Yes

torch.Tensor

logical_xor

No

No

No

Yes

Yes

torch.Tensor

logical_xor_

No

No

No

Yes

Yes

torch.Tensor

logit

No

No

No

No

No

torch.Tensor

logit_

No

No

No

No

No

torch.Tensor

logsumexp

No

No

No

No

No

torch.Tensor

lt

Yes

Yes

No

Yes

Yes

torch.Tensor

lt_

Yes

Yes

No

Yes

Yes

torch.Tensor

masked_fill

Yes

Yes

No

Yes

Yes

torch.Tensor

masked_fill_

Yes

Yes

No

Yes

Yes

torch.Tensor

masked_scatter

Yes

Yes

No

Yes

Yes

torch.Tensor

masked_scatter_

Yes

Yes

No

Yes

Yes

torch.Tensor

masked_select

No

Yes

No

No

No

torch.Tensor

matmul

No

No

No

No

No

torch.Tensor

max

Yes

Yes

No

No

No

torch.Tensor

maximum

Yes

Yes

Yes

Yes

Yes

torch.Tensor

mean

Yes

Yes

Yes

Yes

Yes

torch.Tensor

median

Yes

Yes

No

No

No

torch.Tensor

min

Yes

Yes

No

No

No

torch.Tensor

minimum

Yes

Yes

Yes

Yes

Yes

torch.Tensor

mm

No

No

No

No

No

torch.Tensor

mul

Yes

Yes

Yes

Yes

No

torch.Tensor

mul_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

multinomial

No

No

No

No

No

torch.Tensor

mv

No

No

No

No

No

torch.Tensor

nan_to_num

Yes

Yes

Yes

Yes

Yes

torch.Tensor

nan_to_num_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

nansum

Yes

Yes

Yes

Yes

Yes

torch.Tensor

narrow

No

No

No

No

No

torch.Tensor

ne

Yes

Yes

Yes

Yes

Yes

torch.Tensor

ne_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

neg

Yes

Yes

Yes

Yes

Yes

torch.Tensor

neg_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

new_empty

No

Yes

No

Yes

No

torch.Tensor

new_empty_strided

No

Yes

No

Yes

No

torch.Tensor

new_full

No

Yes

No

Yes

No

torch.Tensor

new_ones

No

Yes

No

Yes

No

torch.Tensor

new_zeros

Yes

Yes

Yes

Yes

Yes

torch.Tensor

nextafter

No

No

No

No

No

torch.Tensor

nextafter_

No

No

No

No

No

torch.Tensor

nonzero

No

Yes

No

No

Yes

torch.Tensor

norm

No

No

No

No

No

torch.Tensor

normal_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

not_equal

Yes

Yes

Yes

Yes

Yes

torch.Tensor

not_equal_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

permute

Yes

Yes

Yes

Yes

Yes

torch.Tensor

pin_memory

No

Yes

No

Yes

Yes

torch.Tensor

pow

Yes

Yes

Yes

Yes

Yes

torch.Tensor

pow_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

prod

Yes

Yes

Yes

Yes

Yes

torch.Tensor

put

No

No

No

No

No

torch.Tensor

put_

No

No

No

No

No

torch.Tensor

random_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

reciprocal

Yes

Yes

Yes

Yes

Yes

torch.Tensor

reciprocal_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

relu

No

No

No

No

No

torch.Tensor

relu_

No

No

No

No

No

torch.Tensor

remainder

Yes

Yes

Yes

Yes

Yes

torch.Tensor

remainder_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

repeat

Yes

Yes

No

Yes

Yes

torch.Tensor

repeat_interleave

Yes

Yes

No

Yes

Yes

torch.Tensor

reshape

Yes

Yes

Yes

Yes

Yes

torch.Tensor

resize_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

resolve_conj

No

No

No

No

No

torch.Tensor

resolve_neg

No

No

No

No

No

torch.Tensor

roll

No

Yes

Yes

Yes

Yes

torch.Tensor

round

Yes

Yes

Yes

Yes

Yes

torch.Tensor

round_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

rsqrt

Yes

Yes

Yes

Yes

Yes

torch.Tensor

rsqrt_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

scatter

Yes

Yes

Yes

Yes

Yes

torch.Tensor

scatter_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

scatter_add

No

No

No

No

No

torch.Tensor

scatter_add_

No

No

No

No

No

torch.Tensor

scatter_reduce

No

No

No

No

No

torch.Tensor

scatter_reduce_

No

No

No

No

No

torch.Tensor

select

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sgn

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sgn_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sigmoid

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sigmoid_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sign

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sign_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

signbit

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sin

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sin_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sinc

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sinc_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sinh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sinh_

No

No

No

No

No

torch.Tensor

sort

Yes

Yes

Yes

No

No

torch.Tensor

split_with_sizes

No

Yes

Yes

No

No

torch.Tensor

sqrt

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sqrt_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

square

No

Yes

No

Yes

Yes

torch.Tensor

square_

No

Yes

No

Yes

Yes

torch.Tensor

squeeze

Yes

Yes

Yes

Yes

Yes

torch.Tensor

squeeze_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

std

No

No

No

No

No

torch.Tensor

sub

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sub_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

sum

Yes

Yes

Yes

Yes

Yes

torch.Tensor

t

Yes

Yes

Yes

Yes

Yes

torch.Tensor

take

Yes

Yes

No

No

No

torch.Tensor

tan

Yes

Yes

Yes

Yes

Yes

torch.Tensor

tan_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

tanh

Yes

Yes

Yes

Yes

Yes

torch.Tensor

tanh_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

to

Yes

Yes

Yes

Yes

Yes

torch.Tensor

topk

Yes

Yes

Yes

No

No

torch.Tensor

trace

Yes

Yes

No

No

No

torch.Tensor

transpose

Yes

Yes

Yes

Yes

Yes

torch.Tensor

tril

No

No

No

Yes

Yes

torch.Tensor

tril_

No

No

No

Yes

Yes

torch.Tensor

triu

No

No

No

Yes

Yes

torch.Tensor

triu_

No

No

No

Yes

Yes

torch.Tensor

trunc

Yes

Yes

Yes

Yes

Yes

torch.Tensor

trunc_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

unbind

No

Yes

No

No

No

torch.Tensor

uniform_

Yes

Yes

No

No

No

torch.Tensor

unique

No

Yes

No

No

No

torch.Tensor

unsqueeze

Yes

Yes

Yes

No

No

torch.Tensor

unsqueeze_

Yes

Yes

Yes

No

No

torch.Tensor

var

No

No

No

No

No

torch.Tensor

vdot

Yes

Yes

No

No

No

torch.Tensor

view

Yes

Yes

Yes

Yes

Yes

torch.Tensor

where

Yes

Yes

Yes

Yes

Yes

torch.Tensor

xlogy

Yes

Yes

Yes

Yes

Yes

torch.Tensor

xlogy_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

zero_

Yes

Yes

Yes

Yes

Yes

torch.Tensor

entr

Yes

Yes

Yes

Yes

Yes

torch.special

erf

Yes

Yes

Yes

Yes

Yes

torch.special

erfc

Yes

Yes

Yes

Yes

Yes

torch.special

erfcx

Yes

Yes

Yes

Yes

Yes

torch.special

erfinv

Yes

Yes

Yes

Yes

Yes

torch.special

exp2

Yes

Yes

Yes

Yes

Yes

torch.special

expit

No

No

No

No

No

torch.special

expm1

Yes

Yes

Yes

Yes

Yes

torch.special

log1p

Yes

Yes

Yes

Yes

Yes

torch.special

logit

No

No

No

No

No

torch.special

logsumexp

No

No

No

No

No

torch.special

round

Yes

Yes

Yes

Yes

Yes

torch.special

sinc

Yes

Yes

Yes

Yes

Yes

torch.special

softmax

No

No

No

No

No

torch.special

xlog1py

Yes

Yes

Yes

Yes

Yes

torch.special

xlogy

Yes

Yes

Yes

Yes

Yes

torch.special

batched_nms

No

No

No

No

No

torchvision.ops

deform_conv2d

No

No

No

No

No

torchvision.ops

nms

No

No

No

No

No

torchvision.ops

roi_align

No

No

No

No

No

torchvision.ops

_safe_softmax

No

No

No

No

No

torch.ops