PyTorch OperatorsĀ¶

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 more details, see Fused Optimizers and Custom Ops for Intel Gaudi.

PyTorch Operators Support SummaryĀ¶

PyTorch Operator

FP32

BF16

FP16

INT64

INT32

INT8

BOOL

Operator Type

adaptive_avg_pool1d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

adaptive_avg_pool2d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

adaptive_avg_pool3d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

avg_pool1d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

avg_pool2d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

avg_pool3d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

batch_norm

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

binary_cross_entropy

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

binary_cross_entropy_with_logits

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

conv_transpose2d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

conv1d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

conv2d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

conv3d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

dropout

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

embedding

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.nn.functional

embedding_bag

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

elu

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

elu_

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

gelu

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

glu

Yes

No

No

No

No

No

No

torch.nn.functional

grid_sample

Yes

No

No

No

No

No

No

torch.nn.functional

hardshrink

Yes

Yes

No

No

No

No

No

torch.nn.functional

hardsigmoid

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

hardtanh

Yes

No

No

No

No

No

No

torch.nn.functional

hardtanh_

Yes

No

No

No

No

No

No

torch.nn.functional

huber_loss

Yes

Yes

No

No

No

No

No

torch.nn.functional

instance_norm

Yes

Yes

No

No

No

No

No

torch.nn.functional

kl_div

Yes

Yes

No

No

No

No

No

torch.nn.functional

l1_loss

Yes

Yes

No

No

No

No

No

torch.nn.functional

layer_norm

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

leaky_relu

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

linear

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

log_softmax

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

logsigmoid

Yes

No

No

No

No

No

No

torch.nn.functional

max_pool2d

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

max_pool2d_with_indices

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

max_pool3d

Yes

Yes

No

No

No

No

No

torch.nn.functional

max_pool3d_with_indices

Yes

Yes

No

No

No

No

No

torch.nn.functional

mish

Yes

Yes

No

No

No

No

No

torch.nn.functional

mse_loss

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

nll_loss

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

one_hot

Yes

Yes

Yes

Yes

Yes

No

No

torch.nn.functional

pad

Yes

Yes

No

No

No

No

No

torch.nn.functional

pixel_shuffle

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.nn.functional

prelu

Yes

Yes

No

No

No

No

No

torch.nn.functional

relu

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

relu_

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

rrelu

Yes

Yes

No

No

No

No

No

torch.nn.functional

sigmoid

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

silu

Yes

Yes

No

No

No

No

No

torch.nn.functional

smooth_l1_loss

Yes

Yes

No

No

No

No

No

torch.nn.functional

softmax

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

softplus

Yes

Yes

No

No

No

No

No

torch.nn.functional

softshrink

Yes

Yes

No

No

No

No

No

torch.nn.functional

tanh

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

threshold

Yes

Yes

No

No

No

No

No

torch.nn.functional

upsample

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

upsample_nearest

Yes

Yes

Yes

No

No

No

No

torch.nn.functional

vector_norm

Yes

Yes

No

No

No

No

No

torch.linalg

abs

Yes

Yes

Yes

No

No

No

No

torch

absolute

Yes

Yes

Yes

No

No

No

No

torch

acos

Yes

Yes

No

No

No

No

No

torch

acosh

Yes

Yes

No

No

No

No

No

torch

add

Yes

Yes

Yes

Yes

Yes

No

No

torch

addbmm

Yes

Yes

Yes

No

No

No

No

torch

addcdiv

Yes

Yes

No

No

No

No

No

torch

addcmul

Yes

Yes

No

No

No

No

No

torch

addmm

Yes

Yes

Yes

No

No

No

No

torch

addmv

Yes

Yes

Yes

No

No

No

No

torch

addmv_

Yes

Yes

Yes

No

No

No

No

torch

addr

Yes

Yes

Yes

No

No

No

No

torch

all

Yes

Yes

No

No

Yes

No

Yes

torch

amax

Yes

Yes

Yes

No

Yes

No

No

torch

amin

Yes

Yes

Yes

No

Yes

No

No

torch

aminmax

Yes

Yes

No

Yes

Yes

Yes

No

torch

any

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

arange

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

arccos

Yes

Yes

No

No

No

No

No

torch

arccosh

Yes

Yes

No

No

No

No

No

torch

arcsin

Yes

Yes

No

No

No

No

No

torch

arcsinh

Yes

Yes

No

No

No

No

No

torch

arctan

Yes

Yes

No

No

No

No

No

torch

arctanh

Yes

Yes

No

No

No

No

No

torch

argmax

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

argmin

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

asin

Yes

Yes

No

No

No

No

No

torch

asinh

Yes

Yes

No

No

No

No

No

torch

as_strided

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

atan

Yes

Yes

No

No

No

No

No

torch

