Op-db Coverage per Op and Type#

This page shows, for every op collected from torch.testing._internal.common_methods_invocations (op_db), which data types are covered by the op-db export tests in:

  • unittests.torch.coverage.test_onnx_export_common_methods

  • unittests.torch.coverage.test_onnx_export_common_methods_tracing

  • unittests.torch.coverage.test_onnx_export_common_methods_new_tracing

Legend:

  • ✔ — converter exists and the test passes for that dtype.

  • ⚠ xfail — converter exists but the test is a known failure for that dtype (incorrect numerical results, unsupported dtype mapping, etc.).

  • ✘ no converter — no ONNX converter has been implemented for this op yet.

  • — — the op does not support that dtype at all.

The exclusion sets (NO_CONVERTER_OPS, XFAIL_OPS, and the per-dtype xfail sets) are defined in yobx.torch.coverage.op_coverage and are imported by the test module so that both sources always stay in sync.

Default export path#

Default export path

Op

float32

float16

bfloat16

int32

int64

H

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

T

✔

✔

✔

✔

✔

__getitem__

✔

✔

✔

✔

✔

__radd__

✔

✔

✔

✔

✔

__rand__

—

—

—

✔

✔

__rdiv__

✔

✔

✔

⚠ xfail

⚠ xfail

__rmatmul__

✔

✔

✔

✔

✔

__rmod__

✔

✔

✔

✔

✔

__rmul__

✔

✔

✔

✔

✔

__ror__

—

—

—

✔

✔

__rpow__

✔

✔

✔

⚠ xfail

⚠ xfail

__rsub__

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

__rxor__

—

—

—

⚠ xfail

⚠ xfail

_batch_norm_with_update

✔

✔

✔

—

—

_chunk_cat

✔

✔

✔

✔

✔

_native_batch_norm_legit

✔

✔

✔

—

—

_softmax_backward_data

✔

✔

✔

—

—

_unsafe_masked_index

✔

✔

✔

✔

✔

_unsafe_masked_index_put_accumulate

✔

✔

✔

✔

✔

_upsample_bilinear2d_aa

✔

—

—

—

—

abs

✔

✔

✔

✔

✔

acos

✔

✔

✔

⚠ xfail

⚠ xfail

acosh

✔

✔

✔

⚠ xfail

⚠ xfail

add

✔

✔

✔

✔

✔

addbmm

✔

✔

✔

✔

✔

addcdiv

✘ no converter

✘ no converter

✘ no converter

—

—

addcmul

✔

⚠ xfail

⚠ xfail

✔

✔

addmm

✔

✔

✔

✔

✔

addmv

✔

✔

✔

✔

✔

addr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

alias_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

all

✔

✔

✔

✔

✔

allclose

✔

✔

✔

—

—

amax

✔

✔

✔

✔

✔

amin

✔

✔

✔

✔

✔

aminmax

✔

✔

✔

✔

✔

angle

✔

✔

✔

⚠ xfail

⚠ xfail

any

✔

✔

✔

✔

✔

arange

✔

✔

✔

✔

✔

argmax

✔

✔

✔

✔

✔

argmin

✔

✔

✔

✔

✔

argsort

✔

⚠ xfail

⚠ xfail

✔

✔

argwhere

✔

✔

✔

✔

✔

as_strided

✔

✔

✔

✔

✔

as_strided_copy

✔

✔

✔

✔

✔

as_strided_scatter

✔

✔

✔

✔

✔

asin

✔

✔

✔

⚠ xfail

⚠ xfail

asinh

✔

✔

✔

⚠ xfail

⚠ xfail

atan

✔

✔

✔

⚠ xfail

⚠ xfail

atan2

✔

✔

✔

⚠ xfail

⚠ xfail

atanh

✔

✔

✔

⚠ xfail

⚠ xfail

atleast_1d

✔

✔

✔

✔

✔

atleast_2d

✔

✔

✔

✔

✔

atleast_3d

✔

✔

✔

✔

✔

baddbmm

✔

✔

✔

✔

✔

bernoulli

✘ no converter

✘ no converter

✘ no converter

—

—

bfloat16

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

bincount

—

—

—

✔

✔

bitwise_and

—

—

—

✔

✔

bitwise_left_shift

—

—

—

✔

✔

bitwise_not

—

—

—

✔

✔

bitwise_or

—

—

—

✔

✔

bitwise_right_shift

—

—

—

✔

✔

bitwise_xor

—

—

—

✔

✔

block_diag

✔

✔

✔

✔

✔

bmm

✔

✔

⚠ xfail

✔

✔

bool

✔

✔

✔

✔

✔

broadcast_shapes

✔

—

—

—

—

broadcast_tensors

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

broadcast_to

✔

✔

✔

✔

✔

bucketize

✔

✔

✔

✔

✔

byte

✔

✔

✔

✔

✔

cartesian_prod

✔

✔

✔

✔

✔

cat

✔

✔

✔

✔

✔

cdist

✔

—

—

—

—

cdouble

✔

✔

✔

✔

✔

ceil

✔

✔

✔

⚠ xfail

⚠ xfail

cfloat

✔

✔

✔

✔

✔

chalf

✔

✔

✔

✔

✔

char

✔

✔

✔

✔

✔

cholesky_inverse

⚠ xfail

—

—

—

—

cholesky_solve

⚠ xfail

—

—

—

—

chunk

✔

✔

✔

✔

✔

clamp

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

clamp_max

✔

✔

✔

✔

✔

clamp_min

✔

✔

✔

✔

✔

clone

✔

✔

✔

✔

✔

column_stack

✔

✔

✔

✔

✔

combinations

✔

✔

✔

✔

✔

complex

✔

✔

—

—

—

conj

✔

✔

✔

✔

✔

conj_physical

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

constant_pad_nd

✔

✔

✔

✔

✔

contiguous

✔

✔

✔

✔

✔

copysign

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

corrcoef

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

cos

✔

✔

✔

⚠ xfail

⚠ xfail

cosh

✔

⚠ xfail

✔

⚠ xfail

⚠ xfail

count_nonzero

✔

✔

✔

✔

✔

cov

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

cross

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

cummax

✔

✔

✔

✔

✔

cummin

✔

✔

✔

✔

✔

cumprod

✔

✔

✔

✔

✔

cumsum

✔

✔

✔

✔

✔

cumulative_trapezoid

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

deg2rad

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diag

✔

✔

✔

✔

✔

diag_embed

✔

✔

✔

✔

✔

diagflat

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal_scatter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diff

✔

✔

✔

✔

✔

digamma

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

dist

✔

✔

✔

—

—

dot

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

double

✔

✔

✔

✔

✔

dsplit

✔

✔

✔

✔

✔

dstack

✔

✔

✔

✔

✔

einsum

✔

✔

✔

✔

✔

empty_permuted

✔

✔

✔

✔

✔

empty_strided

✔

✔

✔

✔

✔

eq

✔

✔

✔

✔

✔

equal

✔

✔

✔

✔

✔

erf

✔

✔

✔

⚠ xfail

⚠ xfail

erfc

✔

✔

✔

⚠ xfail

⚠ xfail

erfinv

✔

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

exp

✔

✔

✔

⚠ xfail

⚠ xfail

exp2

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

expand

✔

✔

✔

✔

✔

expand_as

✔

✔

✔

✔

✔

expand_copy

✔

✔

✔

✔

✔

expm1

✔

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

eye

✔

✔

✔

✔

✔

fft.fft

✔

—

—

✔

✔

fft.fft2

✔

—

—

✔

✔

fft.fftn

✔

—

—

✔

✔

fft.fftshift

✔

✔

✔

✔

✔

fft.hfft

✔

—

—

✔

✔

fft.hfft2

✔

—

—

✔

✔

fft.hfftn

✔

—

—

✔

✔

fft.ifft

✔

—

—

✔

✔

fft.ifft2

✔

—

—

✔

✔

fft.ifftn

✔

—

—

✔

✔

fft.ifftshift

✔

✔

✔

✔

✔

fft.ihfft

✔

—

—

✔

✔

fft.ihfft2

✔

—

—

✔

✔

fft.ihfftn

✔

—

—

✔

✔

fft.irfft

✔

—

—

✔

✔

fft.irfft2

✔

—

—

✔

✔

fft.irfftn

✔

—

—

✔

✔

fft.rfft

✔

—

—

✔

✔

fft.rfft2

✔

—

—

✔

✔

fft.rfftn

✔

—

—

✔

✔

fill

✔

✔

✔

✔

✔

flatten

✔

✔

✔

✔

✔

flip

✔

✔

✔

✔

✔

fliplr

✔

✔

✔

✔

✔

flipud

✔

✔

✔

✔

✔

float

✔

✔

✔

✔

✔

float_power

✔

✔

✔

✔

✔

floor

✔

✔

✔

⚠ xfail

⚠ xfail

floor_divide

✔

✔

✔

⚠ xfail

⚠ xfail

fmax

✔

✔

✔

✔

✔

fmin

✔

✔

✔

✔

✔

fmod

✔

✔

✔

✔

✔

frac

✔

✔

✔

—

—

frexp

✔

✔

⚠ xfail

—

—

full

✔

✔

✔

✔

✔

full_like

✔

✔

✔

✔

✔

gather

✔

✔

✔

✔

✔

gcd

—

—

—

⚠ xfail

⚠ xfail

ge

✔

✔

✔

✔

✔

geqrf

✔

—

—

—

—

gradient

✔

✔

✔

✔

✔

grid_sampler_2d

✔

✔

✔

—

—

grid_sampler_3d

✔

✔

✔

—

—

gt

✔

✔

✔

✔

✔

half

✔

✔

✔

✔

✔

hash_tensor

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

heaviside

✔

✔

✔

✔

✔

histc

✔

✔

✔

—

—

histogram

✔

—

—

—

—

histogramdd

✔

—

—

—

—

hsplit

✔

✔

✔

✔

✔

hstack

✔

✔

✔

✔

✔

hypot

✘ no converter

✘ no converter

✘ no converter

—

—

i0

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

igamma

✘ no converter

✘ no converter

✘ no converter

—

—

igammac

✘ no converter

✘ no converter

✘ no converter

—

—

imag

—

—

—

—

—

index_add

✔

✔

✔

✔

✔

index_copy

✔

✔

✔

✔

✔

index_fill

✔

✔

✔

✔

✔

index_put

✔

✔

✔

✔

✔

index_select

✔

✔

✔

✔

✔

inner

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

int

✔

✔

✔

✔

✔

isclose

✔

✔

✔

✔

✔

isfinite

✔

✔

✔

✔

✔

isin

✔

✔

✔

✔

✔

isinf

✔

✔

✔

⚠ xfail

⚠ xfail

isnan

✔

✔

✔

⚠ xfail

⚠ xfail

isneginf

✔

✔

✔

✔

✔

isposinf

✔

✔

✔

✔

✔

isreal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

istft

—

—

—

—

—

item

✔

✔

✔

✔

✔

jiterator_2inputs_2outputs

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_4inputs_with_extra_args

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary_return_by_ref

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_unary

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

kron

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

kthvalue

✔

✔

✔

✔

✔

lcm

—

—

—

⚠ xfail

⚠ xfail

ldexp

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

le

✔

✔

✔

✔

✔

lerp

✔

✔

✔

—

—

lgamma

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

linalg.cholesky

✔

—

—

—

—

linalg.cholesky_ex

✔

—

—

—

—

linalg.cond

✘ no converter

—

—

—

—

linalg.cross

✔

✔

✔

✔

✔

linalg.det

✔

—

—

—

—

linalg.diagonal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

linalg.eig

✔

—

—

—

—

linalg.eigh

✔

—

—

—

—

linalg.eigvals

✔

—

—

—

—

linalg.eigvalsh

✔

—

—

—

—

linalg.householder_product

✘ no converter

—

—

—

—

linalg.inv

✘ no converter

—

—

—

—

linalg.inv_ex

✘ no converter

—

—

—

—

linalg.ldl_factor

✔

—

—

—

—

linalg.ldl_factor_ex

✔

—

—

—

—

linalg.ldl_solve

✔

—

—

—

—

linalg.lstsq

✔

—

—

—

—

linalg.lu

✔

—

—

—

—

linalg.lu_factor

✔

—

—

—

—

linalg.lu_factor_ex

✔

—

—

—

—

linalg.lu_solve

✔

—

—

—

—

linalg.matrix_norm

✔

✔

✔

—

—

linalg.matrix_power

✔

—

—

—

—

linalg.matrix_rank

✔

—

—

—

—

linalg.matrix_sqrth

✔

—

—

—

—

linalg.multi_dot

✔

✔

✔

✔

✔

linalg.norm

✔

✔

✔

—

—

linalg.pinv

✔

—

—

—

—

linalg.polar

✔

—

—

—

—

linalg.qr

✘ no converter

—

—

—

—

linalg.slogdet

✔

—

—

—

—

linalg.solve

✘ no converter

—

—

—

—

linalg.solve_ex

✘ no converter

—

—

—

—

linalg.solve_triangular

✔

—

—

—

—

linalg.svd

✔

—

—

—

—

linalg.svdvals

✘ no converter

—

—

—

—

linalg.tensorinv

✔

—

—

—

—

linalg.tensorsolve

✔

—

—

—

—

linalg.vander

✔

—

—

✔

✔

linalg.vecdot

✔

✔

✔

—

—

linalg.vector_norm

✔

✔

✔

—

—

linspace

✔

✔

✔

✔

✔

log

✔

✔

✔

⚠ xfail

⚠ xfail

log10

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log1p

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log2

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log_softmax

✔

✔

✔

—

—

logaddexp

✔

✔

⚠ xfail

—

—

logaddexp2

✔

✔

⚠ xfail

—

—

logcumsumexp

✔

✔

✔

—

—

logdet

✔

—

—

—

—

logical_and

✔

✔

✔

✔

✔

logical_not

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

logical_or

✔

✔

✔

✔

✔

logical_xor

✔

✔

✔

✔

✔

logit

✔

✔

✔

⚠ xfail

⚠ xfail

logspace

✔

✔

✔

✔

✔

logsumexp

✔

✔

✔

✔

✔

long

✔

✔

✔

✔

✔

lt

✔

✔

✔

✔

✔

lu

✔

—

—

—

—

lu_solve

✘ no converter

—

—

—

—

lu_unpack

✘ no converter

—

—

—

—

mH

✔

✔

✔

✔

✔

mT

✔

✔

✔

✔

✔

masked.amax

✔

✔

✔

✔

✔

masked.amin

✔

✔

✔

✔

✔

masked.argmax

✔

✔

✔

✔

✔

masked.argmin

✔

✔

✔

✔

✔

masked.cumprod

✔

✔

✔

✔

✔

masked.cumsum

✔

✔

✔

✔

✔

masked.log_softmax

✔

✔

✔

—

—

masked.logaddexp

✔

✔

✔

—

—

