xla_native_functions.yaml

backend: XLA
cpp_namespace: torch_xla
supported:
  - abs
  - abs_
  - acos
  - acos_
  - add.Tensor
  - add.Scalar
  - add_.Scalar
  - all.dim
  - any.dim
  - arange.start_out
  - argmax
  - argmin
  - acosh
  - acosh_
  - asinh
  - asinh_
  - atanh
  - atanh_
  - as_strided
  - as_strided_
  - asin
  - asin_
  - atan
  - atan_
  - baddbmm
  - baddbmm_
  - bernoulli
  - bernoulli_.Tensor
  - bernoulli_.float
  - binary_cross_entropy
  - binary_cross_entropy_backward
  - binary_cross_entropy_with_logits
  - bitwise_not.out
  - bmm
  - cat
  - ceil
  - ceil_
  - clamp
  - clamp.Tensor
  - clamp_
  - clamp_max
  - clamp_max.Tensor_out
  - clamp_max_
  - clamp_min
  - clamp_min.Tensor_out
  - clamp_min_
  - constant_pad_nd
  - convolution_overrideable
  - convolution_backward_overrideable
  - _copy_from
  - _copy_from_and_resize
  - _to_cpu
  - cos
  - cos_
  - cosh
  - cosh_
  - cumprod
  - cumsum
  - diagonal
  - div.Tensor
  - div_.Tensor
  - div.Tensor_mode
  - div_.Tensor_mode
  - div.Scalar
  - div_.Scalar
  - dot
  - embedding
  - embedding_dense_backward
  - empty.memory_format
  - resize_
  - empty_strided
  - erf
  - erf_
  - erfc
  - erfc_
  - exp
  - exp_
  - expm1
  - expm1_
  - expand
  - eye.out
  - eye.m_out
  - fill_.Scalar
  - fill_.Tensor
  - floor
  - floor_
  - frac
  - frac_
  - index.Tensor
  - index_copy_
  - index_put_
  - _index_put_impl_
  - inverse
  - kl_div
  - kl_div_backward
  - kthvalue
  - log
  - log_
  - log10
  - log10_
  - log1p
  - log1p_
  - log2
  - log2_
  - logdet
  - _log_softmax
  - _log_softmax_backward_data
  - logsumexp
  - max.dim
  - max.dim_max
  - mean
  - mean.dim
  - min.dim
  - min.dim_min
  - mm
  - mul.Tensor
  - mul_.Tensor
  - mul.Scalar
  - mul_.Scalar
  - mv
  - mv.out
  - native_batch_norm
  - native_batch_norm_backward
  - permute
  - reciprocal
  - reciprocal_
  - neg
  - neg_
  - repeat
  - round
  - round_
  - relu
  - relu_
  - gelu
  - gelu_backward
  - hardshrink
  - hardshrink_backward
  - rsqrt
  - rsqrt_
  - select.int
  - silu.out
  - sigmoid
  - sigmoid_
  - sin
  - sin_
  - sinh
  - sinh_
  - slice.Tensor
  - _softmax
  - _softmax_backward_data
  - split.Tensor
  - split_with_sizes
  - squeeze
  - squeeze.dim
  - squeeze_
  - squeeze_.dim
  - stack
  - sum
  - sum.dim_IntList
  - sqrt
  - sqrt_
  - std
  - std.dim
  - std.correction
  - prod
  - prod.dim_int
  - t
  - t_
  - tan
  - tan_
  - tanh
  - tanh_
  - threshold
  - threshold_
  - threshold_backward
  - transpose.int
  - transpose_
  - flip
  - _trilinear
  - trunc
  - trunc_
  - _unsafe_view
  - unsqueeze
  - unsqueeze_
  - var
  - var.dim
  - var.correction
  - _s_where
  - norm.ScalarOpt_dtype
  - norm.Scalar
  - norm.ScalarOpt_dim_dtype
  - norm.ScalarOpt_dim
  - clone
  - zero_
  - sub.Tensor
  - sub_.Tensor
  - sub.Scalar
  - sub_.Scalar
  - rsub.Tensor
  - rsub.Scalar
  - addmm
  - unbind.int
  - _local_scalar_dense
  - _pack_padded_sequence
  - masked_fill_.Scalar
  - masked_fill_.Tensor
  - masked_scatter_
  - view
  - put_
  - index_add_
  - index_fill_.int_Scalar
  - index_fill_.int_Tensor
  - scatter_.src
  - scatter_.value
  - scatter_add_
  - eq_.Scalar
  - eq_.Tensor
  - bitwise_and.Tensor_out
  - bitwise_and.Scalar_out
  - bitwise_or.Tensor_out
  - bitwise_or.Scalar_out
  - bitwise_xor.Tensor_out
  - bitwise_xor.Scalar_out
  - __lshift__.Scalar
  - __lshift__.Tensor
  - __ilshift__.Scalar
  - __ilshift__.Tensor
  - __rshift__.Scalar
  - __rshift__.Tensor
  - __irshift__.Scalar
  - __irshift__.Tensor
  - atan2_
  - tril_
  - triu_
  - fmod_.Scalar
  - fmod_.Tensor
  - remainder_.Scalar
  - remainder_.Tensor
  - addcdiv_
  - random_.from
  - random_.to
  - random_
  - uniform_
  - exponential_
  - diag
  - cross
  - triu
  - tril
  - trace
  - ne.Scalar
  - ne.Tensor
  - ne_.Scalar
  - ne_.Tensor
  - eq.Scalar
  - eq.Tensor
  - ge.Scalar
  - ge.Tensor
  - ge_.Scalar
  - ge_.Tensor
  - le.Scalar
  - le.Tensor
  - le_.Scalar
  - le_.Tensor
  - gt.Scalar
  - gt.Tensor
  - gt_.Scalar
  - gt_.Tensor
  - lt.Scalar
  - lt.Tensor
  - lt_.Scalar
  - lt_.Tensor
  - take
  - index_select
  - masked_select
  - nonzero
  - gather
  - addcmul
  - addcmul_
  - addcdiv
  - triangular_solve
  - symeig
  - svd
  - cholesky
  - qr
  - erfinv
  - erfinv_
  - sign
  - sign_
  - atan2
  - fmod.Scalar
  - fmod.Tensor
  - remainder.Scalar
  - remainder.Tensor
  - min
  - max
  - maximum
  - minimum
  - sort
  - topk
  - all
  - any
  - pow.Tensor_Tensor
  - pow.Scalar
  - pow.Tensor_Scalar
  - pow_.Scalar
  - pow_.Tensor
  - normal_
  - normal.Tensor_float
  - normal.float_Tensor
  - normal.Tensor_Tensor
  - alias
  - _amp_foreach_non_finite_check_and_unscale_
  - _amp_update_scale_
  - mse_loss
  - mse_loss_backward
  - l1_loss
  - l1_loss_backward
  - nll_loss_forward
  - nll_loss_backward
  - nll_loss2d_forward
  - nll_loss2d_backward
  - smooth_l1_loss
  - smooth_l1_loss_backward
  - elu
  - elu_backward
  - elu_
  - hardsigmoid
  - hardsigmoid_
  - hardsigmoid_backward
  - hardtanh
  - hardtanh_backward
  - hardtanh_
  - leaky_relu
  - leaky_relu_backward
  - leaky_relu_
  - log_sigmoid_forward
  - log_sigmoid_backward
  - rrelu_with_noise
  - rrelu_with_noise_backward
  - softplus
  - softplus_backward
  - softshrink
  - softshrink_backward
  - _adaptive_avg_pool2d
  - _adaptive_avg_pool2d_backward
  - _adaptive_avg_pool3d
  - _adaptive_avg_pool3d_backward
  - avg_pool2d
  - avg_pool2d_backward
  - avg_pool3d
  - avg_pool3d_backward
  - max_pool2d_with_indices
  - max_pool2d_with_indices_backward
  - max_pool3d_with_indices
  - max_pool3d_with_indices_backward
  - max_unpool2d
  - max_unpool2d_backward
  - max_unpool3d
  - max_unpool3d_backward
  - reflection_pad2d
  - reflection_pad2d_backward
  - replication_pad1d
  - replication_pad1d_backward
  - replication_pad2d
  - replication_pad2d_backward
  - upsample_nearest2d.vec
  - upsample_nearest2d_backward.vec
  - upsample_bilinear2d
  - upsample_bilinear2d_backward
  - upsample_nearest2d
  - upsample_nearest2d_backward
  - sigmoid_backward
  - tanh_backward
  - ger
autograd:
  - max_pool2d
  - max_pool3d