atan2

Yes

Yes

Yes

No

No

No

No

torch

arctan2

Yes

Yes

No

No

No

No

No

torch

atanh

Yes

Yes

No

No

No

No

No

torch

baddbmm

Yes

Yes

Yes

No

No

No

No

torch

bernoulli

Yes

Yes

Yes

No

No

No

No

torch

bitwise_and

No

No

No

No

Yes

Yes

Yes

torch

bitwise_left_shift

No

No

No

No

Yes

Yes

Yes

torch

bitwise_not

No

No

No

No

No

No

Yes

torch

bitwise_or

No

No

No

No

Yes

Yes

Yes

torch

bitwise_right_shift

No

No

No

No

Yes

Yes

Yes

torch

bitwise_xor

No

No

No

No

Yes

Yes

Yes

torch

bmm

Yes

Yes

Yes

No

No

No

No

torch

broadcast_tensors

Yes

Yes

No

No

No

No

No

torch

cat

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

ceil

Yes

Yes

Yes

No

Yes

Yes

No

torch

chunk

Yes

Yes

No

No

Yes

No

No

torch

clamp

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

clip

Yes

Yes

No

No

Yes

No

No

torch

clone

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

conj

Yes

Yes

No

No

Yes

No

No

torch

copysign

Yes

Yes

No

No

No

No

No

torch

cos

Yes

Yes

Yes

No

No

No

No

torch

cosh

Yes

Yes

Yes

No

No

No

No

torch

count_nonzero

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

cross

Yes

Yes

No

No

Yes

No

No

torch

cumsum

Yes

Yes

Yes

Yes

Yes

No

No

torch

cumprod

Yes

Yes

No

No

Yes

No

No

torch

diag

Yes

Yes

Yes

No

No

No

No

torch

div

Yes

Yes

Yes

No

Yes

Yes

No

torch

divide

Yes

Yes

Yes

No

Yes

Yes

No

torch

dot

Yes

Yes

Yes

No

Yes

No

No

torch

embedding

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

empty

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

empty_like

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

empty_strided

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

eq

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

equal

Yes

Yes

No

Yes

Yes

Yes

Yes

torch

erf

Yes

Yes

No

No

No

No

No

torch

erfc

Yes

Yes

No

No

No

No

No

torch

erfinv

Yes

No

No

No

No

No

No

torch

exp

Yes

Yes

Yes

No

No

No

No

torch

exp2

Yes

Yes

No

No

No

No

No

torch

expm1

Yes

No

No

No

No

No

No

torch

eye

Yes

Yes

No

No

Yes

No

No

torch

fill

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

fill_

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

flatten

Yes

Yes

No

No

Yes

No

No

torch

flip

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

floor

Yes

Yes

Yes

No

Yes

Yes

No

torch

floor_divide

Yes

Yes

No

Yes

Yes

Yes

Yes

torch

fmax

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

fmin

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

fmod

Yes

Yes

No

No

Yes

No

No

torch

frac

Yes

Yes

No

No

No

No

No

torch

frexp

Yes

Yes

Yes

No

No

No

No

torch

full

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

full_like

Yes

Yes

Yes

Yes

Yes

Yes

No

torch

gather

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

ge

Yes

Yes

Yes

Yes

Yes

No

No

torch

greater

Yes

Yes

Yes

Yes

Yes

Yes

No

torch

greater_equal

Yes

Yes

Yes

Yes

Yes

Yes

No

torch

grid_sampler_2d

Yes

Yes

No

No

No

No

No

torch

gt

Yes

Yes

Yes

Yes

No

No

No

torch

heaviside

Yes

Yes

No

No

Yes

No

No

torch

hypot

Yes

Yes

No

No

No

No

No

torch

index_fill

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

index_select

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

is_complex

Yes

Yes

No

No

Yes

No

No

torch

is_floating_point

Yes

Yes

No

No

Yes

No

No

torch

is_nonzero

Yes

Yes

No

No

Yes

No

No

torch

isfinite

Yes

Yes

Yes

No

No

No

No

torch

isinf

Yes

Yes

Yes

No

No

No

No

torch

isnan

Yes

Yes

Yes

No

Yes

No

No

torch

isneginf

Yes

Yes

Yes

No

No

No

No

torch

isposinf

Yes

Yes

Yes

No

No

No

No

torch

kthvalue

Yes

Yes

Yes

No

Yes

No

No

torch

le

Yes

Yes

Yes

No

Yes

Yes

No

torch

lerp

Yes

Yes

No

No

Yes

No

No

torch

less

Yes

Yes

Yes

Yes

Yes

Yes

No

torch

less_equal

Yes

Yes

Yes

No

Yes

Yes

No

torch

linspace

Yes

Yes

Yes

No

Yes

No

No

torch

log