masked.logsumexp

✔

✔

✔

✔

✔

masked.mean

✔

✔

✔

—

—

masked.median

✔

✔

✔

—

—

masked.norm

✔

✔

✔

—

—

masked.normalize

✔

✔

✔

—

—

masked.prod

✔

✔

✔

✔

✔

masked.softmax

✔

✔

✔

—

—

masked.softmin

✔

✔

✔

—

—

masked.std

✔

✔

✔

✔

✔

masked.sum

✔

✔

✔

✔

✔

masked.var

✔

✔

✔

✔

✔

masked_fill

✔

✔

✔

✔

✔

masked_scatter

✔

✔

✔

✔

✔

masked_select

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

matmul

✔

✔

✔

✔

✔

matrix_exp

✘ no converter

✘ no converter

✘ no converter

—

—

max_pool2d_with_indices_backward

✔

✔

✔

—

—

maximum

✔

✔

✔

✔

✔

mean

✔

✔

✔

—

—

median

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

minimum

✔

✔

✔

✔

✔

mm

✔

✔

✔

✔

✔

mode

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

movedim

✔

✔

✔

✔

✔

msort

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

mul

✔

✔

✔

✔

✔

multinomial

✔

✔

✔

—

—

mv

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

nan_to_num

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

nanmean

✔

✔

✔

—

—

nanmedian

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

nanquantile

✔

—

—

—

—

nansum

✔

✔

✔

⚠ xfail

⚠ xfail

narrow

✔

✔

✔

✔

✔

narrow_copy

✔

✔

✔

✔

✔

native_batch_norm

✔

✔

✔

—

—

native_dropout_backward

✔

✔

✔

✔

✔

native_group_norm

✔

✔

✔

—

—

native_layer_norm

✔

✔

✔

—

—

ne

✔

✔

✔

✔

✔

neg

✔

✔

✔

✔

✔

new_empty

✔

✔

✔

✔

✔

new_empty_strided

✔

✔

✔

✔

✔

new_full

✔

✔

✔

✔

✔

new_ones

✔

✔

✔

✔

✔

new_zeros

✔

✔

✔

✔

✔

nextafter

✘ no converter

✘ no converter

✘ no converter

—

—

nn.functional.adaptive_avg_pool1d

✔

✔

✔

—

—

nn.functional.adaptive_avg_pool2d

✔

✔

✔

—

—

nn.functional.adaptive_avg_pool3d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool1d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool2d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool3d

✔

✔

✔

—

—

nn.functional.alpha_dropout

✔

✔

✔

—

—

nn.functional.avg_pool1d

✔

✔

✔

—

✔

nn.functional.avg_pool2d

✔

✔

✔

—

✔

nn.functional.avg_pool3d

✔

—

—

—

✔

nn.functional.batch_norm

✔

✔

✔

—

—

nn.functional.bilinear

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.binary_cross_entropy

✔

✔

✔

—

—

nn.functional.binary_cross_entropy_with_logits

✔

✔

✔

—

—

nn.functional.celu

✔

✔

✔

—

—

nn.functional.channel_shuffle

✔

✔

✔

✔

✔

nn.functional.conv1d

✔

✔

✔

—

✔

nn.functional.conv2d

✔

✔

✔

—

✔

nn.functional.conv3d

✔

✔

✔

—

✔

nn.functional.conv_transpose1d

✔

✔

✔

—

✔

nn.functional.conv_transpose2d

✔

✔

✔

—

✔

nn.functional.conv_transpose3d

✔

✔

✔

—

✔

nn.functional.cosine_embedding_loss

✔

✔

✔

✔

✔

nn.functional.cosine_similarity

✔

✔

✔

—

—

nn.functional.cross_entropy

✔

✔

✔

—

—

nn.functional.ctc_loss

✔

—

—

—

—

nn.functional.dropout

✔

✔

✔

—

—

nn.functional.dropout2d

✔

✔

✔

—

—

nn.functional.dropout3d

✔

✔

✔

—

—

nn.functional.elu

✔

✔

✔

—

—

nn.functional.embedding

✔

✔

✔

—

—

nn.functional.embedding_bag

✔

✔

✔

—

—

nn.functional.fractional_max_pool2d

✔

✔

✔

—

—

nn.functional.fractional_max_pool3d

✔

✔

✔

—

—

nn.functional.gaussian_nll_loss

✔

✔

✔

—

—

nn.functional.gelu

✔

✔

✔

—

—

nn.functional.glu

✔

✔

✔

—

—

nn.functional.grid_sample

✔

✔

✔

—

—

nn.functional.group_norm

✔

✔

✔

—

—

nn.functional.hardshrink

✔

✔

✔

—

—

nn.functional.hardsigmoid

✔

✔

✔

—

—

nn.functional.hardswish

✔

✔

✔

—

—

nn.functional.hardtanh

✔

✔

✔

✔

✔

nn.functional.hinge_embedding_loss

✔

✔

✔

—

—

nn.functional.huber_loss

✔

✔

✔

—

—

nn.functional.instance_norm

✔

✔

✔

—

—

nn.functional.kl_div

✔

✔

✔

—

—

nn.functional.l1_loss

✔

✔

✔

—

—

nn.functional.layer_norm

✔

✔

✔

—

—

nn.functional.leaky_relu

✔

✔

✔

—

—

nn.functional.linear

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.linear_cross_entropy

⚠ xfail

⚠ xfail

⚠ xfail

—

—

nn.functional.local_response_norm

✔

✔

✔

—

✔

nn.functional.logsigmoid

✔

✔

✔

—

—

nn.functional.margin_ranking_loss

✔

✔

✔

✔

✔

nn.functional.max_pool1d

✔

✔

✔

—

—

nn.functional.max_pool2d

✔

✔

✔

✔

✔

nn.functional.max_pool3d

✔

✔

✔

✔

✔

nn.functional.max_unpool1d

✔

✔

✔

—

—

nn.functional.max_unpool2d

✔

✔

✔

—

—

nn.functional.max_unpool3d

✔

✔

✔

—

—

nn.functional.mish

✔

✔

✔

—

—

nn.functional.mse_loss

✔

✔

✔

—

—

nn.functional.multi_head_attention_forward

✔

✔

✔

—

—

nn.functional.multi_margin_loss

✘ no converter

—

—

—

—

nn.functional.multilabel_margin_loss

✘ no converter

—

—

—

—

nn.functional.multilabel_soft_margin_loss

✘ no converter

✘ no converter

✘ no converter

—

—

nn.functional.nll_loss

✔

✔

✔

—

—

nn.functional.normalize

✔

✔

✔

—

—

nn.functional.one_hot

—

—

—

—

✔

nn.functional.pairwise_distance

✔

✔

✔

✔

✔

nn.functional.pdist

✘ no converter

—

—

—

—

nn.functional.pixel_shuffle

✔

✔

✔

✔

✔

nn.functional.pixel_unshuffle

✔

✔

✔

✔

✔

nn.functional.poisson_nll_loss

✔

✔

✔

✔

✔

nn.functional.prelu

✔

✔

✔

—

—

nn.functional.relu

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.relu6

✔

✔

✔

✔

✔

nn.functional.rms_norm

✔

✔

✔

—

—

nn.functional.rrelu

✔

—

✔

—

—

nn.functional.scaled_dot_product_attention

✔

✔

✔

—

—

nn.functional.selu

✔

✔

✔

—

—

nn.functional.silu

✔

✔

✔

—

—

nn.functional.smooth_l1_loss

✔

✔

✔

—

—

nn.functional.soft_margin_loss

✔

⚠ xfail

✔

—

—

nn.functional.softmin

✔

✔

✔

—

—

nn.functional.softplus

✔

✔

✔

—

—

nn.functional.softshrink

✔

✔

✔

—

—

nn.functional.softsign

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.tanhshrink

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.threshold

✔

✔

✔

✔

✔

nn.functional.triplet_margin_loss

✔

✔

✔

✔

✔

nn.functional.triplet_margin_with_distance_loss

✔

✔

✔

✔

✔

nn.functional.unfold

✔

✔

✔

—

—

nn.functional.upsample_bilinear

✔

✔

✔

—

—

nn.functional.upsample_nearest

✔

✔

✔

—

—

nonzero

✔

✔

✔

✔

✔

nonzero_static

✔

✔

✔

✔

✔

norm

✔

✔

✔

—

—

ones

✔

✔

✔

✔

✔

ones_like

✔

✔

✔

✔

✔

ormqr

✔

—

—

—

—

outer

✔

✔

✔

✔

✔

pca_lowrank

✔

—

—

—

—

permute

✔

✔

✔

✔

✔

permute_copy

✔

✔

✔

✔

✔

pinverse

✘ no converter

—

—

—

—

polar

✔

✔

—

—

—

positive

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

pow

✔

✔

✔

✔

✔

prod

✔

✔

✔

⚠ xfail

✔

put

✔

✔

✔

✔

✔

quantile

✔

—

—

—

—

rad2deg

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

randint_like

✔

✔

✔

✔

✔

ravel

✔

✔

✔

✔

✔

real

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

reciprocal

✔

✔

✔

⚠ xfail

⚠ xfail

remainder

✔

✔

✔

✔

✔

renorm

✔

✔

✔

—

—

repeat

✔

✔

✔

✔

✔

repeat_interleave

✔

✔

✔

✔

✔

reshape

✔

✔

✔

✔

✔

reshape_as

✔

✔

✔

✔

✔

resize_

✔

✔

✔

✔

✔

resize_as_

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

resolve_conj

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

resolve_neg

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

roll

✔

✔

✔

✔

✔

rot90

✔

✔

✔

✔

✔

round

✔

✔

✔

⚠ xfail

⚠ xfail

rsqrt

✔

✔

✔

⚠ xfail

⚠ xfail

rsub

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

scalar_tensor

✔

✔

✔

✔

✔

scatter

✔

✔

✔

✔

✔

scatter_add

✔

✔

✔

✔

✔

searchsorted

✔

✔

✔

✔

✔

select

✔

✔

✔

✔

✔

select_scatter

✔

✔

✔

✔

✔

sgn

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

short

✔

✔

✔

✔

✔

sigmoid

✔

✔

✔

⚠ xfail

⚠ xfail

sign

✔

✔

✔

✔

✔

signal.windows.bartlett

✔

—

—

—

—

signal.windows.blackman

✔

—

—

—

—

signal.windows.cosine

✔

—

—

—

—

signal.windows.exponential

✔

—

—

—

—

signal.windows.gaussian

✔

—

—

—

—

signal.windows.general_cosine

✔

—

—

—

—

signal.windows.general_hamming

✔

—

—

—

—

signal.windows.hamming

✔

—

—

—

—

signal.windows.hann

✔

—

—

—

—

signal.windows.kaiser

✔

—

—

—

—

signal.windows.nuttall

✔

—

—

—

—

signbit

✔

✔

✔

✔

✔

sin

✔

✔

✔

⚠ xfail

⚠ xfail

sinc

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

sinh

✔

⚠ xfail

✔

⚠ xfail

⚠ xfail

slice

✔

✔

✔

✔

✔

slice_scatter

✔

✔

✔

✔

✔

softmax

✔

✔

✔

—

—

sort

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

sparse.sampled_addmm

✔

—

—

—

—

special.airy_ai

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_j0

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_j1

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_y0

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_y1

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_t

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_u

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_v

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_w

✘ no converter

—

—

✘ no converter

✘ no converter

special.entr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.erfcx

✘ no converter

—

—

✘ no converter

✘ no converter

special.hermite_polynomial_h

✘ no converter

—

—

✘ no converter

✘ no converter

special.hermite_polynomial_he

✘ no converter

—

—

✘ no converter

✘ no converter

special.i0e

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.i1

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.i1e

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.laguerre_polynomial_l

✘ no converter

—

—

✘ no converter

✘ no converter

special.legendre_polynomial_p

✘ no converter

—

—

✘ no converter

✘ no converter

special.log_ndtr

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_i0

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_i1

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_k0

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_k1

✘ no converter

—

—

✘ no converter

✘ no converter

special.ndtr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.ndtri

✘ no converter

—

—

✘ no converter

✘ no converter

special.scaled_modified_bessel_k0

✘ no converter

—

—

✘ no converter

✘ no converter

special.scaled_modified_bessel_k1

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_t

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_u

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_v

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_w

✘ no converter

—

—

✘ no converter

✘ no converter

special.spherical_bessel_j0

✘ no converter

—

—

✘ no converter

✘ no converter

special.xlog1py

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.zeta

✘ no converter

—

—

✘ no converter

✘ no converter

split

✔

✔

✔

✔

✔

split_with_sizes

✔

✔

✔

✔

✔

split_with_sizes_copy

✔

✔

✔

✔

✔

sqrt

✔

✔

✔

⚠ xfail

⚠ xfail

square

✔

✔

✔

✔

✔

squeeze

✔

✔

✔

✔

✔

squeeze_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

stack

✔

✔

✔

✔

✔

std

✔

✔

✔

—

—

std_mean

✔

✔

✔

—

—

stft

✔

—

—

—

—

sub

✔

✔

✔

✔

✔

sum

✔

✔

✔

⚠ xfail

✔

sum_to_size

✔

✔

✔

✔

✔

svd

✔

—

—

—

—

svd_lowrank

✔

—

—

—

—

t

✔

✔

✔

✔

✔

t_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

take

✔

✔

✔

✔

✔

take_along_dim

✔

✔

✔

✔

✔

tan

✔

✔

✔

⚠ xfail

⚠ xfail

tanh

✔

✔

✔

⚠ xfail

⚠ xfail

tensor_split

✔

✔

✔

✔

✔

tensordot

✔

✔

✔

✔

✔

tile

✔

✔

✔

✔

✔

to

✔

✔

✔

✔

✔

to_sparse

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

topk

✔

✔

✔

✔

✔

torch._scaled_mm

—

—

—

—

—

torch._scaled_mm_v2

—

—

—

—

—

torch.ops.aten._efficient_attention_forward

—

—

—

—

—

torch.ops.aten._flash_attention_forward

—

—

—

—

—

torch.ops.aten._safe_softmax.default

✔

✔

✔

✔

✔

torch.ops.aten._scaled_dot_product_flash_attention_for_cpu

✔

✔

✔

—

—

trace

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

transpose

✔

✔

✔

✔

✔

transpose_copy

✔

✔

✔

✔

✔

trapezoid

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

trapz

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

triangular_solve

✘ no converter

—

—

—

—

tril

✔

✔

✔

⚠ xfail

✔

tril_indices

—

—

—

✔

✔

triu

✔

✔

✔

⚠ xfail

✔

triu_indices

—

—

—

✔

✔

true_divide

✔

✔

✔

✔

✔

trunc

✔

✔

✔

⚠ xfail

⚠ xfail

unbind

✔

✔

✔

✔

✔

unbind_copy

✔

✔

✔

✔

✔

unflatten

✔