Yes

Yes

Yes

No

No

No

No

torch

log10

Yes

Yes

No

No

No

No

No

torch

log1p

Yes

Yes

No

No

No

No

No

torch

log2

Yes

Yes

Yes

No

No

No

No

torch

logaddexp

Yes

Yes

No

No

No

No

No

torch

logaddexp2

Yes

Yes

No

No

No

No

No

torch

logcumsumexp

Yes

Yes

No

No

No

No

No

torch

logical_and

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

logical_not

No

No

Yes

No

No

Yes

Yes

torch

logical_or

Yes

Yes

Yes

No

No

Yes

Yes

torch

logical_xor

Yes

Yes

Yes

No

No

Yes

Yes

torch

logit

Yes

Yes

No

No

No

No

No

torch

logspace

Yes

Yes

No

No

Yes

No

No

torch

logsumexp

Yes

Yes

No

No

No

No

No

torch

lt

Yes

Yes

Yes

Yes

Yes

Yes

No

torch

masked_fill

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

masked_scatter

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

masked_select

Yes

Yes

No

No

Yes

No

No

torch

matmul

Yes

Yes

No

No

No

No

No

torch

max

Yes

Yes

No

No

Yes

No

No

torch

maximum

Yes

Yes

Yes

No

Yes

No

No

torch

mean

Yes

Yes

Yes

No

No

No

No

torch

median

Yes

No

No

No

Yes

No

No

torch

meshgrid

Yes

Yes

No

No

No

No

No

torch

min

Yes

Yes

Yes

No

Yes

No

No

torch

minimum

Yes

Yes

Yes

No

Yes

No

No

torch

mm

Yes

Yes

Yes

No

No

No

No

torch

mul

Yes

Yes

Yes

No

No

No

No

torch

mv

Yes

Yes

Yes

No

Yes

No

No

torch

nan_to_num

Yes

Yes

No

No

Yes

No

No

torch

nansum

Yes

Yes

Yes

No

No

No

No

torch

narrow

Yes

No

No

No

No

No

No

torch

native_group_norm

Yes

Yes

No

No

No

No

No

torch

native_layer_norm

Yes

Yes

Yes

No

No

No

No

torch

ne

Yes

Yes

Yes

No

Yes

No

No

torch

neg

Yes

Yes

Yes

No

No

No

No

torch

nextafter

Yes

Yes

Yes

No

No

No

No

torch

nonzero

Yes

Yes

No

No

Yes

No

Yes

torch

norm

Yes

Yes

Yes

No

No

No

No

torch

normal

Yes

Yes

Yes

No

No

No

No

torch

not_equal

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

ones

Yes

Yes

No

No

Yes

Yes

Yes

torch

ones_like

Yes

Yes

No

No

Yes

Yes

Yes

torch

pixel_shuffle

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

poisson

Yes

Yes

No

No

No

No

No

torch

pow

Yes

Yes

Yes

No

No

No

No

torch

prod

Yes

Yes

Yes

Yes

No

No

No

torch

randperm

No

No

Yes

Yes

Yes

No

No

torch

reciprocal

Yes

Yes

Yes

No

No

No

No

torch

remainder

Yes

Yes

No

No

Yes

Yes

Yes

torch

reshape

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

resolve_conj

Yes

Yes

No

No

No

No

No

torch

resolve_neg

Yes

Yes

No

No

No

No

No

torch

result_type

Yes

Yes

No

No

Yes

No

No

torch

roll

Yes

Yes

No

No

Yes

Yes

Yes

torch

round

Yes

Yes

Yes

No

No

No

No

torch

rsqrt

Yes

Yes

Yes

No

No

No

No

torch

rsub

Yes

Yes

Yes

Yes

No

No

No

torch

scalar_tensor

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

scatter

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

scatter_add

Yes

Yes

Yes

No

No

No

No

torch

scatter_reduce

Yes

Yes

Yes

No

No

No

No

torch

searchsorted

Yes

Yes

Yes

Yes

Yes

No

No

torch

select

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

sgn

Yes

Yes

Yes

No

No

No

No

torch

sigmoid

Yes

Yes

Yes

No

No

No

No

torch

sign

Yes

Yes

Yes

No

No

No

No

torch

signbit

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

sin

Yes

Yes

Yes

No

No

No

No

torch

sinc

Yes

Yes

No

No

No

No

No

torch

sinh

Yes

Yes

No

No

No

No

No

torch

sort

Yes

Yes

Yes

No

No

No

No

torch

split_with_sizes

Yes

Yes

No

No

Yes

No

No

torch

sqrt

Yes

Yes

Yes

No

No

No

No

torch

square

Yes

Yes

No

No

Yes

No

No

torch

squeeze

Yes

Yes

No

No

Yes

Yes

Yes

torch

stack

Yes

Yes

No

No

Yes

No

No

torch

std

Yes

Yes

No

No

No

No

No

torch

std_mean

Yes

Yes

No

No

No

No

No

torch

sub

Yes

Yes

Yes

No

No

No

No

torch

sum

Yes

Yes

Yes

No

Yes

No

Yes

torch

t

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

take

Yes

Yes

No

No

Yes

No

No

torch

tan

Yes

Yes

No

No

No

No

No

torch

tanh

Yes

Yes

Yes

No

No

No

No

torch

topk

Yes

Yes

Yes

No

No

No

No

torch

trace

Yes

Yes

No

No

Yes

No

No

torch

transpose

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch

tril

Yes

Yes

No

No

No

Yes

Yes

torch

triu

Yes

Yes

No

No

No

Yes

Yes

torch

trunc

Yes

Yes

Yes

No

Yes

Yes

No

torch

unbind

Yes

Yes

No

No

Yes

No

No

torch

unique

Yes

No

No

No

Yes

No

No

torch

_unique

Yes

No

No

No

Yes

No

No

torch

_unique2

Yes

No

No

No

Yes

No

No

torch

unsqueeze

Yes

Yes

No

No

Yes

No

No

torch

var

Yes

Yes

No

No

No

No

No

torch

var_mean

Yes

Yes

No

No

No

No

No

torch

vdot

Yes

Yes

No

No

No

No

No

torch

where

Yes

Yes

Yes

Yes

Yes

No

No

torch

xlogy

Yes

Yes

Yes

No

No

No

No

torch

zero_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

zeros

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

zeros_like

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

_adaptive_avg_pool2d

Yes

Yes

Yes

No

No

No

No

torch

_adaptive_avg_pool3d

Yes

Yes

Yes

No

No

No

No

torch

_efficientzerotensor

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

_fused_dropout

Yes

Yes

Yes

No

No

No

No

torch

_masked_scale

Yes

Yes

Yes

No

Yes

Yes

Yes

torch

_native_batch_norm_legit

Yes

Yes

Yes

No

No

No

No

torch

_native_batch_norm_legit_no_training

Yes

Yes

Yes

No

No

No

No

torch

_resize_output_

Yes

Yes

No

Yes

Yes

Yes

Yes

torch

_weight_norm_interface

Yes

Yes

Yes

No

No

No

No

torch

AdaptiveAvgPool1d

Yes

Yes

Yes

No

No

No

No

torch.nn

AdaptiveAvgPool2d

Yes

Yes

Yes

No

No

No

No

torch.nn

AdaptiveAvgPool3d

Yes

Yes

Yes

No

No

No

No

torch.nn

AvgPool1d

Yes

Yes

Yes

No

No

No

No

torch.nn

AvgPool2d

Yes

Yes

Yes

No

No

No

No

torch.nn

AvgPool3d

Yes

Yes

Yes

No

No

No

No

torch.nn

BatchNorm1d

Yes

Yes

Yes

No

No

No

No

torch.nn

BatchNorm2d

Yes

Yes

Yes

No

No

No

No

torch.nn

BCELoss

Yes

Yes

Yes

No

No

No

No

torch.nn

BCEWithLogitsLoss

Yes

Yes

Yes

No

No

No

No

torch.nn

ChannelShuffle

Yes

Yes

No

No

Yes

No

No

torch.nn

ConstantPad1d

Yes

Yes

Yes

No

No

No

No

torch.nn

Conv1d

Yes

Yes

Yes

No

No

No

No

torch.nn

Conv2d

Yes

Yes

Yes

No

No

No

No

torch.nn

Conv3d

Yes

Yes

Yes

No

No

No

No

torch.nn

ConvTranspose2d

Yes

Yes

Yes

No

No

No

No

torch.nn

ConvTranspose3d

Yes

Yes

Yes

No

No

No

No

torch.nn

CrossEntropyLoss

Yes

Yes

No

No

No

No

No

torch.nn

Dropout

Yes

Yes

Yes

No

No

No

No

torch.nn

ELU

Yes

Yes

Yes

No

No

No

No

torch.nn

Embedding

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.nn

EmbeddingBag

Yes

Yes

Yes

No

No

No

No

torch.nn

GELU

Yes

Yes

Yes

No

No

No

No

torch.nn

GLU

Yes

No

No

No

No

No

No

torch.nn

Hardshrink

Yes

Yes

No

No

No

No

No

torch.nn

Hardsigmoid

Yes

Yes

Yes

No

No

No

No

torch.nn

Hardtanh

Yes

No

No

No

No

No

No

torch.nn

HuberLoss

Yes

Yes

No

No

No

No

No

torch.nn

InstanceNorm2d

Yes

Yes

No

No

No

No

No

torch.nn