✔

✔

✔

✔

unfold

✔

✔

✔

✔

✔

unfold_copy

✔

✔

✔

✔

✔

unique

✔

✔

✔

✔

✔

unique_consecutive

✔

✔

✔

✔

✔

unravel_index

—

—

—

✔

✔

unsafe_chunk

✔

✔

✔

✔

✔

unsafe_split

✔

✔

✔

✔

✔

unsqueeze

✔

✔

✔

✔

✔

unsqueeze_copy

✔

✔

✔

✔

✔

var

✘ no converter

✘ no converter

✘ no converter

—

—

var_mean

✘ no converter

✘ no converter

✘ no converter

—

—

vdot

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

view

✔

✔

✔

✔

✔

view_as

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

view_as_complex

✔

✔

—

—

—

view_as_real

—

—

—

—

—

view_copy

✔

✔

✔

✔

✔

vsplit

✔

✔

✔

✔

✔

vstack

✔

✔

✔

✔

✔

where

✔

✔

✔

✔

✔

xlogy

✔

✔

✔

⚠ xfail

⚠ xfail

zero_

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

zeros

✔

✔

✔

✔

✔

zeros_like

✔

✔

✔

✔

✔

Torch tracing export path#

Torch tracing export path (tracing=True)

Op

float32

float16

bfloat16

int32

int64

H

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

T

✔

✔

✔

✔

✔

__getitem__

✔

✔

✔

✔

✔

__radd__

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

__rand__

—

—

—

⚠ xfail

⚠ xfail

__rdiv__

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

__rmatmul__

✔

✔

✔

✔

✔

__rmod__

✔

✔

✔

✔

✔

__rmul__

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

__ror__

—

—

—

⚠ xfail

⚠ xfail

__rpow__

✔

✔

✔

⚠ xfail

⚠ xfail

__rsub__

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

__rxor__

—

—

—

⚠ xfail

⚠ xfail

_batch_norm_with_update

✔

✔

✔

—

—

_chunk_cat

✔

✔

✔

✔

✔

_native_batch_norm_legit

✔

✔

✔

—

—

_softmax_backward_data

✔

✔

✔

—

—

_unsafe_masked_index

✔

✔

✔

✔

✔

_unsafe_masked_index_put_accumulate

✔

✔

✔

✔

✔

_upsample_bilinear2d_aa

✔

—

—

—

—

abs

✔

✔

✔

✔

✔

acos

✔

✔

✔

⚠ xfail

⚠ xfail

acosh

✔

✔

✔

⚠ xfail

⚠ xfail

add

✔

✔

✔

✔

✔

addbmm

✔

✔

✔

✔

✔

addcdiv

✘ no converter

✘ no converter

✘ no converter

—

—

addcmul

✔

⚠ xfail

⚠ xfail

✔

✔

addmm

✔

✔

✔

✔

✔

addmv

✔

✔

✔

✔

✔

addr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

alias_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

all

✔

✔

✔

✔

✔

allclose

✔

✔

✔

—

—

amax

✔

✔

✔

✔

✔

amin

✔

✔

✔

✔

✔

aminmax

✔

✔

✔

✔

✔

angle

✔

✔

✔

⚠ xfail

⚠ xfail

any

✔

✔

✔

✔

✔

arange

✔

✔

✔

✔

✔

argmax

✔

✔

✔

✔

✔

argmin

✔

✔

✔

✔

✔

argsort

✔

⚠ xfail

⚠ xfail

✔

✔

argwhere

✔

✔

✔

✔

✔

as_strided

✔

✔

✔

✔

✔

as_strided_copy

✔

✔

✔

✔

✔

as_strided_scatter

✔

✔

✔

✔

✔

asin

✔

✔

✔

⚠ xfail

⚠ xfail

asinh

✔

✔

✔

⚠ xfail

⚠ xfail

atan

✔

✔

✔

⚠ xfail

⚠ xfail

atan2

✔

✔

✔

⚠ xfail

⚠ xfail

atanh

✔

✔

✔

⚠ xfail

⚠ xfail

atleast_1d

✔

✔

✔

✔

✔

atleast_2d

✔

✔

✔

✔

✔

atleast_3d

✔

✔

✔

✔

✔

baddbmm

✔

✔

✔

✔

✔

bernoulli

✘ no converter

✘ no converter

✘ no converter

—

—

bfloat16

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

bincount

—

—

—

✔

✔

bitwise_and

—

—

—

⚠ xfail

⚠ xfail

bitwise_left_shift

—

—

—

✔

✔

bitwise_not

—

—

—

✔

✔

bitwise_or

—

—

—

⚠ xfail

⚠ xfail

bitwise_right_shift

—

—

—

✔

✔

bitwise_xor

—

—

—

✔

✔

block_diag

✔

✔

✔

✔

✔

bmm

✔

✔

⚠ xfail

✔

✔

bool

✔

✔

✔

✔

✔

broadcast_shapes

✔

—

—

—

—

broadcast_tensors

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

broadcast_to

✔

✔

✔

✔

✔

bucketize

✔

✔

✔

✔

✔

byte

✔

✔

✔

✔

✔

cartesian_prod

✔

✔

✔

✔

✔

cat

✔

✔

✔

✔

✔

cdist

✔

—

—

—

—

cdouble

✔

✔

✔

✔

✔

ceil

✔

✔

✔

⚠ xfail

⚠ xfail

cfloat

✔

✔

✔

✔

✔

chalf

✔

✔

✔

✔

✔

char

✔

✔

✔

✔

✔

cholesky_inverse

⚠ xfail

—

—

—

—

cholesky_solve

⚠ xfail

—

—

—

—

chunk

✔

✔

✔

✔

✔

clamp

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

clamp_max

✔

✔

✔

✔

✔

clamp_min

✔

✔

✔

✔

✔

clone

✔

✔

✔

✔

✔

column_stack

✔

✔

✔

✔

✔

combinations

✔

✔

✔

✔

✔

complex

✔

✔

—

—

—

conj

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

conj_physical

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

constant_pad_nd

✔

✔

✔

✔

✔

contiguous

✔

✔

✔

✔

✔

copysign

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

corrcoef

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

cos

✔

✔

✔

⚠ xfail

⚠ xfail

cosh

✔

⚠ xfail

✔

⚠ xfail

⚠ xfail

count_nonzero

✔

✔

✔

✔

✔

cov

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

cross

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

cummax

✔

✔

✔

✔

✔

cummin

✔

✔

✔

✔

✔

cumprod

✔

✔

✔

✔

✔

cumsum

✔

✔

✔

✔

✔

cumulative_trapezoid

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

deg2rad

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diag

✔

✔

✔

✔

✔

diag_embed

✔

✔

✔

✔

✔

diagflat

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal_scatter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diff

✔

✔

✔

✔

✔

digamma

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

dist

✔

✔

✔

—

—

dot

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

double

✔

✔

✔

✔

✔

dsplit

✔

✔

✔

✔

✔

dstack

✔

✔

✔

✔

✔

einsum

✔

✔

✔

✔

✔

empty_permuted

✔

✔

✔

✔

✔

empty_strided

✔

✔

✔

✔

✔

eq

✔

✔

✔

✔

✔

equal

✔

✔

✔

✔

✔

erf

✔

✔

✔

⚠ xfail

⚠ xfail

erfc

✔

✔

✔

⚠ xfail

⚠ xfail

erfinv

✔

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

exp

✔

✔

✔

⚠ xfail

⚠ xfail

exp2

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

expand

✔

✔

✔

✔

✔

expand_as

✔

✔

✔

✔

✔

expand_copy

✔

✔

✔

✔

✔

expm1

✔

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

eye

✔

✔

✔

✔

✔

fft.fft

✔

—

—

✔

✔

fft.fft2

✔

—

—

✔

✔

fft.fftn

✔

—

—

✔

✔

fft.fftshift

✔

✔

✔

✔

✔

fft.hfft

✔

—

—

✔

✔

fft.hfft2

✔

—

—

✔

✔

fft.hfftn

✔

—

—

✔

✔

fft.ifft

✔

—

—

✔

✔

fft.ifft2

✔

—

—

✔

✔

fft.ifftn

✔

—

—

✔

✔

fft.ifftshift

✔

✔

✔

✔

✔

fft.ihfft

✔

—

—

✔

✔

fft.ihfft2

✔

—

—

✔

✔

fft.ihfftn

✔

—

—

✔

✔

fft.irfft

✔

—

—

✔

✔

fft.irfft2

✔

—

—

✔

✔

fft.irfftn

✔

—

—

✔

✔

fft.rfft

✔

—

—

✔

✔

fft.rfft2

✔

—

—

✔

✔

fft.rfftn

✔

—

—

✔

✔

fill

✔

✔

✔

✔

✔

flatten

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

flip

✔

✔

✔

✔

✔

fliplr

✔

✔

✔

✔

✔

flipud

✔

✔

✔

✔

✔

float

✔

✔

✔

✔

✔

float_power

✔

✔

⚠ xfail

✔

✔

floor

✔

✔

✔

⚠ xfail

⚠ xfail

floor_divide

✔

✔

✔

⚠ xfail

⚠ xfail

fmax

✔

✔

✔

✔

✔

fmin

✔

✔

✔

✔

✔

fmod

✔

✔

✔

✔

✔

frac

✔

✔

✔

—

—

frexp

✔

✔

⚠ xfail

—

—

full

✔

✔

✔

✔

✔

full_like

✔

✔

✔

✔

✔

gather

✔

✔

✔

✔

✔

gcd

—

—

—

⚠ xfail

⚠ xfail

ge

✔

✔

✔

✔

✔

geqrf

✔

—

—

—

—

gradient

✔

✔

✔

✔

✔

grid_sampler_2d

✔

✔

✔

—

—

grid_sampler_3d

✔

✔

✔

—

—

gt

✔

✔

✔

✔

✔

half

✔

✔

✔

✔

✔

hash_tensor

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

heaviside

✔

✔

✔

✔

✔

histc

✔

✔

✔

—

—

histogram

✔

—

—

—

—

histogramdd

✔

—

—

—

—

hsplit

✔

✔

✔

✔

✔

hstack

✔

✔

✔

✔

✔

hypot

✘ no converter

✘ no converter

✘ no converter

—

—

i0

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

igamma

✘ no converter

✘ no converter

✘ no converter

—

—

igammac

✘ no converter

✘ no converter

✘ no converter

—

—

imag

—

—

—

—

—

index_add

✔

✔

✔

✔

✔

index_copy

✔

✔

✔

✔

✔

index_fill

✔

✔

✔

✔

✔

index_put

✔

✔

✔

✔

✔

index_select

✔

✔

✔

✔

✔

inner

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

int

✔

✔

✔

✔

✔

isclose

✔

✔

✔

✔

✔

isfinite

✔

✔

✔

✔

✔

isin

✔

✔

✔

✔

✔

isinf

✔

✔

✔

⚠ xfail

⚠ xfail

isnan

✔

✔

✔

⚠ xfail

⚠ xfail

isneginf

✔

✔

✔

✔

✔

isposinf

✔

✔

✔

✔

✔

isreal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

istft

—

—

—

—

—

item

✔

✔

✔

✔

✔

jiterator_2inputs_2outputs

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_4inputs_with_extra_args

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary_return_by_ref

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_unary

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

kron

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

kthvalue

✔

✔

✔

✔

✔

lcm

—

—

—

⚠ xfail

⚠ xfail

ldexp

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

le

✔

✔

✔

✔

✔

lerp

✔

✔

✔

—

—

lgamma

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

linalg.cholesky

✔

—

—

—

—

linalg.cholesky_ex

✔

—

—

—

—

linalg.cond

✘ no converter

—

—

—

—

linalg.cross

✔

✔

✔

✔

✔

linalg.det

✔

—

—

—

—

linalg.diagonal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

linalg.eig

✔

—

—

—

—

linalg.eigh

✔

—

—

—

—

linalg.eigvals

✔

—

—

—

—

linalg.eigvalsh

✔

—

—

—

—

linalg.householder_product

✘ no converter

—

—

—

—

linalg.inv

✘ no converter

—

—

—

—

linalg.inv_ex

✘ no converter

—

—

—

—

linalg.ldl_factor

✔

—

—

—

—

linalg.ldl_factor_ex

✔

—

—

—

—

linalg.ldl_solve

✔

—

—

—

—

linalg.lstsq

✔

—

—

—

—

linalg.lu

✔

—

—

—

—

linalg.lu_factor

✔

—

—

—

—

linalg.lu_factor_ex

✔

—

—

—

—

linalg.lu_solve

✔

—

—

—

—

linalg.matrix_norm

✔

✔

✔

—

—

linalg.matrix_power

✔

—

—

—

—

linalg.matrix_rank

✔

—

—

—

—

linalg.matrix_sqrth

✔

—

—

—

—

linalg.multi_dot

✔

✔

✔

✔

✔

linalg.norm

✔

✔

✔

—

—

linalg.pinv

✔

—

—

—

—

linalg.polar

✔

—

—

—

—

linalg.qr

✘ no converter

—

—

—

—

linalg.slogdet

✔

—

—

—

—

linalg.solve

✘ no converter

—

—

—

—

linalg.solve_ex

✘ no converter

—

—

—

—

linalg.solve_triangular

✔

—

—

—

—

linalg.svd

✔

—

—

—

—

linalg.svdvals

✘ no converter

—

—

—

—

linalg.tensorinv

✔

—

—

—

—

linalg.tensorsolve

✔

—

—

—

—

linalg.vander

✔

—

—

✔

✔

linalg.vecdot

✔

✔

✔

—

—

linalg.vector_norm

✔

✔

✔

—

—

linspace

✔

✔

✔

✔

✔

log

✔

✔

✔

⚠ xfail

⚠ xfail

log10

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log1p

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log2

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log_softmax

✔

✔

✔

—

—

logaddexp

✔

✔

⚠ xfail

—

—

logaddexp2

✔

✔

⚠ xfail

—

—

logcumsumexp

✔

✔

✔

—

—

logdet

✔

—

—

—

—

logical_and

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

logical_not

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

logical_or

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

logical_xor

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

logit

✔

✔

✔

⚠ xfail

⚠ xfail

logspace

✔

✔

✔

✔

✔

logsumexp

✔

✔

✔

✔

✔

long

✔

✔

✔

✔

✔

lt

✔

✔

✔

✔

✔

lu

✔

—

—

—

—

lu_solve

✘ no converter

—

—

—

—

lu_unpack

✘ no converter

—

—

—

—

mH

✔

✔

✔

✔

✔

mT

✔

✔

✔

✔

✔

masked.amax

✔

✔

✔

✔

✔

masked.amin

✔

✔

✔

✔

✔

masked.argmax

✔

✔

✔

✔

✔

masked.argmin

✔

✔

✔

✔

✔

masked.cumprod

✔

✔

✔

✔

✔

masked.cumsum

✔

✔