KLDivLoss

Yes

Yes

No

No

No

No

No

torch.nn

LayerNorm

Yes

Yes

Yes

No

No

No

No

torch.nn

LeakyReLU

Yes

Yes

Yes

No

No

No

No

torch.nn

Linear

Yes

Yes

Yes

No

No

No

No

torch.nn

LogSigmoid

Yes

No

No

No

No

No

No

torch.nn

LogSoftmax

Yes

Yes

Yes

No

No

No

No

torch.nn

MaxPool2d

Yes

Yes

Yes

No

No

No

No

torch.nn

MaxPool3d

Yes

Yes

No

No

No

No

No

torch.nn

Mish

Yes

Yes

No

No

No

No

No

torch.nn

MSELoss

Yes

Yes

Yes

No

No

No

No

torch.nn

NLLLoss

Yes

Yes

Yes

No

No

No

No

torch.nn

PixelShuffle

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.nn

PReLU

Yes

Yes

No

No

No

No

No

torch.nn

ReLU

Yes

Yes

Yes

No

No

No

No

torch.nn

ReflectionPad1d

Yes

No

No

No

No

No

No

torch.nn

ReflectionPad2d

Yes

No

No

No

No

No

No

torch.nn

ReflectionPad3d

Yes

No

No

No

No

No

No

torch.nn

ReplicationPad1d

Yes

Yes

No

No

Yes

No

No

torch.nn

ReplicationPad2d

Yes

Yes

No

No

Yes

No

No

torch.nn

ReplicationPad3d

Yes

Yes

No

No

Yes

No

No

torch.nn

RReLU

Yes

Yes

No

No

No

No

No

torch.nn

SiLU

Yes

Yes

No

No

No

No

No

torch.nn

Softmax

Yes

Yes

Yes

No

No

No

No

torch.nn

Softplus

Yes

Yes

No

No

No

No

No

torch.nn

Softshrink

Yes

Yes

No

No

No

No

No

torch.nn

Threshold

Yes

Yes

No

No

No

No

No

torch.nn

SmoothL1Loss

Yes

Yes

No

No

No

No

No

torch.nn

Upsample

Yes

Yes

Yes

No

No

No

No

torch.nn

UpsamplingNearest2d

Yes

Yes

Yes

No

No

No

No

torch.nn

weight_norm

Yes

Yes

No

No

No

No

No

torch.nn.utils

__and__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__iand__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__ilshift__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__ior__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__irshift__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__ixor__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__lshift__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__or__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__rshift__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

__xor__

No

No

No

No

Yes

Yes

Yes

torch.Tensor

abs

Yes

Yes

Yes

No

No

No

No

torch.Tensor

acos

Yes

Yes

No

No

No

No

No

torch.Tensor

acos_

Yes

Yes

No

No

No

No

No

torch.Tensor

acosh

Yes

Yes

No

No

No

No

No

torch.Tensor

acosh_

Yes

Yes

No

No

No

No

No

torch.Tensor

add

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

add_

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

addbmm

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addbmm_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addcdiv

Yes

Yes

No

No

No

No

No

torch.Tensor

addcdiv_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addcmul

Yes

Yes

No

No

No

No

No

torch.Tensor

addcmul_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addmm

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addmm_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addmv

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addmv_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addr

Yes

Yes

Yes

No

No

No

No

torch.Tensor

addr_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

all

Yes

Yes

No

No

Yes

No

Yes

torch.Tensor

amax

Yes

Yes

Yes

No

No

No

No

torch.Tensor

amin

Yes

Yes

Yes

No

No

No

No

torch.Tensor

aminmax

Yes

Yes

No

Yes

Yes

Yes

No

torch.Tensor

any

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

arccos

Yes

Yes

No

No

No

No

No

torch.Tensor

arccos_

Yes

Yes

No

No

No

No

No

torch.Tensor

arccosh

Yes

Yes

No

No

No

No

No

torch.Tensor

arccosh_

Yes

Yes

No

No

No

No

No

torch.Tensor

arcsin

Yes

Yes

No

No

No

No

No

torch.Tensor

arcsin_

Yes

Yes

No

No

No

No

No

torch.Tensor

arcsinh

Yes

Yes

No

No

No

No

No

torch.Tensor

arcsinh_

Yes

Yes

No

No

No

No

No

torch.Tensor

arctan

Yes

Yes

No

No

No

No

No

torch.Tensor

arctan_

Yes

Yes

No

No

No

No

No

torch.Tensor

arctanh

Yes

Yes

No

No

No

No

No

torch.Tensor

arctanh_

Yes

Yes

No

No

No

No

No

torch.Tensor

argmax

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

argmin

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

asin

Yes

Yes

No

No

No

No

No

torch.Tensor

asin_

Yes

Yes

No