✔

✔

✔

masked.log_softmax

✔

✔

✔

—

—

masked.logaddexp

✔

✔

✔

—

—

masked.logsumexp

✔

✔

✔

✔

✔

masked.mean

✔

✔

✔

—

—

masked.median

✔

✔

✔

—

—

masked.norm

✔

✔

✔

—

—

masked.normalize

✔

✔

✔

—

—

masked.prod

✔

✔

✔

✔

✔

masked.softmax

✔

✔

✔

—

—

masked.softmin

✔

✔

✔

—

—

masked.std

✔

✔

✔

✔

✔

masked.sum

✔

✔

✔

✔

✔

masked.var

✔

✔

✔

✔

✔

masked_fill

✔

✔

✔

✔

✔

masked_scatter

✔

✔

✔

✔

✔

masked_select

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

matmul

✔

✔

✔

✔

✔

matrix_exp

✘ no converter

✘ no converter

✘ no converter

—

—

max_pool2d_with_indices_backward

✔

✔

✔

—

—

maximum

✔

✔

✔

✔

✔

mean

✔

✔

✔

—

—

median

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

minimum

✔

✔

✔

✔

✔

mm

✔

✔

✔

✔

✔

mode

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

movedim

✔

✔

✔

✔

✔

msort

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

mul

✔

✔

✔

✔

✔

multinomial

✔

✔

✔

—

—

mv

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

nan_to_num

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

nanmean

✔

✔

✔

—

—

nanmedian

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

nanquantile

✔

—

—

—

—

nansum

✔

✔

✔

⚠ xfail

⚠ xfail

narrow

✔

✔

✔

✔

✔

narrow_copy

✔

✔

✔

✔

✔

native_batch_norm

✔

✔

✔

—

—

native_dropout_backward

✔

✔

✔

✔

✔

native_group_norm

✔

✔

✔

—

—

native_layer_norm

✔

✔

✔

—

—

ne

✔

✔

✔

✔

✔

neg

✔

✔

✔

✔

✔

new_empty

✔

✔

✔

✔

✔

new_empty_strided

✔

✔

✔

✔

✔

new_full

✔

✔

✔

✔

✔

new_ones

✔

✔

✔

✔

✔

new_zeros

✔

✔

✔

✔

✔

nextafter

✘ no converter

✘ no converter

✘ no converter

—

—

nn.functional.adaptive_avg_pool1d

✔

✔

✔

—

—

nn.functional.adaptive_avg_pool2d

✔

✔

✔

—

—

nn.functional.adaptive_avg_pool3d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool1d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool2d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool3d

✔

✔

✔

—

—

nn.functional.alpha_dropout

✔

✔

✔

—

—

nn.functional.avg_pool1d

✔

✔

✔

—

✔

nn.functional.avg_pool2d

✔

✔

✔

—

✔

nn.functional.avg_pool3d

✔

—

—

—

✔

nn.functional.batch_norm

✔

✔

✔

—

—

nn.functional.bilinear

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

nn.functional.binary_cross_entropy

✔

✔

✔

—

—

nn.functional.binary_cross_entropy_with_logits

✔

✔

✔

—

—

nn.functional.celu

✔

✔

✔

—

—

nn.functional.channel_shuffle

✔

✔

✔

✔

✔

nn.functional.conv1d

✔

✔

✔

—

✔

nn.functional.conv2d

✔

✔

✔

—

✔

nn.functional.conv3d

✔

✔

✔

—

✔

nn.functional.conv_transpose1d

✔

✔

✔

—

✔

nn.functional.conv_transpose2d

✔

✔

✔

—

✔

nn.functional.conv_transpose3d

✔

✔

✔

—

✔

nn.functional.cosine_embedding_loss

✔

✔

✔

✔

✔

nn.functional.cosine_similarity

✔

✔

✔

—

—

nn.functional.cross_entropy

✔

✔

✔

—

—

nn.functional.ctc_loss

✔

—

—

—

—

nn.functional.dropout

✔

✔

✔

—

—

nn.functional.dropout2d

✔

✔

✔

—

—

nn.functional.dropout3d

✔

✔

✔

—

—

nn.functional.elu

✔

✔

✔

—

—

nn.functional.embedding

✔

✔

✔

—

—

nn.functional.embedding_bag

✔

✔

✔

—

—

nn.functional.fractional_max_pool2d

✔

✔

✔

—

—

nn.functional.fractional_max_pool3d

✔

✔

✔

—

—

nn.functional.gaussian_nll_loss

✔

✔

✔

—

—

nn.functional.gelu

✔

✔

✔

—

—

nn.functional.glu

✔

✔

✔

—

—

nn.functional.grid_sample

✔

✔

✔

—

—

nn.functional.group_norm

✔

✔

✔

—

—

nn.functional.hardshrink

✔

✔

✔

—

—

nn.functional.hardsigmoid

⚠ xfail

⚠ xfail

⚠ xfail

—

—

nn.functional.hardswish

✔

✔

✔

—

—

nn.functional.hardtanh

✔

✔

✔

✔

✔

nn.functional.hinge_embedding_loss

✔

✔

✔

—

—

nn.functional.huber_loss

✔

✔

✔

—

—

nn.functional.instance_norm

✔

✔

✔

—

—

nn.functional.kl_div

✔

✔

✔

—

—

nn.functional.l1_loss

✔

✔

✔

—

—

nn.functional.layer_norm

✔

✔

✔

—

—

nn.functional.leaky_relu

✔

✔

✔

—

—

nn.functional.linear

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.linear_cross_entropy

⚠ xfail

⚠ xfail

⚠ xfail

—

—

nn.functional.local_response_norm

✔

✔

✔

—

✔

nn.functional.logsigmoid

✔

✔

✔

—

—

nn.functional.margin_ranking_loss

✔

✔

✔

✔

✔

nn.functional.max_pool1d

✔

✔

✔

—

—

nn.functional.max_pool2d

✔

✔

✔

✔

✔

nn.functional.max_pool3d

✔

✔

✔

✔

✔

nn.functional.max_unpool1d

✔

✔

✔

—

—

nn.functional.max_unpool2d

✔

✔

✔

—

—

nn.functional.max_unpool3d

✔

✔

✔

—

—

nn.functional.mish

✔

✔

✔

—

—

nn.functional.mse_loss

✔

✔

✔

—

—

nn.functional.multi_head_attention_forward

✔

✔

✔

—

—

nn.functional.multi_margin_loss

✘ no converter

—

—

—

—

nn.functional.multilabel_margin_loss

✘ no converter

—

—

—

—

nn.functional.multilabel_soft_margin_loss

✘ no converter

✘ no converter

✘ no converter

—

—

nn.functional.nll_loss

✔

✔

✔

—

—

nn.functional.normalize

✔

✔

✔

—

—

nn.functional.one_hot

—

—

—

—

✔

nn.functional.pairwise_distance

✔

✔

✔

✔

✔

nn.functional.pdist

✘ no converter

—

—

—

—

nn.functional.pixel_shuffle

✔

✔

✔

✔

✔

nn.functional.pixel_unshuffle

✔

✔

✔

✔

✔

nn.functional.poisson_nll_loss

✔

✔

✔

✔

✔

nn.functional.prelu

✔

✔

✔

—

—

nn.functional.relu

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.relu6

✔

✔

✔

✔

✔

nn.functional.rms_norm

✔

✔

✔

—

—

nn.functional.rrelu

✔

—

✔

—

—

nn.functional.scaled_dot_product_attention

✔

✔

✔

—

—

nn.functional.selu

✔

✔

✔

—

—

nn.functional.silu

✔

✔

✔

—

—

nn.functional.smooth_l1_loss

✔

✔

✔

—

—

nn.functional.soft_margin_loss

✔

⚠ xfail

✔

—

—

nn.functional.softmin

✔

✔

✔

—

—

nn.functional.softplus

✔

✔

✔

—

—

nn.functional.softshrink

✔

✔

✔

—

—

nn.functional.softsign

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.tanhshrink

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.threshold

✔

✔

✔

✔

✔

nn.functional.triplet_margin_loss

✔

✔

✔

✔

✔

nn.functional.triplet_margin_with_distance_loss

✔

✔

✔

✔

✔

nn.functional.unfold

✔

✔

✔

—

—

nn.functional.upsample_bilinear

✔

✔

✔

—

—

nn.functional.upsample_nearest

✔

✔

✔

—

—

nonzero

✔

✔

✔

✔

✔

nonzero_static

✔

✔

✔

✔

✔

norm

✔

✔

✔

—

—

ones

✔

✔

✔

✔

✔

ones_like

✔

✔

✔

✔

✔

ormqr

✔

—

—

—

—

outer

✔

✔

✔

✔

✔

pca_lowrank

✔

—

—

—

—

permute

✔

✔

✔

✔

✔

permute_copy

✔

✔

✔

✔

✔

pinverse

✘ no converter

—

—

—

—

polar

✔

✔

—

—

—

positive

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

pow

✔

✔

✔

✔

✔

prod

✔

✔

✔

⚠ xfail

✔

put

✔

✔

✔

✔

✔

quantile

✔

—

—

—

—

rad2deg

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

randint_like

✔

✔

✔

✔

✔

ravel

✔

✔

✔

✔

✔

real

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

reciprocal

✔

✔

✔

⚠ xfail

⚠ xfail

remainder

✔

✔

✔

✔

✔

renorm

✔

✔

✔

—

—

repeat

✔

✔

✔

✔

✔

repeat_interleave

✔

✔

✔

✔

✔

reshape

✔

✔

✔

✔

✔

reshape_as

✔

✔

✔

✔

✔

resize_

✔

✔

✔

✔

✔

resize_as_

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

resolve_conj

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

resolve_neg

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

roll

✔

✔

✔

✔

✔

rot90

✔

✔

✔

✔

✔

round

✔

✔

✔

⚠ xfail

⚠ xfail

rsqrt

✔

✔

✔

⚠ xfail

⚠ xfail

rsub

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

scalar_tensor

✔

✔

✔

✔

✔

scatter

✔

✔

✔

✔

✔

scatter_add

✔

✔

✔

✔

✔

searchsorted

✔

✔

✔

✔

✔

select

✔

✔

✔

✔

✔

select_scatter

✔

✔

✔

✔

✔

sgn

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

short

✔

✔

✔

✔

✔

sigmoid

✔

✔

✔

⚠ xfail

⚠ xfail

sign

✔

✔

✔

✔

✔

signal.windows.bartlett

✔

—

—

—

—

signal.windows.blackman

✔

—

—

—

—

signal.windows.cosine

✔

—

—

—

—

signal.windows.exponential

✔

—

—

—

—

signal.windows.gaussian

✔

—

—

—

—

signal.windows.general_cosine

✔

—

—

—

—

signal.windows.general_hamming

✔

—

—

—

—

signal.windows.hamming

✔

—

—

—

—

signal.windows.hann

✔

—

—

—

—

signal.windows.kaiser

✔

—

—

—

—

signal.windows.nuttall

✔

—

—

—

—

signbit

✔

✔

✔

✔

✔

sin

✔

✔

✔

⚠ xfail

⚠ xfail

sinc

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

sinh

✔

⚠ xfail

✔

⚠ xfail

⚠ xfail

slice

✔

✔

✔

✔

✔

slice_scatter

✔

✔

✔

✔

✔

softmax

✔

✔

✔

—

—

sort

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

sparse.sampled_addmm

✔

—

—

—

—

special.airy_ai

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_j0

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_j1

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_y0

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_y1

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_t

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_u

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_v

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_w

✘ no converter

—

—

✘ no converter

✘ no converter

special.entr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.erfcx

✘ no converter

—

—

✘ no converter

✘ no converter

special.hermite_polynomial_h

✘ no converter

—

—

✘ no converter

✘ no converter

special.hermite_polynomial_he

✘ no converter

—

—

✘ no converter

✘ no converter

special.i0e

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.i1

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.i1e

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.laguerre_polynomial_l

✘ no converter

—

—

✘ no converter

✘ no converter

special.legendre_polynomial_p

✘ no converter

—

—

✘ no converter

✘ no converter

special.log_ndtr

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_i0

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_i1

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_k0

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_k1

✘ no converter

—

—

✘ no converter

✘ no converter

special.ndtr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.ndtri

✘ no converter

—

—

✘ no converter

✘ no converter

special.scaled_modified_bessel_k0

✘ no converter

—

—

✘ no converter

✘ no converter

special.scaled_modified_bessel_k1

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_t

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_u

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_v

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_w

✘ no converter

—

—

✘ no converter

✘ no converter

special.spherical_bessel_j0

✘ no converter

—

—

✘ no converter

✘ no converter

special.xlog1py

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.zeta

✘ no converter

—

—

✘ no converter

✘ no converter

split

✔

✔

✔

✔

✔

split_with_sizes

✔

✔

✔

✔

✔

split_with_sizes_copy

✔

✔

✔

✔

✔

sqrt

✔

✔

✔

⚠ xfail

⚠ xfail

square

✔

✔

✔

✔

✔

squeeze

✔

✔

✔

✔

✔

squeeze_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

stack

✔

✔

✔

✔

✔

std

⚠ xfail

⚠ xfail

⚠ xfail

—

—

std_mean

⚠ xfail

⚠ xfail

⚠ xfail

—

—

stft

✔

—

—

—

—

sub

✔

✔

✔

✔

✔

sum

✔

✔

✔

⚠ xfail

✔

sum_to_size

✔

✔

✔

✔

✔

svd

✔

—

—

—

—

svd_lowrank

✔

—

—

—

—

t

✔

✔

✔

✔

✔

t_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

take

✔

✔

✔

✔

✔

take_along_dim

✔

✔

✔

✔

✔

tan

✔

✔

✔

⚠ xfail

⚠ xfail

tanh

✔

✔

✔

⚠ xfail

⚠ xfail

tensor_split

✔

✔

✔

✔

✔

tensordot

✔

✔

✔

✔

✔

tile

✔

✔

✔

✔

✔

to

✔

✔

✔

✔

✔

to_sparse

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

topk

✔

✔

✔

✔

✔

torch._scaled_mm

—

—

—

—

—

torch._scaled_mm_v2

—

—

—

—

—

torch.ops.aten._efficient_attention_forward

—

—

—

—

—

torch.ops.aten._flash_attention_forward

—

—

—

—

—

torch.ops.aten._safe_softmax.default

✔

✔

✔

✔

✔

torch.ops.aten._scaled_dot_product_flash_attention_for_cpu

✔

✔

✔

—

—

trace

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

transpose

✔

✔

✔

✔

✔

transpose_copy

✔

✔

✔

✔

✔

trapezoid

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

trapz

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

triangular_solve

✘ no converter

—

—

—

—

tril

✔

✔

✔

⚠ xfail

✔

tril_indices

—

—

—

✔

✔

triu

✔

✔

✔

⚠ xfail

✔

triu_indices

—

—

—

✔

✔

true_divide

✔

✔

✔

✔

✔

trunc

✔

✔

✔

⚠ xfail

⚠ xfail

unbind

✔

✔

✔

✔

✔

unbind_copy

✔

✔

✔

✔

✔

unflatten

✔

✔

✔

✔

✔

unfold

✔

✔

✔

✔

✔

unfold_copy

✔

✔

✔

✔

✔

unique

✔

✔

✔

✔

✔

unique_consecutive

✔

✔

✔

✔

✔

unravel_index

—

—

—

✔

✔

unsafe_chunk

✔

✔

✔

✔

✔

unsafe_split

✔

✔

✔

✔

✔

unsqueeze

✔

✔

✔

✔

✔

unsqueeze_copy

✔

✔

✔

✔

✔

var

✘ no converter

✘ no converter

✘ no converter

—

—

var_mean

✘ no converter

✘ no converter

✘ no converter

—

—

vdot

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

view

✔

✔

✔

✔

✔

view_as

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

view_as_complex

✔

✔

—

—

—

view_as_real

—

—

—

—

—

view_copy

✔

✔

✔

✔

✔

vsplit

✔

✔

✔

✔

✔

vstack

✔

✔

✔

✔

✔

where

✔

✔

✔

✔

✔

xlogy

✔

✔

✔

⚠ xfail

⚠ xfail

zero_

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

zeros

✔

✔

✔

✔

✔

zeros_like

✔

✔

✔

✔

✔

New-tracing export path#

New-tracing export path (tracing=TracingMode.NEW_TRACING)

Op

float32

float16

bfloat16

int32

int64

H

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

T

✔

✔

✔

✔

✔

__getitem__

✔

✔

✔

✔

✔

__radd__

✔

✔

✔

✔

✔

__rand__

—

—

—

✔

✔

__rdiv__

✔

✔

✔

⚠ xfail

⚠ xfail

__rmatmul__

✔

✔

✔

✔

✔

__rmod__

✔

✔

✔

✔

✔

__rmul__

✔

✔

✔

✔

✔

__ror__

—

—

—

✔

✔

__rpow__

✔

✔

✔

⚠ xfail

⚠ xfail

__rsub__

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

__rxor__

—

—

—

⚠ xfail

⚠ xfail

_batch_norm_with_update

✔

✔

✔

—

—

_chunk_cat

✔

✔

✔

✔

✔

_native_batch_norm_legit

✔

✔

✔

—

—

_softmax_backward_data

✔

✔

✔

—

—

_unsafe_masked_index

✔

✔

✔

✔

✔

_unsafe_masked_index_put_accumulate

✔

✔

✔

✔

✔

_upsample_bilinear2d_aa

✔

—

—

—

—

abs

✔

✔

✔

✔

✔

acos

✔

✔

✔

⚠ xfail

⚠ xfail

acosh

✔

✔

✔

⚠ xfail

⚠ xfail

add

✔

✔

✔

✔

✔

addbmm

✔

✔

✔

✔

✔

addcdiv

✘ no converter

✘ no converter

✘ no converter

—

—

addcmul

✔

⚠ xfail

⚠ xfail

✔

✔

addmm

✔

✔

✔

✔

✔

addmv

✔

✔

✔

✔

✔

addr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

alias_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

all

✔

✔

✔

✔

✔

allclose

✔

✔

✔

—

—

amax

✔

✔

✔

✔

✔

amin

✔

✔

✔

✔

✔

aminmax

✔

✔

✔

✔

✔

angle

✔

✔

✔

⚠ xfail

⚠ xfail

any

✔

✔

✔

✔

✔

arange

✔

✔

✔

✔

✔

argmax

✔

✔

✔

✔

✔

argmin

✔

✔

✔

✔

✔

argsort

✔

⚠ xfail

⚠ xfail

✔

✔

argwhere

✔

✔

✔

✔

✔

as_strided

✔

✔

✔

✔

✔

as_strided_copy

✔

✔

✔

✔

✔

as_strided_scatter

✔

✔

✔

✔

✔

asin

✔

✔

✔

⚠ xfail

⚠ xfail

asinh

✔

✔

✔

⚠ xfail

⚠ xfail

atan

✔

✔

✔

⚠ xfail

⚠ xfail

atan2

✔

✔

✔

⚠ xfail

⚠ xfail

atanh

✔

✔

✔

⚠ xfail

⚠ xfail

atleast_1d

✔

✔

✔

✔

✔

atleast_2d

✔

✔

✔

✔

✔

atleast_3d

✔

✔

✔

✔

✔

baddbmm

✔

✔

✔

✔

✔

bernoulli

✘ no converter

✘ no converter

✘ no converter

—

—

bfloat16

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

bincount

—

—

—

✔

✔

bitwise_and

—

—

—

✔

✔

bitwise_left_shift

—

—

—

✔

✔

bitwise_not

—

—

—

✔

✔

bitwise_or

—

—

—

✔

✔

bitwise_right_shift

—

—

—

✔

✔

bitwise_xor

—

—

—

✔

✔

block_diag

✔

✔

✔

✔

✔

bmm

✔

✔

⚠ xfail

✔

✔

bool

✔

✔

✔

✔

✔

broadcast_shapes

✔

—

—

—

—

broadcast_tensors

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

broadcast_to

✔

✔

✔

✔

✔

bucketize

✔

✔

✔

✔

✔

byte

✔

✔

✔

✔

✔

cartesian_prod

✔

✔

✔

✔

✔

cat

✔

✔

✔

✔

✔

cdist

✔

—

—

—

—

cdouble

✔

✔

✔

✔

✔

ceil

✔

✔

✔

⚠ xfail

⚠ xfail

cfloat

✔

✔

✔

✔

✔

chalf

✔

✔

✔

✔

✔

char

✔

✔

✔

✔

✔

cholesky_inverse

⚠ xfail

—

—

—

—

cholesky_solve

⚠ xfail

—

—

—

—

chunk

✔

✔

✔

✔

✔

clamp

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

clamp_max

✔

✔

✔

✔

✔

clamp_min

✔

✔

✔

✔

✔

clone

✔

✔

✔

✔

✔

column_stack

✔

✔

✔

✔

✔

combinations

✔

✔

✔

✔

✔

complex

✔

✔

—

—

—

conj

✔

✔

✔

✔

✔

conj_physical

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

constant_pad_nd

✔

✔

✔

✔

✔

contiguous

✔

✔

✔

✔

✔

copysign

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

corrcoef

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

cos

✔

✔

✔

⚠ xfail

⚠ xfail

cosh

✔

⚠ xfail

✔

⚠ xfail

⚠ xfail

count_nonzero

✔

✔

✔

✔

✔

cov

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

cross

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

cummax

✔

✔

✔

✔

✔

cummin

✔

✔

✔

✔

✔

cumprod

✔

✔

✔

✔

✔

cumsum

✔

✔

✔

✔

✔

cumulative_trapezoid

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

deg2rad

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diag

✔

✔

✔

✔

✔

diag_embed

✔

✔

✔

✔

✔

diagflat

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diagonal_scatter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

diff

✔

✔

✔

✔

✔

digamma

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

dist

✔

✔

✔

—

—

dot

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

double

✔

✔

✔

✔

✔

dsplit

✔

✔

✔

✔

✔

dstack

✔

✔

✔

✔

✔

einsum

✔

✔

✔

✔

✔

empty_permuted

✔

✔

✔

✔

✔

empty_strided

✔

✔

✔

✔

✔

eq

✔

✔

✔

✔

✔

equal

✔

✔

✔

✔

✔

erf

✔

✔

✔

⚠ xfail

⚠ xfail

erfc

✔

✔

✔

⚠ xfail

⚠ xfail

erfinv

✔

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

exp

✔

✔

✔

⚠ xfail

⚠ xfail

exp2

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

expand

✔

✔

✔

✔

✔

expand_as

✔

✔

✔

✔

✔

expand_copy

✔

✔

✔

✔

✔

expm1

✔

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

eye

✔

✔

✔

✔

✔

fft.fft

✔

—

—

✔

✔

fft.fft2

✔

—

—

✔

✔

fft.fftn

✔

—

—

✔

✔

fft.fftshift

✔

✔

✔

✔

✔

fft.hfft

✔

—

—

✔

✔

fft.hfft2

✔

—

—

✔

✔

fft.hfftn

✔

—

—

✔

✔

fft.ifft

✔

—

—

✔

✔

fft.ifft2

✔

—

—

✔

✔

fft.ifftn

✔

—

—

✔

✔

fft.ifftshift

✔

✔

✔

✔

✔

fft.ihfft

✔

—

—

✔

✔

fft.ihfft2

✔

—

—

✔

✔

fft.ihfftn

✔

—

—

✔

✔

fft.irfft

✔

—

—

✔

✔

fft.irfft2

✔

—

—

✔

✔

fft.irfftn

✔

—

—

✔

✔

fft.rfft

✔

—

—

✔

✔

fft.rfft2

✔

—

—

✔

✔

fft.rfftn

✔

—

—

✔

✔

fill

✔

✔

✔

✔

✔

flatten

✔

✔

✔

✔

✔

flip

✔

✔

✔

✔

✔

fliplr

✔

✔

✔

✔

✔

flipud

✔

✔

✔

✔

✔

float

✔

✔

✔

✔

✔

float_power

✔

✔

✔

✔

✔

floor

✔

✔

✔

⚠ xfail

⚠ xfail

floor_divide

✔

✔

✔

⚠ xfail

⚠ xfail

fmax

✔

✔

✔

✔

✔

fmin

✔

✔

✔

✔

✔

fmod

✔

✔

✔

✔

✔

frac

✔

✔

✔

—

—

frexp

✔

✔

⚠ xfail

—

—

full

✔

✔

✔

✔

✔

full_like

✔

✔

✔

✔

✔

gather

✔

✔

✔

✔

✔

gcd

—

—

—

⚠ xfail

⚠ xfail

ge

✔

✔

✔

✔

✔

geqrf

✔

—

—

—

—

gradient

✔

✔

✔

✔

✔

grid_sampler_2d

✔

✔

✔

—

—

grid_sampler_3d

✔

✔

✔

—

—

gt

✔

✔

✔

✔

✔

half

✔

✔

✔

✔

✔

hash_tensor

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

heaviside

✔

✔

✔

✔

✔

histc

✔

✔

✔

—

—

histogram

✔

—

—

—

—

histogramdd

✔

—

—

—

—

hsplit

✔

✔

✔

✔

✔

hstack

✔

✔

✔

✔

✔

hypot

✘ no converter

✘ no converter

✘ no converter

—

—

i0

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

igamma

✘ no converter

✘ no converter

✘ no converter

—

—

igammac

✘ no converter

✘ no converter

✘ no converter

—

—

imag

—

—

—

—

—

index_add

✔

✔

✔

✔

✔

index_copy

✔

✔

✔

✔

✔

index_fill

✔

✔

✔

✔

✔

index_put

✔

✔

✔

✔

✔

index_select

✔

✔

✔

✔

✔

inner

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

int

✔

✔

✔

✔

✔

isclose

✔

✔

✔

✔

✔

isfinite

✔

✔

✔

✔

✔

isin

✔

✔

✔

✔

✔

isinf

✔

✔

✔

⚠ xfail

⚠ xfail

isnan

✔

✔

✔

⚠ xfail

⚠ xfail

isneginf

✔

✔

✔

✔

✔

isposinf

✔

✔

✔

✔

✔

isreal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

istft

—

—

—

—

—

item

✔

✔

✔

✔

✔

jiterator_2inputs_2outputs

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_4inputs_with_extra_args

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary_return_by_ref

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_unary

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

kron

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

kthvalue

✔

✔

✔

✔

✔

lcm

—

—

—

⚠ xfail

⚠ xfail

ldexp

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

le

✔

✔

✔

✔

✔

lerp

✔

✔

✔

—

—

lgamma

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

linalg.cholesky

✔

—

—

—

—

linalg.cholesky_ex

✔

—

—

—

—

linalg.cond

✘ no converter

—

—

—

—

linalg.cross

✔

✔

✔

✔

✔

linalg.det

✔

—

—

—

—

linalg.diagonal

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

linalg.eig

✔

—

—

—

—

linalg.eigh

✔

—

—

—

—

linalg.eigvals

✔

—

—

—

—

linalg.eigvalsh

✔

—

—

—

—

linalg.householder_product

✘ no converter

—

—

—

—

linalg.inv

✘ no converter

—

—

—

—

linalg.inv_ex

✘ no converter

—

—

—

—

linalg.ldl_factor

✔

—

—

—

—

linalg.ldl_factor_ex

✔

—

—

—

—

linalg.ldl_solve

✔

—

—

—

—

linalg.lstsq

✔

—

—

—

—

linalg.lu

✔

—

—

—

—

linalg.lu_factor

✔

—

—

—

—

linalg.lu_factor_ex

✔

—

—

—

—

linalg.lu_solve

✔

—

—

—

—

linalg.matrix_norm

✔

✔

✔

—

—

linalg.matrix_power

✔

—

—

—

—

linalg.matrix_rank

✔

—

—

—

—

linalg.matrix_sqrth

✔

—

—

—

—

linalg.multi_dot

✔

✔

✔

✔

✔

linalg.norm

✔

✔

✔

—

—

linalg.pinv

✔

—

—

—

—

linalg.polar

✔

—

—

—

—

linalg.qr

✘ no converter

—

—

—

—

linalg.slogdet

✔

—

—

—

—

linalg.solve

✘ no converter

—

—

—

—

linalg.solve_ex

✘ no converter

—

—

—

—

linalg.solve_triangular

✔

—

—

—

—

linalg.svd

✔

—

—

—

—

linalg.svdvals

✘ no converter

—

—

—

—

linalg.tensorinv

✔

—

—

—

—

linalg.tensorsolve

✔

—

—

—

—

linalg.vander

✔

—

—

✔

✔

linalg.vecdot

✔

✔

✔

—

—

linalg.vector_norm

✔

✔

✔

—

—

linspace

✔

✔

✔

✔

✔

log

✔

✔

✔

⚠ xfail

⚠ xfail

log10

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log1p

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log2

✔

✔

⚠ xfail

⚠ xfail

⚠ xfail

log_softmax

✔

✔

✔

—

—

logaddexp

✔

✔

⚠ xfail

—

—

logaddexp2

✔

✔

⚠ xfail

—

—

logcumsumexp

✔

✔

✔

—

—

logdet

✔

—

—

—

—

logical_and

✔

✔

✔

✔

✔

logical_not

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

logical_or

✔

✔

✔

✔

✔

logical_xor

✔

✔

✔

✔

✔

logit

✔

✔

✔

⚠ xfail

⚠ xfail

logspace

✔

✔

✔

✔

✔

logsumexp

✔

✔

✔

✔

✔

long

✔

✔

✔

✔

✔

lt

✔

✔

✔

✔