No

No

No

No

torch.Tensor

asinh

Yes

Yes

No

No

No

No

No

torch.Tensor

asinh_

Yes

Yes

No

No

No

No

No

torch.Tensor

atan2

Yes

Yes

No

No

No

No

No

torch.Tensor

atan2_

Yes

Yes

No

No

No

No

No

torch.Tensor

arctan2

Yes

Yes

No

No

No

No

No

torch.Tensor

arctan2_

Yes

Yes

No

No

No

No

No

torch.Tensor

atanh

Yes

Yes

No

No

No

No

No

torch.Tensor

atanh_

Yes

Yes

No

No

No

No

No

torch.Tensor

baddbmm

Yes

Yes

Yes

No

No

No

No

torch.Tensor

baddbmm_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

bernoulli

Yes

Yes

Yes

No

No

No

No

torch.Tensor

bitwise_and

No

No

No

No

No

No

Yes

torch.Tensor

bitwise_left_shift

No

No

No

No

Yes

Yes

No

torch.Tensor

bitwise_not

No

No

No

No

Yes

Yes

Yes

torch.Tensor

bitwise_or

No

No

No

No

No

No

Yes

torch.Tensor

bitwise_right_shift

No

No

No

No

Yes

Yes

No

torch.Tensor

bitwise_xor

No

No

No

No

No

No

Yes

torch.Tensor

bmm

Yes

Yes

Yes

No

No

No

No

torch.Tensor

ceil

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

clamp

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_max

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clamp_min

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

clip

Yes

Yes

No

No

Yes

No

No

torch.Tensor

clip_

Yes

Yes

No

No

Yes

No

No

torch.Tensor

clone

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

conj

Yes

Yes

No

No

Yes

No

No

torch.Tensor

copy_

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

copysign

Yes

Yes

No

No

No

No

No

torch.Tensor

copysign_

Yes

Yes

No

No

No

No

No

torch.Tensor

cos

Yes

Yes

Yes

No

No

No

No

torch.Tensor

cos_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

cosh_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

count_nonzero

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

cross

Yes

Yes

No

No

Yes

No

No

torch.Tensor

cumprod

Yes

Yes

No

No

Yes

No

No

torch.Tensor

cumsum

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

diag

Yes

Yes

Yes

No

No

No

No

torch.Tensor

div

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

div_

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

dot

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

fill_

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

eq

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

eq_

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

equal

Yes

Yes

No

Yes

Yes

Yes

No

torch.Tensor

erf

Yes

Yes

No

No

No

No

No

torch.Tensor

erf_

Yes

Yes

No

No

No

No

No

torch.Tensor

erfinv

Yes

No

No

No

No

No

No

torch.Tensor

erfinv_

Yes

No

No

No

No

No

No

torch.Tensor

erfc_

Yes

Yes

No

No

No

No

No

torch.Tensor

erfc

Yes

Yes

No

No

No

No

No

torch.Tensor

exp

Yes

Yes

Yes

No

No

No

No

torch.Tensor

exp_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

expand

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

expand_as

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

expm1

Yes

No

No

No

No

No

No

torch.Tensor

expm1_

Yes

No

No

No

No

No

No

torch.Tensor

exponential_

Yes

Yes

No

No

No

No

No

torch.Tensor

flatten

Yes

Yes

No

No

Yes

No

No

torch.Tensor

flip

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

floor

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

floor_

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

floor_divide

Yes

Yes

No

Yes

Yes

Yes

Yes

torch.Tensor

floor_divide_

Yes

Yes

No

Yes

Yes

Yes

Yes

torch.Tensor

fmax

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

fmin

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

fmod

Yes

Yes

No

No

Yes

No

No

torch.Tensor

fmod_

Yes

Yes

No

No

Yes

No

No

torch.Tensor

frac

Yes

Yes

No

No

No

No

No

torch.Tensor

frexp

Yes

Yes

Yes

No

No

No

No

torch.Tensor

gather

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

ge

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

ge_

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

geometric_

Yes

Yes

No

No

No

No

No

torch.Tensor

greater

Yes

Yes

Yes

Yes

Yes

Yes

No

torch.Tensor

greater_equal

Yes

Yes

Yes

Yes

Yes

Yes

No

torch.Tensor

gt

Yes

Yes

Yes

Yes

Yes

Yes

No

torch.Tensor

hardshrink

Yes

Yes

No

No

No

No

No

torch.Tensor

heaviside

Yes

Yes

No

No

Yes

No

No

torch.Tensor

hypot

Yes

Yes

No

No

No

No

No

torch.Tensor

index_add_

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

index_copy_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

index_fill

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

index_fill_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

index_put

Yes

Yes

No

No

Yes

No

No

torch.Tensor

index_put_

Yes

Yes

No

No

Yes

No

No

torch.Tensor

index_select

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

is_complex

Yes

Yes

No

No

Yes

No

No

torch.Tensor

is_floating_point

Yes

Yes

No

No

Yes

No

No

torch.Tensor

isfinite

Yes

Yes

Yes

No

No

No

No

torch.Tensor

isinf

Yes

Yes

Yes

No

No

No

No

torch.Tensor

isnan

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

isneginf

Yes

Yes

Yes

No

No

No

No

torch.Tensor

isposinf

Yes

Yes

Yes

No

No

No

No

torch.Tensor

item

Yes

No

No

No

No

No

No

torch.Tensor

kthvalue

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

le

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

le_

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

lerp

Yes

Yes

No

No

Yes

No

No

torch.Tensor

lerp_

Yes

Yes

No

No

Yes

No

No

torch.Tensor

less

Yes

Yes

Yes

Yes

Yes

Yes

No

torch.Tensor

less_

Yes

Yes

Yes

Yes

Yes

Yes

No

torch.Tensor

less_equal

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

less_equal_

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

log

Yes

Yes

Yes

No

No

No

No

torch.Tensor

log_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

log_normal_

Yes

Yes

No

No

No

No

No

torch.Tensor

log_softmax

Yes

Yes

Yes

No

No

No

No

torch.Tensor

log10

Yes

Yes

No

No

No

No

No

torch.Tensor

log1p

Yes

Yes

No

No

No

No

No

torch.Tensor

log1p_

Yes

Yes

No

No

No

No

No

torch.Tensor

log2

Yes

Yes

Yes

No

No

No

No

torch.Tensor

log2_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

logaddexp

Yes

Yes

No

No

No

No

No

torch.Tensor

logaddexp2

Yes

Yes

No

No

No

No

No

torch.Tensor

logcumsumexp

Yes

Yes

No

No

No

No

No

torch.Tensor

logical_and

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

logical_and_

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

logical_not

No

No

Yes

No

No

Yes

Yes

torch.Tensor

logical_not_

No

No

Yes

No

No

Yes

Yes

torch.Tensor

logical_or

Yes

Yes

Yes

No

No

Yes

Yes

torch.Tensor

logical_or_

Yes

Yes

Yes

No

No

Yes

Yes

torch.Tensor

logical_xor_

Yes

Yes

Yes

No

No

Yes

Yes

torch.Tensor

logical_xor_

Yes

Yes

Yes

No

No

Yes

Yes

torch.Tensor

logit

Yes

Yes

No

No

No

No

No

torch.Tensor

logsumexp

Yes

Yes

No

No

No

No

No

torch.Tensor

lt

Yes

Yes

Yes

Yes

Yes

Yes

No

torch.Tensor

masked_fill

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

masked_fill_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

masked_scatter

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

masked_scatter_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

masked_select

Yes

Yes

No

No

Yes

No

No

torch.Tensor

matmul

Yes

Yes

No

No

No

No

No

torch.Tensor

max

Yes

Yes

No

No

Yes

No

No

torch.Tensor

maximum

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

mean

Yes

Yes

Yes

No

No

No

No

torch.Tensor

median

Yes

No

No

No

Yes

No

No

torch.Tensor

min

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

minimum

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

mm

Yes

Yes

Yes

No

No

No

No

torch.Tensor

mul

Yes

Yes

Yes

No

No

No

No

torch.Tensor

mul_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

mv

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

nan_to_num

Yes

Yes

No

No

Yes

No

No

torch.Tensor

nansum

Yes

Yes

Yes

No

No

No

No

torch.Tensor

narrow

Yes

No

No

No

No

No

No

torch.Tensor

ne

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

ne_

Yes

Yes

Yes

No

Yes

No

No

torch.Tensor

neg

Yes

Yes

Yes

No

No

No

No

torch.Tensor

new_empty

Yes

Yes

No

No

Yes

Yes

No

torch.Tensor

new_empty_strided

Yes

Yes

No

No

Yes

Yes

No

torch.Tensor

new_full

Yes

Yes

No

No

Yes

Yes

No

torch.Tensor

new_ones

Yes

Yes

No

No

Yes

Yes

No

torch.Tensor

new_zeros

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

nextafter

Yes

Yes

Yes

No

No

No

No

torch.Tensor

nonzero

Yes

Yes

No

No

Yes

No

Yes

torch.Tensor

norm

Yes

Yes

Yes

No

No

No

No

torch.Tensor

normal_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

permute

Yes

Yes

No

No

Yes

No

No

torch.Tensor

pin_memory

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

pow

Yes

Yes

Yes

No

No

No

No

torch.Tensor

pow_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

prod

Yes

Yes

Yes

Yes

No

No

No

torch.Tensor

put_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

random_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

reciprocal