✔

lu

✔

—

—

—

—

lu_solve

✘ no converter

—

—

—

—

lu_unpack

✘ no converter

—

—

—

—

mH

✔

✔

✔

✔

✔

mT

✔

✔

✔

✔

✔

masked.amax

✔

✔

✔

✔

✔

masked.amin

✔

✔

✔

✔

✔

masked.argmax

✔

✔

✔

✔

✔

masked.argmin

✔

✔

✔

✔

✔

masked.cumprod

✔

✔

✔

✔

✔

masked.cumsum

✔

✔

✔

✔

✔

masked.log_softmax

✔

✔

✔

—

—

masked.logaddexp

✔

✔

✔

—

—

masked.logsumexp

✔

✔

✔

✔

✔

masked.mean

✔

✔

✔

—

—

masked.median

✔

✔

✔

—

—

masked.norm

✔

✔

✔

—

—

masked.normalize

✔

✔

✔

—

—

masked.prod

✔

✔

✔

✔

✔

masked.softmax

✔

✔

✔

—

—

masked.softmin

✔

✔

✔

—

—

masked.std

✔

✔

✔

✔

✔

masked.sum

✔

✔

✔

✔

✔

masked.var

✔

✔

✔

✔

✔

masked_fill

✔

✔

✔

✔

✔

masked_scatter

✔

✔

✔

✔

✔

masked_select

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

matmul

✔

✔

✔

✔

✔

matrix_exp

✘ no converter

✘ no converter

✘ no converter

—

—

max_pool2d_with_indices_backward

✔

✔

✔

—

—

maximum

✔

✔

✔

✔

✔

mean

✔

✔

✔

—

—

median

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

minimum

✔

✔

✔

✔

✔

mm

✔

✔

✔

✔

✔

mode

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

movedim

✔

✔

✔

✔

✔

msort

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

mul

✔

✔

✔

✔

✔

multinomial

✔

✔

✔

—

—

mv

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

nan_to_num

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

⚠ xfail

nanmean

✔

✔

✔

—

—

nanmedian

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

nanquantile

✔

—

—

—

—

nansum

✔

✔

✔

⚠ xfail

⚠ xfail

narrow

✔

✔

✔

✔

✔

narrow_copy

✔

✔

✔

✔

✔

native_batch_norm

✔

✔

✔

—

—

native_dropout_backward

✔

✔

✔

✔

✔

native_group_norm

✔

✔

✔

—

—

native_layer_norm

✔

✔

✔

—

—

ne

✔

✔

✔

✔

✔

neg

✔

✔

✔

✔

✔

new_empty

✔

✔

✔

✔

✔

new_empty_strided

✔

✔

✔

✔

✔

new_full

✔

✔

✔

✔

✔

new_ones

✔

✔

✔

✔

✔

new_zeros

✔

✔

✔

✔

✔

nextafter

✘ no converter

✘ no converter

✘ no converter

—

—

nn.functional.adaptive_avg_pool1d

✔

✔

✔

—

—

nn.functional.adaptive_avg_pool2d

✔

✔

✔

—

—

nn.functional.adaptive_avg_pool3d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool1d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool2d

✔

✔

✔

—

—

nn.functional.adaptive_max_pool3d

✔

✔

✔

—

—

nn.functional.alpha_dropout

✔

✔

✔

—

—

nn.functional.avg_pool1d

✔

✔

✔

—

✔

nn.functional.avg_pool2d

✔

✔

✔

—

✔

nn.functional.avg_pool3d

✔

—

—

—

✔

nn.functional.batch_norm

✔

✔

✔

—

—

nn.functional.bilinear

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.binary_cross_entropy

✔

✔

✔

—

—

nn.functional.binary_cross_entropy_with_logits

✔

✔

✔

—

—

nn.functional.celu

✔

✔

✔

—

—

nn.functional.channel_shuffle

✔

✔

✔

✔

✔

nn.functional.conv1d

✔

✔

✔

—

✔

nn.functional.conv2d

✔

✔

✔

—

✔

nn.functional.conv3d

✔

✔

✔

—

✔

nn.functional.conv_transpose1d

✔

✔

✔

—

✔

nn.functional.conv_transpose2d

✔

✔

✔

—

✔

nn.functional.conv_transpose3d

✔

✔

✔

—

✔

nn.functional.cosine_embedding_loss

✔

✔

✔

✔

✔

nn.functional.cosine_similarity

✔

✔

✔

—

—

nn.functional.cross_entropy

✔

✔

✔

—

—

nn.functional.ctc_loss

✔

—

—

—

—

nn.functional.dropout

✔

✔

✔

—

—

nn.functional.dropout2d

✔

✔

✔

—

—

nn.functional.dropout3d

✔

✔

✔

—

—

nn.functional.elu

✔

✔

✔

—

—

nn.functional.embedding

✔

✔

✔

—

—

nn.functional.embedding_bag

✔

✔

✔

—

—

nn.functional.fractional_max_pool2d

✔

✔

✔

—

—

nn.functional.fractional_max_pool3d

✔

✔

✔

—

—

nn.functional.gaussian_nll_loss

✔

✔

✔

—

—

nn.functional.gelu

✔

✔

✔

—

—

nn.functional.glu

✔

✔

✔

—

—

nn.functional.grid_sample

✔

✔

✔

—

—

nn.functional.group_norm

✔

✔

✔

—

—

nn.functional.hardshrink

✔

✔

✔

—

—

nn.functional.hardsigmoid

✔

✔

✔

—

—

nn.functional.hardswish

✔

✔

✔

—

—

nn.functional.hardtanh

✔

✔

✔

✔

✔

nn.functional.hinge_embedding_loss

✔

✔

✔

—

—

nn.functional.huber_loss

✔

✔

✔

—

—

nn.functional.instance_norm

✔

✔

✔

—

—

nn.functional.kl_div

✔

✔

✔

—

—

nn.functional.l1_loss

✔

✔

✔

—

—

nn.functional.layer_norm

✔

✔

✔

—

—

nn.functional.leaky_relu

✔

✔

✔

—

—

nn.functional.linear

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.linear_cross_entropy

⚠ xfail

⚠ xfail

⚠ xfail

—

—

nn.functional.local_response_norm

✔

✔

✔

—

✔

nn.functional.logsigmoid

✔

✔

✔

—

—

nn.functional.margin_ranking_loss

✔

✔

✔

✔

✔

nn.functional.max_pool1d

✔

✔

✔

—

—

nn.functional.max_pool2d

✔

✔

✔

✔

✔

nn.functional.max_pool3d

✔

✔

✔

✔

✔

nn.functional.max_unpool1d

✔

✔

✔

—

—

nn.functional.max_unpool2d

✔

✔

✔

—

—

nn.functional.max_unpool3d

✔

✔

✔

—

—

nn.functional.mish

✔

✔

✔

—

—

nn.functional.mse_loss

✔

✔

✔

—

—

nn.functional.multi_head_attention_forward

✔

✔

✔

—

—

nn.functional.multi_margin_loss

✘ no converter

—

—

—

—

nn.functional.multilabel_margin_loss

✘ no converter

—

—

—

—

nn.functional.multilabel_soft_margin_loss

✘ no converter

✘ no converter

✘ no converter

—

—

nn.functional.nll_loss

✔

✔

✔

—

—

nn.functional.normalize

✔

✔

✔

—

—

nn.functional.one_hot

—

—

—

—

✔

nn.functional.pairwise_distance

✔

✔

✔

✔

✔

nn.functional.pdist

✘ no converter

—

—

—

—

nn.functional.pixel_shuffle

✔

✔

✔

✔

✔

nn.functional.pixel_unshuffle

✔

✔

✔

✔

✔

nn.functional.poisson_nll_loss

✔

✔

✔

✔

✔

nn.functional.prelu

✔

✔

✔

—

—

nn.functional.relu

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.relu6

✔

✔

✔

✔

✔

nn.functional.rms_norm

✔

✔

✔

—

—

nn.functional.rrelu

✔

—

✔

—

—

nn.functional.scaled_dot_product_attention

✔

✔

✔

—

—

nn.functional.selu

✔

✔

✔

—

—

nn.functional.silu

✔

✔

✔

—

—

nn.functional.smooth_l1_loss

✔

✔

✔

—

—

nn.functional.soft_margin_loss

✔

⚠ xfail

✔

—

—

nn.functional.softmin

✔

✔

✔

—

—

nn.functional.softplus

✔

✔

✔

—

—

nn.functional.softshrink

✔

✔

✔

—

—

nn.functional.softsign

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.tanhshrink

✔

✔

✔

⚠ xfail

⚠ xfail

nn.functional.threshold

✔

✔

✔

✔

✔

nn.functional.triplet_margin_loss

✔

✔

✔

✔

✔

nn.functional.triplet_margin_with_distance_loss

✔

✔

✔

✔

✔

nn.functional.unfold

✔

✔

✔

—

—

nn.functional.upsample_bilinear

✔

✔

✔

—

—

nn.functional.upsample_nearest

✔

✔

✔

—

—

nonzero

✔

✔

✔

✔

✔

nonzero_static

✔

✔

✔

✔

✔

norm

✔

✔

✔

—

—

ones

✔

✔

✔

✔

✔

ones_like

✔

✔

✔

✔

✔

ormqr

✔

—

—

—

—

outer

✔

✔

✔

✔

✔

pca_lowrank

✔

—

—

—

—

permute

✔

✔

✔

✔

✔

permute_copy

✔

✔

✔

✔

✔

pinverse

✘ no converter

—

—

—

—

polar

✔

✔

—

—

—

positive

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

pow

✔

✔

✔

✔

✔

prod

✔

✔

✔

⚠ xfail

✔

put

✔

✔

✔

✔

✔

quantile

✔

—

—

—

—

rad2deg

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

randint_like

✔

✔

✔

✔

✔

ravel

✔

✔

✔

✔

✔

real

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

reciprocal

✔

✔

✔

⚠ xfail

⚠ xfail

remainder

✔

✔

✔

✔

✔

renorm

✔

✔

✔

—

—

repeat

✔

✔

✔

✔

✔

repeat_interleave

✔

✔

✔

✔

✔

reshape

✔

✔

✔

✔

✔

reshape_as

✔

✔

✔

✔

✔

resize_

✔

✔

✔

✔

✔

resize_as_

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

resolve_conj

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

resolve_neg

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

roll

✔

✔

✔

✔

✔

rot90

✔

✔

✔

✔

✔

round

✔

✔

✔

⚠ xfail

⚠ xfail

rsqrt

✔

✔

✔

⚠ xfail

⚠ xfail

rsub

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

scalar_tensor

✔

✔

✔

✔

✔

scatter

✔

✔

✔

✔

✔

scatter_add

✔

✔

✔

✔

✔

searchsorted

✔

✔

✔

✔

✔

select

✔

✔

✔

✔

✔

select_scatter

✔

✔

✔

✔

✔

sgn

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

short

✔

✔

✔

✔

✔

sigmoid

✔

✔

✔

⚠ xfail

⚠ xfail

sign

✔

✔

✔

✔

✔

signal.windows.bartlett

✔

—

—

—

—

signal.windows.blackman

✔

—

—

—

—

signal.windows.cosine

✔

—

—

—

—

signal.windows.exponential

✔

—

—

—

—

signal.windows.gaussian

✔

—

—

—

—

signal.windows.general_cosine

✔

—

—

—

—

signal.windows.general_hamming

✔

—

—

—

—

signal.windows.hamming

✔

—

—

—

—

signal.windows.hann

✔

—

—

—

—

signal.windows.kaiser

✔

—

—

—

—

signal.windows.nuttall

✔

—

—

—

—

signbit

✔

✔

✔

✔

✔

sin

✔

✔

✔

⚠ xfail

⚠ xfail

sinc

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

sinh

✔

⚠ xfail

✔

⚠ xfail

⚠ xfail

slice

✔

✔

✔

✔

✔

slice_scatter

✔

✔

✔

✔

✔

softmax

✔

✔

✔

—

—

sort

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

sparse.sampled_addmm

✔

—

—

—

—

special.airy_ai

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_j0

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_j1

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_y0

✘ no converter

—

—

✘ no converter

✘ no converter

special.bessel_y1

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_t

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_u

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_v

✘ no converter

—

—

✘ no converter

✘ no converter

special.chebyshev_polynomial_w

✘ no converter

—

—

✘ no converter

✘ no converter

special.entr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.erfcx

✘ no converter

—

—

✘ no converter

✘ no converter

special.hermite_polynomial_h

✘ no converter

—

—

✘ no converter

✘ no converter

special.hermite_polynomial_he

✘ no converter

—

—

✘ no converter

✘ no converter

special.i0e

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.i1

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.i1e

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.laguerre_polynomial_l

✘ no converter

—

—

✘ no converter

✘ no converter

special.legendre_polynomial_p

✘ no converter

—

—

✘ no converter

✘ no converter

special.log_ndtr

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_i0

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_i1

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_k0

✘ no converter

—

—

✘ no converter

✘ no converter

special.modified_bessel_k1

✘ no converter

—

—

✘ no converter

✘ no converter

special.ndtr

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.ndtri

✘ no converter

—

—

✘ no converter

✘ no converter

special.scaled_modified_bessel_k0

✘ no converter

—

—

✘ no converter

✘ no converter

special.scaled_modified_bessel_k1

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_t

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_u

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_v

✘ no converter

—

—

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_w

✘ no converter

—

—

✘ no converter

✘ no converter

special.spherical_bessel_j0

✘ no converter

—

—

✘ no converter

✘ no converter

special.xlog1py

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

special.zeta

✘ no converter

—

—

✘ no converter

✘ no converter

split

✔

✔

✔

✔

✔

split_with_sizes

✔

✔

✔

✔

✔

split_with_sizes_copy

✔

✔

✔

✔

✔

sqrt

✔

✔

✔

⚠ xfail

⚠ xfail

square

✔

✔

✔

✔

✔

squeeze

✔

✔

✔

✔

✔

squeeze_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

stack

✔

✔

✔

✔

✔

std

✔

✔

✔

—

—

std_mean

✔

✔

✔

—

—

stft

✔

—

—

—

—

sub

✔

✔

✔

✔

✔

sum

✔

✔

✔

⚠ xfail

✔

sum_to_size

✔

✔

✔

✔

✔

svd

✔

—

—

—

—

svd_lowrank

✔

—

—

—

—

t

✔

✔

✔

✔

✔

t_copy

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

take

✔

✔

✔

✔

✔

take_along_dim

✔

✔

✔

✔

✔

tan

✔

✔

✔

⚠ xfail

⚠ xfail

tanh

✔

✔

✔

⚠ xfail

⚠ xfail

tensor_split

✔

✔

✔

✔

✔

tensordot

✔

✔

✔

✔

✔

tile

✔

✔

✔

✔

✔

to

✔

✔

✔

✔

✔

to_sparse

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

topk