Yes

Yes

Yes

No

No

No

No

torch.Tensor

reciprocal_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

remainder

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

remainder_

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

repeat

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

reshape

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

roll

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

resize_

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

round

Yes

Yes

Yes

No

No

No

No

torch.Tensor

round_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

rsqrt

Yes

Yes

Yes

No

No

No

No

torch.Tensor

rsqrt_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

scatter

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

scatter_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

scatter_add

Yes

Yes

Yes

No

No

No

No

torch.Tensor

scatter_add_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

scatter_reduce

Yes

Yes

Yes

No

No

No

No

torch.Tensor

scatter_reduce_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

select

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

sgn

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sgn_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sigmoid

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sigmoid_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sign

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sign_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

signbit

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

sin

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sin_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sinh

Yes

Yes

No

No

No

No

No

torch.Tensor

sinh_

Yes

Yes

No

No

No

No

No

torch.Tensor

sinc

Yes

Yes

No

No

No

No

No

torch.Tensor

sort

Yes

Yes

Yes

No

No

No

No

torch.Tensor

split_with_sizes

Yes

Yes

No

No

Yes

No

No

torch.Tensor

sqrt

Yes

Yes

Yes

No

No

No

No

torch.Tensor

square

Yes

Yes

No

No

No

No

No

torch.Tensor

square_

Yes

Yes

No

No

No

No

No

torch.Tensor

squeeze

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

squeeze_

Yes

Yes

No

No

Yes

Yes

Yes

torch.Tensor

std

Yes

Yes

No

No

No

No

No

torch.Tensor

sub

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sub_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

sum

Yes

Yes

Yes

No

No

No

No

torch.Tensor

T

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

t

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

tan

Yes

Yes

No

No

No

No

No

torch.Tensor

tan_

Yes

Yes

No

No

No

No

No

torch.Tensor

tanh

Yes

Yes

Yes

No

No

No

No

torch.Tensor

tanh_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

to

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

topk

Yes

Yes

Yes

No

No

No

No

torch.Tensor

trace

Yes

Yes

No

No

Yes

No

No

torch.Tensor

take

Yes

Yes

No

No

Yes

No

No

torch.Tensor

transpose

Yes

Yes

Yes

Yes

Yes

Yes

Yes

torch.Tensor

tril

Yes

Yes

No

No

No

Yes

Yes

torch.Tensor

tril_

Yes

Yes

No

No

No

Yes

Yes

torch.Tensor

triu

Yes

Yes

No

No

No

Yes

Yes

torch.Tensor

triu_

Yes

Yes

No

No

No

Yes

Yes

torch.Tensor

trunc

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

trunc_

Yes

Yes

Yes

No

Yes

Yes

No

torch.Tensor

unbind

Yes

Yes

No

No

Yes

No

No

torch.Tensor

unsqueeze

Yes

Yes

No

No

Yes

No

No

torch.Tensor

uniform_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

unique

Yes

No

No

No

Yes

No

No

torch.Tensor

var

Yes

Yes

No

No

No

No

No

torch.Tensor

vdot

Yes

Yes

No

No

No

No

No

torch.Tensor

view

Yes

No

No

No

No

No

No

torch.Tensor

where

Yes

Yes

Yes

Yes

Yes

No

No

torch.Tensor

xlogy

Yes

Yes

Yes

No

No

No

No

torch.Tensor

xlogy_

Yes

Yes

Yes

No

No

No

No

torch.Tensor

zero_

Yes

Yes

Yes

No

Yes

Yes

Yes

torch.Tensor

entr

Yes

No

No

No

No

No

No

torch.special

erf

Yes

Yes

No

No

No

No

No

torch.special

erfc

Yes

Yes

No

No

No

No

No

torch.special

exp2

Yes

Yes

No

No

No

No

No

torch.special

expit

Yes

Yes

No

No

No

No

No

torch.special

expm1

Yes

No

No

No

No

No

No

torch.special

erfinv

Yes

No

No

No

No

No

No

torch.special

log1p

Yes

Yes

No

No

No

No

No

torch.special

logit

Yes

Yes

No

No

No

No

No

torch.special

logsumexp

Yes

Yes

No

No

No

No

No

torch.special

round

Yes

Yes

Yes

No

No

No

No

torch.special

sinc

Yes

Yes

No

No

No

No

No

torch.special

softmax

Yes

Yes

Yes

No

No

No

No

torch.special

xlog1py

Yes

Yes

No

No

No

No

No

torch.special

xlogy

Yes

Yes

Yes

No

No

No

No

torch.special

batched_nms

Yes

Yes

No

No

No

No

No

torchvision.ops

nms

Yes

Yes

No

No

No

No

No

torchvision.ops

roi_align

Yes

No

No

No

No

No

No

torchvision.ops