✔

✔

✔

✔

✔

torch._scaled_mm

—

—

—

—

—

torch._scaled_mm_v2

—

—

—

—

—

torch.ops.aten._efficient_attention_forward

—

—

—

—

—

torch.ops.aten._flash_attention_forward

—

—

—

—

—

torch.ops.aten._safe_softmax.default

✔

✔

✔

✔

✔

torch.ops.aten._scaled_dot_product_flash_attention_for_cpu

✔

✔

✔

—

—

trace

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

transpose

✔

✔

✔

✔

✔

transpose_copy

✔

✔

✔

✔

✔

trapezoid

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

trapz

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

triangular_solve

✘ no converter

—

—

—

—

tril

✔

✔

✔

⚠ xfail

✔

tril_indices

—

—

—

✔

✔

triu

✔

✔

✔

⚠ xfail

✔

triu_indices

—

—

—

✔

✔

true_divide

✔

✔

✔

✔

✔

trunc

✔

✔

✔

⚠ xfail

⚠ xfail

unbind

✔

✔

✔

✔

✔

unbind_copy

✔

✔

✔

✔

✔

unflatten

✔

✔

✔

✔

✔

unfold

✔

✔

✔

✔

✔

unfold_copy

✔

✔

✔

✔

✔

unique

✔

✔

✔

✔

✔

unique_consecutive

✔

✔

✔

✔

✔

unravel_index

—

—

—

✔

✔

unsafe_chunk

✔

✔

✔

✔

✔

unsafe_split

✔

✔

✔

✔

✔

unsqueeze

✔

✔

✔

✔

✔

unsqueeze_copy

✔

✔

✔

✔

✔

var

✘ no converter

✘ no converter

✘ no converter

—

—

var_mean

✘ no converter

✘ no converter

✘ no converter

—

—

vdot

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

view

✔

✔

✔

✔

✔

view_as

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

view_as_complex

✔

✔

—

—

—

view_as_real

—

—

—

—

—

view_copy

✔

✔

✔

✔

✔

vsplit

✔

✔

✔

✔

✔

vstack

✔

✔

✔

✔

✔

where

✔

✔

✔

✔

✔

xlogy

✔

✔

✔

⚠ xfail

⚠ xfail

zero_

✘ no converter

✘ no converter

✘ no converter

✘ no converter

✘ no converter

zeros

✔

✔

✔

✔

✔

zeros_like

✔

✔

✔

✔

✔

Float32 comparison across export paths#

This section compares, for every op that supports float32, whether it passes, is a known failure (xfail), or has no converter for the three export paths side by side.

Float32 op support: default vs tracing vs new-tracing

Op

Default

Tracing

New-tracing

H

✘ no converter

✘ no converter

✘ no converter

T

✔

✔

✔

__getitem__

✔

✔

✔

__radd__

✔

⚠ xfail

✔

__rdiv__

✔

⚠ xfail

✔

__rmatmul__

✔

✔

✔

__rmod__

✔

✔

✔

__rmul__

✔

⚠ xfail

✔

__rpow__

✔

✔

✔

__rsub__

✘ no converter

✘ no converter

✘ no converter

_batch_norm_with_update

✔

✔

✔

_chunk_cat

✔

✔

✔

_native_batch_norm_legit

✔

✔

✔

_softmax_backward_data

✔

✔

✔

_unsafe_masked_index

✔

✔

✔

_unsafe_masked_index_put_accumulate

✔

✔

✔

_upsample_bilinear2d_aa

✔

✔

✔

abs

✔

✔

✔

acos

✔

✔

✔

acosh

✔

✔

✔

add

✔

✔

✔

addbmm

✔

✔

✔

addcdiv

✘ no converter

✘ no converter

✘ no converter

addcmul

✔

✔

✔

addmm

✔

✔

✔

addmv

✔

✔

✔

addr

✘ no converter

✘ no converter

✘ no converter

alias_copy

✘ no converter

✘ no converter

✘ no converter

all

✔

✔

✔

allclose

✔

✔

✔

amax

✔

✔

✔

amin

✔

✔

✔

aminmax

✔

✔

✔

angle

✔

✔

✔

any

✔

✔

✔

arange

✔

✔

✔

argmax

✔

✔

✔

argmin

✔

✔

✔

argsort

✔

✔

⚠ xfail

argwhere

✔

✔

✔

as_strided

✔

✔

✔

as_strided_copy

✔

✔

✔

as_strided_scatter

✔

✔

✔

asin

✔

✔

✔

asinh

✔

✔

✔

atan

✔

✔

✔

atan2

✔

✔

✔

atanh

✔

✔

✔

atleast_1d

✔

✔

✔

atleast_2d

✔

✔

✔

atleast_3d

✔

✔

✔

baddbmm

✔

✔

✔

bernoulli

✘ no converter

✘ no converter

✘ no converter

bfloat16

⚠ xfail

⚠ xfail

⚠ xfail

block_diag

✔

✔

✔

bmm

✔

✔

✔

bool

✔

✔

✔

broadcast_shapes

✔

✔

✔

broadcast_tensors

⚠ xfail

⚠ xfail

⚠ xfail

broadcast_to

✔

✔

✔

bucketize

✔

✔

✔

byte

✔

✔

✔

cartesian_prod

✔

✔

✔

cat

✔

✔

✔

cdist

✔

✔

✔

cdouble

✔

✔

✔

ceil

✔

✔

✔

cfloat

✔

✔

✔

chalf

✔

✔

✔

char

✔

✔

⚠ xfail

cholesky_inverse

⚠ xfail

⚠ xfail

⚠ xfail

cholesky_solve

⚠ xfail

⚠ xfail

⚠ xfail

chunk

✔

✔

✔

clamp

⚠ xfail

⚠ xfail

⚠ xfail

clamp_max

✔

✔

✔

clamp_min

✔

✔

✔

clone

✔

✔

✔

column_stack

✔

✔

✔

combinations

✔

✔

✔

complex

✔

✔

✔

conj

✔

⚠ xfail

✔

conj_physical

✘ no converter

✘ no converter

✘ no converter

constant_pad_nd

✔

✔

✔

contiguous

✔

✔

✔

copysign

✘ no converter

✘ no converter

✘ no converter

corrcoef

⚠ xfail

⚠ xfail

⚠ xfail

cos

✔

✔

✔

cosh

✔

✔

✔

count_nonzero

✔

✔

✔

cov

⚠ xfail

⚠ xfail

⚠ xfail

cross

✘ no converter

✘ no converter

✘ no converter

cummax

✔

✔

✔

cummin

✔

✔

✔

cumprod

✔

✔

✔

cumsum

✔

✔

✔

cumulative_trapezoid

✘ no converter

✘ no converter

✘ no converter

deg2rad

✘ no converter

✘ no converter

✘ no converter

diag

✔

✔

✔

diag_embed

✔

✔

✔

diagflat

✘ no converter

✘ no converter

✘ no converter

diagonal

✘ no converter

✘ no converter

✘ no converter

diagonal_copy

✘ no converter

✘ no converter

✘ no converter

diagonal_scatter

✘ no converter

✘ no converter

✘ no converter

diff

✔

✔

✔

digamma

✘ no converter

✘ no converter

✘ no converter

dist

✔

✔

✔

dot

✘ no converter

✘ no converter

✘ no converter

double

✔

✔

✔

dsplit

✔

✔

✔

dstack

✔

✔

✔

einsum

✔

✔

✔

empty_permuted

✔

✔

✔

empty_strided

✔

✔

✔

eq

✔

✔

✔

equal

✔

✔

✔

erf

✔

✔

✔

erfc

✔

✔

✔

erfinv

✔

✔

⚠ xfail

exp

✔

✔

✔

exp2

✘ no converter

✘ no converter

✘ no converter

expand

✔

✔

✔

expand_as

✔

✔

✔

expand_copy

✔

✔

✔

expm1

✔

✔

✔

eye

✔

✔

✔

fft.fft

✔

✔

✔

fft.fft2

✔

✔

✔

fft.fftn

✔

✔

✔

fft.fftshift

✔

✔

✔

fft.hfft

✔

✔

✔

fft.hfft2

✔

✔

✔

fft.hfftn

✔

✔

✔

fft.ifft

✔

✔

✔

fft.ifft2

✔

✔

✔

fft.ifftn

✔

✔

✔

fft.ifftshift

✔

✔

✔

fft.ihfft

✔

✔

✔

fft.ihfft2

✔

✔

✔

fft.ihfftn

✔

✔

✔

fft.irfft

✔

✔

✔

fft.irfft2

✔

✔

✔

fft.irfftn

✔

✔

✔

fft.rfft

✔

✔

✔

fft.rfft2

✔

✔

✔

fft.rfftn

✔

✔

✔

fill

✔

✔

✔

flatten

✔

⚠ xfail

✔

flip

✔

✔

✔

fliplr

✔

✔

✔

flipud

✔

✔

✔

float

✔

✔

✔

float_power

✔

✔

✔

floor

✔

✔

✔

floor_divide

✔

✔

✔

fmax

✔

✔

✔

fmin

✔

✔

✔

fmod

✔

✔

✔

frac

✔

✔

✔

frexp

✔

✔

✔

full

✔

✔

✔

full_like

✔

✔

✔

gather

✔

✔

✔

ge

✔

✔

✔

geqrf

✔

✔

✔

gradient

✔

✔

✔

grid_sampler_2d

✔

✔

✔

grid_sampler_3d

✔

✔

✔

gt

✔

✔

✔

half

✔

✔

⚠ xfail

hash_tensor

✘ no converter

✘ no converter

✘ no converter

heaviside

✔

✔

✔

histc

✔

✔

✔

histogram

✔

✔

✔

histogramdd

✔

✔

✔

hsplit

✔

✔

✔

hstack

✔

✔

✔

hypot

✘ no converter

✘ no converter

✘ no converter

i0

✘ no converter

✘ no converter

✘ no converter

igamma

✘ no converter

✘ no converter

✘ no converter

igammac

✘ no converter

✘ no converter

✘ no converter

index_add

✔

✔

✔

index_copy

✔

✔

✔

index_fill

✔

✔

✔

index_put

✔

✔

✔

index_select

✔

✔

✔

inner

✘ no converter

✘ no converter

✘ no converter

int

✔

✔

⚠ xfail

isclose

✔

✔

⚠ xfail

isfinite

✔

✔

⚠ xfail

isin

✔

✔

✔

isinf

✔

✔

✔

isnan

✔

✔

✔

isneginf

✔

✔

✔

isposinf

✔

✔

✔

isreal

✘ no converter

✘ no converter

✘ no converter

item

✔

✔

✔

jiterator_2inputs_2outputs

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_4inputs_with_extra_args

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_binary_return_by_ref

⚠ xfail

⚠ xfail

⚠ xfail

jiterator_unary

⚠ xfail

⚠ xfail

⚠ xfail

kron

✘ no converter

✘ no converter

✘ no converter

kthvalue

✔

✔

✔

ldexp

✘ no converter

✘ no converter

✘ no converter

le

✔

✔

✔

lerp

✔

✔

✔

lgamma

✘ no converter

✘ no converter

✘ no converter

linalg.cholesky

✔

✔

✔

linalg.cholesky_ex

✔

✔

✔

linalg.cond

✘ no converter

✘ no converter

✘ no converter

linalg.cross

✔

✔

✔

linalg.det

✔

✔

✔

linalg.diagonal

✘ no converter

✘ no converter

✘ no converter

linalg.eig

✔

✔

✔

linalg.eigh

✔

✔

✔

linalg.eigvals

✔

✔

✔

linalg.eigvalsh

✔

✔

✔

linalg.householder_product

✘ no converter

✘ no converter

✘ no converter

linalg.inv

✘ no converter

✘ no converter

✘ no converter

linalg.inv_ex

✘ no converter

✘ no converter

✘ no converter

linalg.ldl_factor

✔

✔

✔

linalg.ldl_factor_ex

✔

✔

✔

linalg.ldl_solve

✔

✔

✔

linalg.lstsq

✔

✔

✔

linalg.lu

✔

✔

✔

linalg.lu_factor

✔

✔

✔

linalg.lu_factor_ex

✔

✔

✔

linalg.lu_solve

✔

✔

✔

linalg.matrix_norm

✔

✔

✔

linalg.matrix_power

✔

✔

✔

linalg.matrix_rank

✔

✔

✔

linalg.matrix_sqrth

✔

✔

✔

linalg.multi_dot

✔

✔

✔

linalg.norm

✔

✔

✔

linalg.pinv

✔

✔

✔

linalg.polar

✔

✔

✔

linalg.qr

✘ no converter

✘ no converter

✘ no converter

linalg.slogdet

✔

✔

✔

linalg.solve

✘ no converter

✘ no converter

✘ no converter

linalg.solve_ex

✘ no converter

✘ no converter

✘ no converter

linalg.solve_triangular

✔

✔

✔

linalg.svd

✔

✔

✔

linalg.svdvals

✘ no converter

✘ no converter

✘ no converter

linalg.tensorinv

✔

✔

✔

linalg.tensorsolve

✔

✔

✔

linalg.vander

✔

✔

✔

linalg.vecdot

✔

✔

✔

linalg.vector_norm

✔

✔

✔

linspace

✔

✔

✔

log

✔

✔

✔

log10

✔

✔

✔

log1p

✔

✔

✔

log2

✔

✔

✔

log_softmax

✔

✔

✔

logaddexp

✔

✔

✔

logaddexp2

✔

✔

✔

logcumsumexp

✔

✔

✔

logdet

✔

✔

✔

logical_and

✔

⚠ xfail

✔

logical_not

⚠ xfail

⚠ xfail

⚠ xfail

logical_or

✔

⚠ xfail

✔

logical_xor

✔

⚠ xfail

✔

logit

✔

✔

✔

logspace

✔

✔

✔

logsumexp

✔

✔

✔

long

✔

✔

⚠ xfail

lt

✔

✔

✔

lu

✔

✔

✔

lu_solve

✘ no converter

✘ no converter

✘ no converter

lu_unpack

✘ no converter

✘ no converter

✘ no converter

mH

✔

✔

✔

mT

✔

✔

✔

masked.amax

✔

✔

✔

masked.amin

✔

✔

✔

masked.argmax

✔

✔

✔

masked.argmin

✔

✔

✔

masked.cumprod

✔

✔

✔

masked.cumsum

✔

✔

✔

masked.log_softmax

✔

✔

✔

masked.logaddexp

✔

✔

✔

masked.logsumexp

✔

✔

✔

masked.mean

✔

✔

✔

masked.median

✔

✔

✔

masked.norm

✔

✔

✔

masked.normalize

✔

✔

✔

masked.prod

✔

✔

✔

masked.softmax

✔

✔

✔

masked.softmin

✔

✔

✔

masked.std

✔

✔

✔

masked.sum

✔

✔

✔

masked.var

✔

✔

✔

masked_fill

✔

✔

✔

masked_scatter

✔

✔

✔

masked_select

✘ no converter

✘ no converter

✘ no converter

matmul

✔

✔

✔

matrix_exp

✘ no converter

✘ no converter

✘ no converter

max_pool2d_with_indices_backward

✔

✔

✔

maximum

✔

✔

✔

mean

✔

✔

✔

median

✘ no converter

✘ no converter

✘ no converter

minimum

✔

✔

✔

mm

✔

✔

✔

mode

✘ no converter

✘ no converter

✘ no converter

movedim

✔

✔

✔

msort

✘ no converter

✘ no converter

✘ no converter

mul

✔

✔

✔

multinomial

✔

✔

✔

mv

✘ no converter

✘ no converter

✘ no converter

nan_to_num

⚠ xfail

⚠ xfail

⚠ xfail

nanmean

✔

✔

✔

nanmedian

✘ no converter

✘ no converter

✘ no converter

nanquantile

✔

✔

✔

nansum

✔

✔

✔

narrow

✔

✔

✔

narrow_copy

✔

✔

✔

native_batch_norm

✔

✔

✔

native_dropout_backward

✔

✔

✔

native_group_norm

✔

✔

✔

native_layer_norm

✔

✔

✔

ne

✔

✔

✔

neg

✔

✔

✔

new_empty

✔

✔

✔

new_empty_strided

✔

✔

✔

new_full

✔

✔

✔

new_ones

✔

✔

✔

new_zeros

✔

✔

✔

nextafter

✘ no converter

✘ no converter

✘ no converter

nn.functional.adaptive_avg_pool1d

✔

✔

✔

nn.functional.adaptive_avg_pool2d

✔

✔

✔

nn.functional.adaptive_avg_pool3d

✔

✔

✔

nn.functional.adaptive_max_pool1d

✔

✔

✔

nn.functional.adaptive_max_pool2d

✔

✔

✔

nn.functional.adaptive_max_pool3d

✔

✔

✔

nn.functional.alpha_dropout

✔

✔

✔

nn.functional.avg_pool1d

✔

✔

✔

nn.functional.avg_pool2d

✔

✔

✔

nn.functional.avg_pool3d

✔

✔

✔

nn.functional.batch_norm

✔

✔

✔

nn.functional.bilinear

✔

⚠ xfail

✔

nn.functional.binary_cross_entropy

✔

✔

✔

nn.functional.binary_cross_entropy_with_logits

✔

✔

✔

nn.functional.celu

✔

✔

✔

nn.functional.channel_shuffle

✔

✔

✔

nn.functional.conv1d

✔

✔

✔

nn.functional.conv2d

✔

✔

✔

nn.functional.conv3d

✔

✔

✔

nn.functional.conv_transpose1d

✔

✔

✔

nn.functional.conv_transpose2d

✔

✔

✔

nn.functional.conv_transpose3d

✔

✔

✔

nn.functional.cosine_embedding_loss

✔

✔

✔

nn.functional.cosine_similarity

✔

✔

✔

nn.functional.cross_entropy

✔

✔

✔

nn.functional.ctc_loss

✔

✔

✔

nn.functional.dropout

✔

✔

✔

nn.functional.dropout2d

✔

✔

✔

nn.functional.dropout3d

✔

✔

✔

nn.functional.elu

✔

✔

✔

nn.functional.embedding

✔

✔

✔

nn.functional.embedding_bag

✔

✔

✔

nn.functional.fractional_max_pool2d

✔

✔

✔

nn.functional.fractional_max_pool3d

✔

✔

✔

nn.functional.gaussian_nll_loss

✔

✔

✔

nn.functional.gelu

✔

✔

✔

nn.functional.glu

✔

✔

✔

nn.functional.grid_sample

✔

✔

✔

nn.functional.group_norm

✔

✔

✔

nn.functional.hardshrink

✔

✔

✔

nn.functional.hardsigmoid

✔

⚠ xfail

✔

nn.functional.hardswish

✔

✔

✔

nn.functional.hardtanh

✔

✔

✔

nn.functional.hinge_embedding_loss

✔

✔

✔

nn.functional.huber_loss

✔

✔

✔

nn.functional.instance_norm

✔

✔

✔

nn.functional.kl_div

✔

✔

✔

nn.functional.l1_loss

✔

✔

✔

nn.functional.layer_norm

✔

✔

✔

nn.functional.leaky_relu

✔

✔

✔

nn.functional.linear

✔

✔

✔

nn.functional.linear_cross_entropy

⚠ xfail

⚠ xfail

⚠ xfail

nn.functional.local_response_norm

✔

✔

✔

nn.functional.logsigmoid

✔

✔

✔

nn.functional.margin_ranking_loss

✔

✔

✔

nn.functional.max_pool1d

✔

✔

✔

nn.functional.max_pool2d

✔

✔

✔

nn.functional.max_pool3d

✔

✔

✔

nn.functional.max_unpool1d

✔

✔

✔

nn.functional.max_unpool2d

✔

✔

✔

nn.functional.max_unpool3d

✔

✔

✔

nn.functional.mish

✔

✔

✔

nn.functional.mse_loss

✔

✔

✔

nn.functional.multi_head_attention_forward

✔

✔

✔

nn.functional.multi_margin_loss

✘ no converter

✘ no converter

✘ no converter

nn.functional.multilabel_margin_loss

✘ no converter

✘ no converter

✘ no converter

nn.functional.multilabel_soft_margin_loss

✘ no converter

✘ no converter

✘ no converter

nn.functional.nll_loss

✔

✔

✔

nn.functional.normalize

✔

✔

✔

nn.functional.pairwise_distance

✔

✔

✔

nn.functional.pdist

✘ no converter

✘ no converter

✘ no converter

nn.functional.pixel_shuffle

✔

✔

✔

nn.functional.pixel_unshuffle

✔

✔

✔

nn.functional.poisson_nll_loss

✔

✔

✔

nn.functional.prelu

✔

✔

✔

nn.functional.relu

✔

✔

✔

nn.functional.relu6

✔

✔

✔

nn.functional.rms_norm

✔

✔

✔

nn.functional.rrelu

✔

✔

✔

nn.functional.scaled_dot_product_attention

✔

✔

✔

nn.functional.selu

✔

✔

✔

nn.functional.silu

✔

✔

✔

nn.functional.smooth_l1_loss

✔

✔

✔

nn.functional.soft_margin_loss

✔

✔

✔

nn.functional.softmin

✔

✔

✔

nn.functional.softplus

✔

✔

✔

nn.functional.softshrink

✔

✔

✔

nn.functional.softsign

✔

✔

✔

nn.functional.tanhshrink

✔

✔

✔

nn.functional.threshold

✔

✔

✔

nn.functional.triplet_margin_loss

✔

✔

✔

nn.functional.triplet_margin_with_distance_loss

✔

✔

✔

nn.functional.unfold

✔

✔

✔

nn.functional.upsample_bilinear

✔

✔

✔

nn.functional.upsample_nearest

✔

✔

✔

nonzero

✔

✔

✔

nonzero_static

✔

✔

✔

norm

✔

✔

✔

ones

✔

✔

✔

ones_like

✔

✔

✔

ormqr

✔

✔

✔

outer

✔

✔

✔

pca_lowrank

✔

✔

✔

permute

✔

✔

✔

permute_copy

✔

✔

✔

pinverse

✘ no converter

✘ no converter

✘ no converter

polar

✔

✔

✔

positive

✘ no converter

✘ no converter

✘ no converter

pow

✔

✔

✔

prod

✔

✔

✔

put

✔

✔

✔

quantile

✔

✔

✔

rad2deg

✘ no converter

✘ no converter

✘ no converter

randint_like

✔

✔

✔

ravel

✔

✔

✔

real

✘ no converter

✘ no converter

✘ no converter

reciprocal

✔

✔

✔

remainder

✔

✔

✔

renorm

✔

✔

✔

repeat

✔

✔

✔

repeat_interleave

✔

✔

✔

reshape

✔

✔

✔

reshape_as

✔

✔

✔

resize_

✔

✔

✔

resize_as_

✘ no converter

✘ no converter

✘ no converter

resolve_conj

✘ no converter

✘ no converter

✘ no converter

resolve_neg

✘ no converter

✘ no converter

✘ no converter

roll

✔

✔

✔

rot90

✔

✔

✔

round

✔

✔

✔

rsqrt

✔

✔

✔

rsub

✘ no converter

✘ no converter

✘ no converter

scalar_tensor

✔

✔

✔

scatter

✔

✔

✔

scatter_add

✔

✔

✔

searchsorted

✔

✔

✔

select

✔

✔

✔

select_scatter

✔

✔

✔

sgn

✘ no converter

✘ no converter

✘ no converter

short

✔

✔

⚠ xfail

sigmoid

✔

✔

✔

sign

✔

✔

✔

signal.windows.bartlett

✔

✔

✔

signal.windows.blackman

✔

✔

✔

signal.windows.cosine

✔

✔

✔

signal.windows.exponential

✔

✔

✔

signal.windows.gaussian

✔

✔

✔

signal.windows.general_cosine

✔

✔

✔

signal.windows.general_hamming

✔

✔

✔

signal.windows.hamming

✔

✔

✔

signal.windows.hann

✔

✔

✔

signal.windows.kaiser

✔

✔

✔

signal.windows.nuttall

✔

✔

✔

signbit

✔

✔

✔

sin

✔

✔

✔

sinc

✘ no converter

✘ no converter

✘ no converter

sinh

✔

✔

✔

slice

✔

✔

✔

slice_scatter

✔

✔

✔

softmax

✔

✔

✔

sort

✘ no converter

✘ no converter

✘ no converter

sparse.sampled_addmm

✔

✔

✔

special.airy_ai

✘ no converter

✘ no converter

✘ no converter

special.bessel_j0

✘ no converter

✘ no converter

✘ no converter

special.bessel_j1

✘ no converter

✘ no converter

✘ no converter

special.bessel_y0

✘ no converter

✘ no converter

✘ no converter

special.bessel_y1

✘ no converter

✘ no converter

✘ no converter

special.chebyshev_polynomial_t

✘ no converter

✘ no converter

✘ no converter

special.chebyshev_polynomial_u

✘ no converter

✘ no converter

✘ no converter

special.chebyshev_polynomial_v

✘ no converter

✘ no converter

✘ no converter

special.chebyshev_polynomial_w

✘ no converter

✘ no converter

✘ no converter

special.entr

✘ no converter

✘ no converter

✘ no converter

special.erfcx

✘ no converter

✘ no converter

✘ no converter

special.hermite_polynomial_h

✘ no converter

✘ no converter

✘ no converter

special.hermite_polynomial_he

✘ no converter

✘ no converter

✘ no converter

special.i0e

✘ no converter

✘ no converter

✘ no converter

special.i1

✘ no converter

✘ no converter

✘ no converter

special.i1e

✘ no converter

✘ no converter

✘ no converter

special.laguerre_polynomial_l

✘ no converter

✘ no converter

✘ no converter

special.legendre_polynomial_p

✘ no converter

✘ no converter

✘ no converter

special.log_ndtr

✘ no converter

✘ no converter

✘ no converter

special.modified_bessel_i0

✘ no converter

✘ no converter

✘ no converter

special.modified_bessel_i1

✘ no converter

✘ no converter

✘ no converter

special.modified_bessel_k0

✘ no converter

✘ no converter

✘ no converter

special.modified_bessel_k1

✘ no converter

✘ no converter

✘ no converter

special.ndtr

✘ no converter

✘ no converter

✘ no converter

special.ndtri

✘ no converter

✘ no converter

✘ no converter

special.scaled_modified_bessel_k0

✘ no converter

✘ no converter

✘ no converter

special.scaled_modified_bessel_k1

✘ no converter

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_t

✘ no converter

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_u

✘ no converter

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_v

✘ no converter

✘ no converter

✘ no converter

special.shifted_chebyshev_polynomial_w

✘ no converter

✘ no converter

✘ no converter

special.spherical_bessel_j0

✘ no converter

✘ no converter

✘ no converter

special.xlog1py

✘ no converter

✘ no converter

✘ no converter

special.zeta

✘ no converter

✘ no converter

✘ no converter

split

✔

✔

✔

split_with_sizes

✔

✔

✔

split_with_sizes_copy

✔

✔

✔

sqrt

✔

✔

✔

square

✔

✔

✔

squeeze

✔

✔

✔

squeeze_copy

✘ no converter

✘ no converter

✘ no converter

stack

✔

✔

✔

std

✔

⚠ xfail

⚠ xfail

std_mean

✔

⚠ xfail

⚠ xfail

stft

✔

✔

✔

sub

✔

✔

✔

sum

✔

✔

✔

sum_to_size

✔

✔

✔

svd

✔

✔

✔

svd_lowrank

✔

✔

✔

t

✔

✔

✔

t_copy

✘ no converter

✘ no converter

✘ no converter

take

✔

✔

✔

take_along_dim

✔

✔

✔

tan

✔

✔

✔

tanh

✔

✔

✔

tensor_split

✔

✔

⚠ xfail

tensordot

✔

✔

✔

tile

✔

✔

✔

to

✔

✔

✔

to_sparse

✘ no converter

✘ no converter

✘ no converter

topk

✔

✔

✔

torch.ops.aten._safe_softmax.default

✔

✔

✔

torch.ops.aten._scaled_dot_product_flash_attention_for_cpu

✔

✔

✔

trace

✘ no converter

✘ no converter

✘ no converter

transpose

✔

✔

✔

transpose_copy

✔

✔

✔

trapezoid

✘ no converter

✘ no converter

✘ no converter

trapz

✘ no converter

✘ no converter

✘ no converter

triangular_solve

✘ no converter

✘ no converter

✘ no converter

tril

✔

✔

✔

triu

✔

✔

✔

true_divide

✔

✔

✔

trunc

✔

✔

✔

unbind

✔

✔

✔

unbind_copy

✔

✔

✔

unflatten

✔

✔

✔

unfold

✔

✔

✔

unfold_copy

✔

✔

✔

unique

✔

✔

✔

unique_consecutive

✔

✔

✔

unsafe_chunk

✔

✔

✔

unsafe_split

✔

✔

✔

unsqueeze

✔

✔

✔

unsqueeze_copy

✔

✔

✔

var

✘ no converter

✘ no converter

✘ no converter

var_mean

✘ no converter

✘ no converter

✘ no converter

vdot

✘ no converter

✘ no converter

✘ no converter

view

✔

✔

✔

view_as

✘ no converter

✘ no converter

✘ no converter

view_as_complex

✔

✔

✔

view_copy

✔

✔

✔

vsplit

✔

✔

✔

vstack

✔

✔

✔

where

✔

✔

✔

xlogy

✔

✔

✔

zero_

✘ no converter

✘ no converter

✘ no converter

zeros

✔

✔

✔

zeros_like

✔

✔

✔