Function name Kernel Name In v2.13.0 Starting from macOS version Note
_assert_async_assert_async_mpsTrueN/AN/A
_assert_async.msg_assert_async_msg_mpsTrueN/AN/A
native_dropoutnative_dropout_mpsTrueN/AN/A
native_dropout_backwardnative_dropout_backward_mpsTrueN/AN/A
abs.outabs_outTrueN/AN/A
angleangleTrueN/AN/A
angle.outangle_outTrueN/AN/A
view_as_realview_as_realTrueN/AN/A
view_as_complexview_as_complexTrueN/AN/A
sgn.outsgn_out_mpsTrueN/AN/A
conj_physical.outconj_physical_out_mpsTrueN/AN/A
acos.outacos_outTrueN/AN/A
add.outadd_out_mpsTrueN/AN/A
addmv.outaddmv_out_mpsTrueN/AN/A
addraddr_mpsTrueN/AN/A
addr.outaddr_out_mpsTrueN/AN/A
all.outall_outTrueN/AN/A
any.outany_outTrueN/AN/A
arange.start_outarange_mps_outTrueN/AN/A
argmax.outargmax_out_mpsTrueN/AN/A
argmin.outargmin_out_mpsTrueN/AN/A
acosh.outacosh_out_mpsTrueN/AN/A
asinh.outasinh_out_mpsTrueN/AN/A
atanh.outatanh_out_mpsTrueN/AN/A
as_stridedas_strided_tensorimplTrueN/AN/A
asin.outasin_outTrueN/AN/A
atan.outatan_outTrueN/AN/A
baddbmm.outbaddbmm_out_mpsTrueN/AN/A
bernoulli.outbernoulli_outTrueN/AN/A
bernoulli.Tensorbernoulli_TrueN/AN/A
bernoulli.floatbernoulli_TrueN/AN/A
binary_cross_entropybinary_cross_entropy_mpsTrueN/AN/A
binary_cross_entropy.outbinary_cross_entropy_out_mpsTrueN/AN/A
binary_cross_entropy_backwardbinary_cross_entropy_backward_mpsTrueN/AN/A
binary_cross_entropy_backward.grad_inputbinary_cross_entropy_backward_out_mpsTrueN/AN/A
bincount_bincount_mpsTrueN/AN/A
bitwise_not.outbitwise_not_outTrueN/AN/A
copysign.outcopysign_outTrueN/AN/A
logical_not.outlogical_not_outTrueN/AN/A
logical_xor.outlogical_xor_out_mpsTrueN/AN/A
logical_and.outlogical_and_out_mpsTrueN/AN/A
logical_or.outlogical_or_out_mpsTrueN/AN/A
bmm.outbmm_out_mpsTrueN/AN/A
cat.outcat_out_mpsTrueN/AN/A
ceil.outceil_outTrueN/AN/A
clamp.outclamp_outTrueN/AN/A
clamp.Tensor_outclamp_Tensor_outTrueN/AN/A
clamp_max.outclamp_max_outTrueN/AN/A
clamp_max.Tensor_outclamp_max_Tensor_outTrueN/AN/A
clamp_min.outclamp_min_outTrueN/AN/A
clamp_min.Tensor_outclamp_min_Tensor_outTrueN/AN/A
complex.outcomplex_outTrueN/AN/A
polar.outpolar_outTrueN/AN/A
constant_pad_ndconstant_pad_nd_mpsTrueN/AN/A
_copy_from_copy_from_mpsTrueN/AN/A
_copy_from_and_resize_copy_from_and_resize_mpsTrueN/AN/A
cos.outcos_outTrueN/AN/A
cosh.outcosh_outTrueN/AN/A
count_nonzero.dim_IntListcount_nonzero_mpsTrueN/AN/A
_mps_convolution_transpose_mps_convolution_transposeTrueN/AN/A
mps_convolution_transpose_backwardmps_convolution_transpose_backwardTrueN/AN/A
_cummax_helpercummax_helper_mpsTrueN/AN/A
_cummin_helpercummin_helper_mpsTrueN/AN/A
cumprod.outcumprod_out_mpsTrueN/AN/A
cumsum.outcumsum_out_mpsTrue13.0N/A
_ctc_lossctc_loss_mpsFalseN/AN/A
_ctc_loss.Tensorctc_loss_tensorFalseN/AN/A
_ctc_loss_backwardctc_loss_backward_mpsFalseN/AN/A
_ctc_loss_backward.Tensorctc_loss_backward_tensorFalseN/AN/A
div.outdiv_outTrueN/AN/A
div.out_modediv_out_modeTrueN/AN/A
dotdot_mpsTrueN/AN/A
vdotvdot_mpsTrueN/AN/A
embedding_dense_backwardembedding_dense_backward_mpsTrueN/AN/A
_embedding_bag_forward_only_embedding_bag_forward_only_mpsTrueN/AN/A
_embedding_bag_embedding_bag_mpsTrueN/AN/A
_embedding_bag_backward_embedding_bag_backward_symintTrueN/AN/A
_embedding_bag_dense_backward_embedding_bag_dense_backward_mpsTrueN/AN/A
_embedding_bag_per_sample_weights_backward_embedding_bag_per_sample_weights_backward_mpsTrueN/AN/A
empty.memory_formatempty_mpsTrueN/AN/A
resizeresize_mps_TrueN/AN/A
empty_stridedempty_strided_mpsTrueN/AN/A
erf.outerf_outTrueN/AN/A
erfc.outerfc_outTrueN/AN/A
exp.outexp_outTrueN/AN/A
exp2.outexp2_outTrueN/AN/A
expm1.outexpm1_outTrueN/AN/A
eye.outeye_out_mpsTrueN/AN/A
eye.m_outeye_out_mpsTrueN/AN/A
fill.Scalarfill_TrueN/AN/A
fill.Tensorfill_TrueN/AN/A
floor.outfloor_outTrueN/AN/A
floor_dividefloor_divideTrueN/AN/A
floor_divide.Tensorfloor_divide_TrueN/AN/A
floor_divide.outfloor_divide_outTrueN/AN/A
frac.outfrac_out_mpsTrueN/AN/A
gcd.outgcd_outTrueN/AN/A
lcm.outlcm_outTrueN/AN/A
grid_sampler_2dgrid_sampler_2d_mpsTrueN/AN/A
grid_sampler_2d_backwardgrid_sampler_2d_backward_mpsTrueN/AN/A
grid_sampler_3dgrid_sampler_3d_mpsTrueN/AN/A
grid_sampler_3d_backwardgrid_sampler_3d_backward_mpsTrueN/AN/A
native_group_normmanually registeredTrueN/AN/A
native_group_norm_backwardmanually registeredTrueN/AN/A
_fft_r2c_fft_r2c_mpsTrueN/AN/A
_fft_r2c.out_fft_r2c_mps_outTrueN/AN/A
_fft_c2r_fft_c2r_mpsTrueN/AN/A
_fft_c2r.out_fft_c2r_mps_outTrueN/AN/A
_fft_c2c_fft_c2c_mpsTrueN/AN/A
_fft_c2c.out_fft_c2c_mps_outTrueN/AN/A
index.Tensor_outindex_outTrueN/AN/A
index_copy.outindex_copy_out_mpsTrueN/AN/A
_index_put_impl_index_put_impl_TrueN/AN/A
isin.Tensor_Tensor_outisin_Tensor_Tensor_out_mpsTrueN/AN/A
isin.Tensor_Scalar_outisin_Tensor_Scalar_outTrueN/AN/A
isin.Scalar_Tensor_outisin_Scalar_Tensor_out_mpsTrueN/AN/A
isnanisnanTrueN/AN/A
kthvalue.valueskthvalue_out_mpsTrueN/AN/A
native_layer_normlayer_norm_mpsTrueN/AN/A
native_layer_norm_backwardlayer_norm_backward_mpsTrueN/AN/A
_fused_rms_norm_fused_rms_norm_mpsTrueN/AN/A
nan_to_num.outnan_to_num_outTrueN/AN/A
linear_mps_linearTrueN/AN/A
linear_backwardmps_linear_backwardTrueN/AN/A
linspace.outlinspace_out_mpsTrueN/AN/A
log.outlog_outTrueN/AN/A
log10.outlog10_outTrueN/AN/A
log1p.outlog1p_outTrueN/AN/A
log2.outlog2_outTrueN/AN/A
logaddexp.outlogaddexp_outTrueN/AN/A
logaddexp2.outlogaddexp2_outTrueN/AN/A
xlogy.OutTensorxlogy_outTrueN/AN/A
_log_softmax.outlog_softmax_mps_outTrueN/AN/A
_log_softmax_backward_data.outlog_softmax_backward_mps_outTrueN/AN/A
_logcumsumexp_logcumsumexp_mpsTrueN/AN/A
_logcumsumexp.out_logcumsumexp_out_mpsTrueN/AN/A
aminmax.outaminmax_outTrueN/AN/A
_compute_linear_combination_compute_linear_combinationFalseN/AN/A
_compute_linear_combination.out_compute_linear_combination_outFalseN/AN/A
max.dim_maxmax_out_mpsTrueN/AN/A
amax.outamax_outTrueN/AN/A
max_pool2dmps_max_pool2dTrueN/AN/A
max_pool2d_backwardmps_max_pool2d_backwardTrueN/AN/A
mean.outmean_outTrueN/AN/A
medianmedian_mpsTrueN/AN/A
median.dim_valuesmedian_out_mpsTrueN/AN/A
nanmediannanmedian_mpsTrueN/AN/A
nanmedian.dim_valuesnanmedian_out_mpsTrueN/AN/A
min.dim_minmin_out_mpsTrueN/AN/A
amin.outamin_outTrueN/AN/A
_mps_convolution_mps_convolutionTrueN/AN/A
mps_convolution_backwardmps_convolution_backwardTrueN/AN/A
mm.outmm_out_mpsTrueN/AN/A
_convert_weight_to_int4pack_convert_weight_to_int4pack_mpsTrueN/AN/A
_weight_int4pack_mm_weight_int4pack_mm_mpsTrueN/AN/A
_weight_int8pack_mm_weight_int8pack_mm_mpsTrueN/AN/A
mul.outmul_outTrueN/AN/A
mvlgamma.outmvlgamma_outTrueN/AN/A
native_batch_normbatch_norm_mpsTrueN/AN/A
native_batch_norm.outbatch_norm_mps_outTrueN/AN/A
_native_batch_norm_legit_batch_norm_legit_mpsTrueN/AN/A
_native_batch_norm_legit.out_batch_norm_legit_mps_outTrueN/AN/A
_native_batch_norm_legit.no_stats_batch_norm_legit_no_stats_mpsTrueN/AN/A
_native_batch_norm_legit.no_stats_out_batch_norm_legit_no_stats_mps_outTrueN/AN/A
native_batch_norm_backwardbatch_norm_backward_mpsTrueN/AN/A
_cdist_forward_cdist_forwardTrueN/AN/A
_cdist_backward_cdist_backwardTrueN/AN/A
permutepermute_mpsTrueN/AN/A
pixel_shufflepixel_shuffle_mpsTrueN/AN/A
pixel_unshufflepixel_unshuffle_mpsTrueN/AN/A
randperm.generator_outrandperm_out_mpsTrueN/AN/A
range.outrange_mps_outTrueN/AN/A
reciprocal.outreciprocal_outTrueN/AN/A
neg.outneg_outTrueN/AN/A
repeatrepeat_mpsTrueN/AN/A
repeat_interleave.Tensorrepeat_interleave_mpsTrueN/AN/A
_reshape_alias_reshape_aliasTrueN/AN/A
round.outround_outTrueN/AN/A
round.decimals_outround_decimals_outTrueN/AN/A
relurelu_mps_TrueN/AN/A
_prelu_kernelprelu_mpsTrueN/AN/A
_prelu_kernel_backwardprelu_backward_mpsTrueN/AN/A
gelu.outgelu_out_mpsTrueN/AN/A
gelu_backward.grad_inputgelu_backward_out_mpsTrueN/AN/A
hardshrink.outhardshrink_outTrueN/AN/A
hardshrink_backward.grad_inputhardshrink_backward_outTrueN/AN/A
rsqrt.outrsqrt_outTrueN/AN/A
silu.outsilu_outTrueN/AN/A
silu_backward.grad_inputsilu_backward_outTrueN/AN/A
mish.outmish_outTrueN/AN/A
mish_backwardmish_backwardTrueN/AN/A
sigmoid.outsigmoid_outTrueN/AN/A
logitlogit_mpsTrueN/AN/A
logit.outlogit_out_mpsTrueN/AN/A
sin.outsin_outTrueN/AN/A
sinc.outsinc_outTrueN/AN/A
sinh.outsinh_outTrueN/AN/A
_softmax.outsoftmax_mps_outTrueN/AN/A
_softmax_backward_data.outsoftmax_backward_mps_outTrueN/AN/A
sum.IntList_outsum_outTrueN/AN/A
nansumnansumTrueN/AN/A
nansum.outnansum_outTrueN/AN/A
sqrt.outsqrt_outTrueN/AN/A
std.correctionstd_mpsTrueN/AN/A
std_mean.correctionstd_mean_mpsTrueN/AN/A
prodprod_mpsTrueN/AN/A
prod.int_outprod_out_mpsTrueN/AN/A
tan.outtan_outTrueN/AN/A
tanh.outtanh_outTrueN/AN/A
threshold.outthreshold_out_mpsTrueN/AN/A
threshold_backward.grad_inputthreshold_backward_out_mpsTrueN/AN/A
flipflip_mpsTrueN/AN/A
rollrollTrueN/AN/A
trunc.outtrunc_outTrueN/AN/A
_unique_unique_mpsTrueN/AN/A
unique_dimunique_dim_mpsTrueN/AN/A
unique_consecutiveunique_consecutive_mpsTrueN/AN/A
unique_dim_consecutiveunique_dim_consecutive_mpsTrueN/AN/A
_unique2_unique2_mpsTrueN/AN/A
var.correctionvar_mpsTrueN/AN/A
var_mean.correctionmanually registeredTrueN/AN/A
where.selfwhereTrueN/AN/A
where.self_outwhere_self_outTrueN/AN/A
_weight_norm_interfaceweight_norm_mpsTrueN/AN/A
_weight_norm_interface_backwardweight_norm_backward_mpsTrueN/AN/A
_efficientzerotensor_efficientzerotensor_mpsTrueN/AN/A
_standard_gamma_grad_standard_gamma_grad_mpsTrueN/AN/A
_standard_gamma_s_gamma_mpsTrueN/AN/A
_dirichlet_grad_dirichlet_grad_mpsTrueN/AN/A
_sample_dirichlet_s_dirichlet_mpsTrueN/AN/A
poisson_s_poisson_mpsFalseN/AN/A
binomial_s_binomial_mpsFalseN/AN/A
_batch_norm_with_update_batch_norm_with_update_mpsTrueN/AN/A
_batch_norm_with_update.out_batch_norm_with_update_mps_outTrueN/AN/A
batch_norm_backward_new_batch_norm_backward_mpsTrueN/AN/A
norm.dtype_outnorm_dtype_outTrueN/AN/A
norm.outnorm_outTrueN/AN/A
zerozero_TrueN/AN/A
sub.outsub_out_mpsTrueN/AN/A
rsub.TensorrsubTrueN/AN/A
addmm.outaddmm_out_mpsTrueN/AN/A
_to_sparse.sparse_dimdense_to_sparseTrueN/AN/A
_to_sparsedense_to_sparseTrueN/AN/A
_local_scalar_dense_local_scalar_dense_mpsTrueN/AN/A
_lstm_mps_lstm_mpsTrueN/AN/A
lstm_mps_backwardlstm_mps_backwardTrueN/AN/A
set.source_Storageset_TrueN/AN/A
set.source_Storage_storage_offsetset_storage_mps_TrueN/AN/A
set.source_Tensorset_tensor_TrueN/AN/A
setset_mps_TrueN/AN/A
is_set_tois_set_toTrueN/AN/A
masked_fill.Scalarmasked_fill__mpsTrueN/AN/A
masked_fill.Tensormasked_fill__mpsTrueN/AN/A
masked_scattermasked_scatter__mpsTrueN/AN/A
viewviewTrueN/AN/A
index_add.outindex_add_mps_outTrueN/AN/A
index_reduce.outindex_reduce_mps_outTrueN/AN/A
index_fill.int_Scalarindex_fill_TrueN/AN/A
index_fill.int_Tensorindex_fill_TrueN/AN/A
scatter.src_outscatter_src_outTrueN/AN/A
scatter.value_outscatter_value_outTrueN/AN/A
scatter.reduce_outscatter_reduce_outTrueN/AN/A
scatter.value_reduce_outscatter_value_reduce_outTrueN/AN/A
scatter_add.outscatter_addTrueN/AN/A
scatter_reduce.two_outscatter_reduce_twoTrueN/AN/A
bitwise_and.Tensor_outbitwise_and_outTrueN/AN/A
bitwise_or.Tensor_outbitwise_or_outTrueN/AN/A
bitwise_xor.Tensor_outbitwise_xor_outTrueN/AN/A
lshift.Scalar__ilshift__TrueN/AN/A
lshift.Tensor__ilshift__TrueN/AN/A
bitwise_left_shift.Tensor_outbitwise_left_shift_outTrueN/AN/A
rshift.Scalar__irshift__TrueN/AN/A
rshift.Tensor__irshift__TrueN/AN/A
bitwise_right_shift.Tensor_outbitwise_right_shift_outTrueN/AN/A
addbmmaddbmm_mpsTrueN/AN/A
addbmm.outaddbmm_out_mpsTrueN/AN/A
random.fromrandom_TrueN/AN/A
random.torandom_TrueN/AN/A
randomrandom_TrueN/AN/A
uniformuniform_TrueN/AN/A
cauchycauchy_TrueN/AN/A
log_normallog_normal_TrueN/AN/A
exponentialexponential_TrueN/AN/A
geometricgeometric_TrueN/AN/A
triu.outtriu_mps_outTrueN/AN/A
tril.outtril_mps_outTrueN/AN/A
tril_indicestril_indices_mpsTrueN/AN/A
triu_indicestriu_indices_mpsTrueN/AN/A
tracetrace_mpsTrueN/AN/A
ne.Scalar_outne_Scalar_outTrueN/AN/A
ne.Tensor_outne_Tensor_outTrueN/AN/A
eq.Scalar_outeq_Scalar_outTrueN/AN/A
eq.Tensor_outeq_Tensor_outTrueN/AN/A
ge.Scalar_outge_Scalar_outTrueN/AN/A
ge.Tensor_outge_Tensor_outTrueN/AN/A
le.Scalar_outle_Scalar_outTrueN/AN/A
le.Tensor_outle_Tensor_outTrueN/AN/A
gt.Scalar_outgt_Scalar_outTrueN/AN/A
gt.Tensor_outgt_Tensor_outTrueN/AN/A
lt.Scalar_outlt_Scalar_outTrueN/AN/A
lt.Tensor_outlt_Tensor_outTrueN/AN/A
index_select.outindex_select_out_mpsTrueN/AN/A
index_selectindex_select_mpsTrueN/AN/A
masked_select.outmasked_select_out_mpsTrueN/AN/A
masked_selectmasked_select_mpsTrueN/AN/A
nonzero.outnonzero_out_mpsTrueN/AN/A
nonzerononzero_mpsTrueN/AN/A
nonzero_static.outnonzero_static_out_mpsTrueN/AN/A
nonzero_staticnonzero_static_mpsTrueN/AN/A
gather.outgather_outTrueN/AN/A
addcmul.outaddcmul_out_mpsTrueN/AN/A
addcdiv.outaddcdiv_out_mpsTrueN/AN/A
triangular_solve.Xtriangular_solve_mps_outTrueN/AN/A
linalg_solve_triangular.outlinalg_solve_triangular_mps_outTrueN/AN/A
linalg_solve_triangularlinalg_solve_triangular_mpsTrueN/AN/A
_cholesky_solve_helper_cholesky_solve_helper_mpsTrueN/AN/A
cholesky_inversecholesky_inverseTrueN/AN/A
cholesky_inverse.outcholesky_inverse_outTrueN/AN/A
lu_unpack.outlu_unpack_outTrueN/AN/A
multinomial.outmultinomial_outTrueN/AN/A
multinomialmultinomialTrueN/AN/A
lgamma.outlgamma_outTrueN/AN/A
digamma.outdigamma_outTrueN/AN/A
polygamma.outpolygamma_outTrueN/AN/A
erfinv.outerfinv_outTrueN/AN/A
i0.outi0_outTrueN/AN/A
sign.outsign_out_mpsTrueN/AN/A
signbit.outsignbit_out_mpsTrueN/AN/A
atan2.outatan2_outTrueN/AN/A
lerp.Scalar_outlerp_ScalarTrueN/AN/A
lerp.Tensor_outlerp_TensorTrueN/AN/A
histc.outhistogram_histc_outTrueN/AN/A
histchistogram_histcTrueN/AN/A
histogram.bins_tensor_outhistogram_outTrueN/AN/A
histogram.bins_tensorhistogramTrueN/AN/A
histogram.bin_ct_outhistogram_outTrueN/AN/A
histogram.bin_cthistogramTrueN/AN/A
_histogramdd_bin_edgeshistogramdd_bin_edgesTrueN/AN/A
_histogramdd_from_bin_cts_histogramddTrueN/AN/A
_histogramdd_from_bin_tensors_histogramddTrueN/AN/A
fmod.Tensor_outfmod_outTrueN/AN/A
hypot.outhypot_outTrueN/AN/A
igamma.outigamma_outTrueN/AN/A
igammac.outigammac_outTrueN/AN/A
nextafter.outnextafter_outTrueN/AN/A
remainder.Tensor_outremainder_outTrueN/AN/A
remainder.Scalar_TensorremainderTrueN/AN/A
minmin_mpsTrueN/AN/A
fmin.outfmin_outTrueN/AN/A
maxmax_mpsTrueN/AN/A
fmax.outfmax_outTrueN/AN/A
maximum.outmaximum_outTrueN/AN/A
minimum.outminimum_outTrueN/AN/A
sort.values_stablesort_stable_out_mpsTrueN/AN/A
topk.valuestopk_out_mpsTrueN/AN/A
all.all_outall_all_outTrueN/AN/A
any.all_outany_all_outTrueN/AN/A
renorm.outrenorm_out_mpsTrueN/AN/A
unfoldunfoldTrueN/AN/A
unfold_backwardunfold_backwardTrueN/AN/A
equalmps_equalTrueN/AN/A
pow.Tensor_Tensor_outpow_tensor_tensor_out_mpsTrueN/AN/A
pow.Scalar_outpow_Scalar_out_mpsTrueN/AN/A
pow.Tensor_Scalar_outpow_Tensor_Scalar_outTrueN/AN/A
normalnormal_TrueN/AN/A
normal.Tensor_float_outnormal_outTrueN/AN/A
normal.Tensor_floatnormalTrueN/AN/A
normal.float_Tensor_outnormal_outTrueN/AN/A
normal.float_TensornormalTrueN/AN/A
normal.Tensor_Tensor_outnormal_outTrueN/AN/A
normal.Tensor_TensornormalTrueN/AN/A
_amp_foreach_non_finite_check_and_unscale_amp_foreach_non_finite_check_and_unscale_mps_TrueN/AN/A
_amp_update_scale_amp_update_scale_mps_TrueN/AN/A
bucketize.Tensorbucketize_mpsTrueN/AN/A
bucketize.Tensor_outbucketize_out_mpsTrueN/AN/A
bucketize.Scalarbucketize_mpsTrueN/AN/A
searchsorted.Tensorsearchsorted_mpsTrueN/AN/A
searchsorted.Tensor_outsearchsorted_out_mpsTrueN/AN/A
searchsorted.Scalarsearchsorted_mpsTrueN/AN/A
searchsorted.Scalar_outsearchsorted_out_mpsTrueN/AN/A
mse_loss.outmse_loss_out_mpsTrueN/AN/A
mse_loss_backward.grad_inputmse_loss_backward_out_mpsTrueN/AN/A
mse_loss_backwardmse_loss_backward_mpsTrueN/AN/A
nll_loss_forward.outputnll_loss_forward_out_mpsTrueN/AN/A
nll_loss_backward.grad_inputnll_loss_backward_out_mpsTrueN/AN/A
nll_loss2d_forward.outputnll_loss2d_forward_out_mpsTrueN/AN/A
nll_loss2d_forwardnll_loss2d_forward_mpsTrueN/AN/A
nll_loss2d_backward.grad_inputnll_loss2d_backward_out_mpsTrueN/AN/A
nll_loss2d_backwardnll_loss2d_backward_mpsTrueN/AN/A
smooth_l1_loss.outsmooth_l1_loss_out_mpsTrueN/AN/A
smooth_l1_loss_backward.grad_inputsmooth_l1_loss_backward_out_mpsTrueN/AN/A
huber_loss.outhuber_loss_out_mpsTrueN/AN/A
huber_losshuber_loss_mpsTrueN/AN/A
huber_loss_backward.outhuber_loss_backward_out_mpsTrueN/AN/A
elu.outelu_outTrueN/AN/A
elu_backward.grad_inputelu_backward_outTrueN/AN/A
glu.outglu_out_mpsTrueN/AN/A
glu_backward.grad_inputglu_backward_mps_outTrueN/AN/A
glu_backwardglu_backward_mpsTrueN/AN/A
hardsigmoid.outhardsigmoid_outTrueN/AN/A
hardsigmoid_backward.grad_inputhardsigmoid_backward_outTrueN/AN/A
hardtanh.outhardtanh_outTrueN/AN/A
hardtanhhardtanh_TrueN/AN/A
hardtanh_backward.grad_inputhardtanh_backward_out_mpsTrueN/AN/A
hardtanh_backwardhardtanh_backward_mpsTrueN/AN/A
hardswish.outhardswish_outTrueN/AN/A
hardswishhardswish_TrueN/AN/A
hardswish_backwardhardswish_backwardTrueN/AN/A
leaky_relu.outleaky_relu_outTrueN/AN/A
leaky_relu_backward.grad_inputleaky_relu_backward_outTrueN/AN/A
log_sigmoid_forward.outputlog_sigmoid_forward_out_mpsTrueN/AN/A
log_sigmoid_forwardlog_sigmoid_forward_mpsTrueN/AN/A
log_sigmoid_backward.grad_inputlog_sigmoid_backward_mps_outTrueN/AN/A
log_sigmoid_backwardlog_sigmoid_backward_mpsTrueN/AN/A
softplus.outsoftplus_out_mpsTrueN/AN/A
softplus_backward.grad_inputsoftplus_backward_out_mpsTrueN/AN/A
softshrink.outsoftshrink_outTrueN/AN/A
softshrink_backward.grad_inputsoftshrink_backward_outTrueN/AN/A
adaptive_avg_pool2d.outadaptive_avg_pool2d_out_mpsTrueN/AN/A
_adaptive_avg_pool2dadaptive_avg_pool2d_mpsTrueN/AN/A
_adaptive_avg_pool2d_backwardadaptive_avg_pool2d_backward_mpsTrueN/AN/A
adaptive_max_pool2d.outadaptive_max_pool2d_out_mpsTrueN/AN/A
adaptive_max_pool2d_backward.grad_inputadaptive_max_pool2d_backward_out_mpsTrueN/AN/A
avg_pool2d.outavg_pool2d_out_mpsTrueN/AN/A
avg_pool2d_backward.grad_inputavg_pool2d_backward_out_mpsTrueN/AN/A
avg_pool3d.outavg_pool3d_out_mpsTrueN/AN/A
avg_pool3d_backward.grad_inputavg_pool3d_backward_out_mpsTrueN/AN/A
max_pool2d_with_indices.outmax_pool2d_with_indices_out_mpsTrueN/AN/A
max_pool2d_with_indices_backward.grad_inputmax_pool2d_with_indices_backward_out_mpsTrueN/AN/A
max_pool3d_with_indices.outmax_pool3d_with_indices_out_mpsTrueN/AN/A
max_pool3d_with_indicesmax_pool3d_with_indices_mpsTrueN/AN/A
max_pool3d_with_indices_backward.grad_inputmax_pool3d_with_indices_backward_out_mpsTrueN/AN/A
max_pool3d_with_indices_backwardmax_pool3d_with_indices_backward_mpsTrueN/AN/A
max_unpool2d.outmax_unpooling2d_forward_out_mpsTrueN/AN/A
max_unpool2dmax_unpooling2d_forward_mpsTrueN/AN/A
max_unpool3d.outmax_unpooling3d_forward_out_mpsTrueN/AN/A
max_unpool3dmax_unpooling3d_forward_mpsTrueN/AN/A
reflection_pad1d.outreflection_pad1d_out_mpsTrueN/AN/A
reflection_pad1d_backward.grad_inputreflection_pad1d_backward_out_mpsTrueN/AN/A
reflection_pad2d.outreflection_pad2d_out_mpsTrueN/AN/A
reflection_pad2dreflection_pad2d_mpsTrueN/AN/A
reflection_pad2d_backward.grad_inputreflection_pad2d_backward_out_mpsTrueN/AN/A
reflection_pad2d_backwardreflection_pad2d_backward_mpsTrueN/AN/A
reflection_pad3d.outreflection_pad3d_out_mpsTrueN/AN/A
reflection_pad3d_backward.grad_inputreflection_pad3d_backward_out_mpsTrueN/AN/A
replication_pad1d.outreplication_pad1d_out_mpsTrueN/AN/A
replication_pad1d_backward.grad_inputreplication_pad1d_backward_out_mpsTrueN/AN/A
replication_pad2d.outreplication_pad2d_out_mpsTrueN/AN/A
replication_pad2d_backward.grad_inputreplication_pad2d_backward_out_mpsTrueN/AN/A
replication_pad2d_backwardreplication_pad2d_backward_mpsTrueN/AN/A
replication_pad3d.outreplication_pad3d_out_mpsTrueN/AN/A
replication_pad3d_backward.grad_inputreplication_pad3d_backward_out_mpsTrueN/AN/A
replication_pad3d_backwardreplication_pad3d_backward_mpsTrueN/AN/A
upsample_linear1d.outupsample_linear1d_out_mpsTrueN/AN/A
upsample_linear1d_backward.grad_inputupsample_linear1d_backward_out_mpsTrueN/AN/A
upsample_bilinear2d.outupsample_bilinear2d_out_mpsTrueN/AN/A
upsample_bilinear2d_backward.grad_inputupsample_bilinear2d_backward_out_mpsTrueN/AN/A
_upsample_bilinear2d_aa.out_upsample_bilinear2d_aa_out_mpsTrueN/AN/A
_upsample_bilinear2d_aa_backward.grad_input_upsample_bilinear2d_aa_backward_out_mpsFalseN/AN/A
upsample_bicubic2d.outupsample_bicubic2d_out_mpsTrueN/AN/A
upsample_bicubic2d_backward.grad_inputupsample_bicubic2d_backward_out_mpsTrueN/AN/A
_upsample_bicubic2d_aa.out_upsample_bicubic2d_aa_out_mpsTrueN/AN/A
_upsample_bicubic2d_aa_backward.grad_input_upsample_bicubic2d_aa_backward_out_mpsFalseN/AN/A
upsample_trilinear3d.outupsample_trilinear3d_out_mpsTrueN/AN/A
upsample_trilinear3d_backward.grad_inputupsample_trilinear3d_backward_out_mpsTrueN/AN/A
upsample_nearest1d.outupsample_nearest1d_out_mpsTrueN/AN/A
_upsample_nearest_exact1d.out_upsample_nearest_exact1d_out_mpsTrueN/AN/A
upsample_nearest1d_backward.grad_inputupsample_nearest1d_backward_out_mpsTrueN/AN/A
_upsample_nearest_exact1d_backward.grad_input_upsample_nearest_exact1d_backward_out_mpsTrueN/AN/A
upsample_nearest2d.outupsample_nearest2d_out_mpsTrueN/AN/A
_upsample_nearest_exact2d.out_upsample_nearest_exact2d_out_mpsTrueN/AN/A
upsample_nearest2d_backward.grad_inputupsample_nearest2d_backward_out_mpsTrueN/AN/A
_upsample_nearest_exact2d_backward.grad_input_upsample_nearest_exact2d_backward_out_mpsTrueN/AN/A
upsample_nearest3d.outupsample_nearest3d_out_mpsTrueN/AN/A
_upsample_nearest_exact3d.out_upsample_nearest_exact3d_out_mpsTrueN/AN/A
upsample_nearest3d_backward.grad_inputupsample_nearest3d_backward_out_mpsTrueN/AN/A
_upsample_nearest_exact3d_backward.grad_input_upsample_nearest_exact3d_backward_out_mpsTrueN/AN/A
sigmoid_backward.grad_inputsigmoid_backward_outTrueN/AN/A
logit_backward.grad_inputlogit_backward_out_mpsTrueN/AN/A
tanh_backward.grad_inputtanh_backward_out_mpsTrueN/AN/A
col2im.outcol2im_out_mpsTrueN/AN/A
col2imcol2im_mpsTrueN/AN/A
im2col.outim2col_out_mpsTrueN/AN/A
im2colim2col_mpsTrueN/AN/A
isposinf.outisposinf_outTrueN/AN/A
isneginf.outisneginf_outTrueN/AN/A
special_entr.outspecial_entr_outTrueN/AN/A
special_erfcx.outspecial_erfcx_outTrueN/AN/A
special_xlog1py.outspecial_xlog1py_outTrueN/AN/A
special_zeta.outspecial_zeta_outTrueN/AN/A
special_i0e.outspecial_i0e_outTrueN/AN/A
special_i1.outspecial_i1_outTrueN/AN/A
special_i1e.outspecial_i1e_outTrueN/AN/A
linalg_cholesky_ex.Llinalg_cholesky_ex_outTrueN/AN/A
linalg_cross.outlinalg_cross_outTrueN/AN/A
linalg_lu_factor_ex.outlinalg_lu_factor_ex_out_mpsTrueN/AN/A
linalg_lu.outlinalg_lu_outTrueN/AN/A
linalg_lu_solve.outlinalg_lu_solve_outFalseN/AN/A
_linalg_det.result_linalg_det_outTrueN/AN/A
linalg_lstsq.outlinalg_lstsq_outFalseN/AN/A
linalg_matrix_explinalg_matrix_expFalseN/AN/A
_linalg_slogdet.sign_linalg_slogdet_outTrueN/AN/A
_linalg_eigh.eigenvalues_linalg_eigh_outFalseN/AN/A
linalg_householder_productlinalg_householder_productTrueN/AN/A
linalg_householder_product.outlinalg_householder_product_outTrueN/AN/A
linalg_inv_ex.inverselinalg_inv_ex_out_mpsTrueN/AN/A
linalg_vector_norm.outlinalg_vector_norm_outTrueN/AN/A
_linalg_svd.U_linalg_svd_outFalseN/AN/A
_linalg_solve_ex.result_linalg_solve_ex_out_mpsTrueN/AN/A
linalg_qr.outlinalg_qr_out_mpsTrueN/AN/A
linalg_polar.outlinalg_polar_outFalseN/AN/A
_scaled_dot_product_attention_math_for_mps_scaled_dot_product_attention_math_mpsTrueN/AN/A
special_bessel_j0.outspecial_bessel_j0_outTrueN/AN/A
special_bessel_j1.outspecial_bessel_j1_outTrueN/AN/A
special_bessel_y0.outspecial_bessel_y0_outTrueN/AN/A
special_bessel_y1.outspecial_bessel_y1_outTrueN/AN/A
special_chebyshev_polynomial_t.outspecial_chebyshev_polynomial_t_outTrueN/AN/A
special_chebyshev_polynomial_u.outspecial_chebyshev_polynomial_u_outTrueN/AN/A
special_chebyshev_polynomial_v.outspecial_chebyshev_polynomial_v_outTrueN/AN/A
special_chebyshev_polynomial_w.outspecial_chebyshev_polynomial_w_outTrueN/AN/A
special_hermite_polynomial_h.outspecial_hermite_polynomial_h_outTrueN/AN/A
special_hermite_polynomial_he.outspecial_hermite_polynomial_he_outTrueN/AN/A
special_modified_bessel_i0.outspecial_modified_bessel_i0_outTrueN/AN/A
special_modified_bessel_i1.outspecial_modified_bessel_i1_outTrueN/AN/A
special_modified_bessel_k0.outspecial_modified_bessel_k0_outTrueN/AN/A
special_modified_bessel_k1.outspecial_modified_bessel_k1_outTrueN/AN/A
special_scaled_modified_bessel_k0.outspecial_scaled_modified_bessel_k0_outTrueN/AN/A
special_scaled_modified_bessel_k1.outspecial_scaled_modified_bessel_k1_outTrueN/AN/A
special_shifted_chebyshev_polynomial_t.outspecial_shifted_chebyshev_polynomial_t_outTrueN/AN/A
special_shifted_chebyshev_polynomial_u.outspecial_shifted_chebyshev_polynomial_u_outTrueN/AN/A
special_shifted_chebyshev_polynomial_v.outspecial_shifted_chebyshev_polynomial_v_outTrueN/AN/A
special_shifted_chebyshev_polynomial_w.outspecial_shifted_chebyshev_polynomial_w_outTrueN/AN/A
special_spherical_bessel_j0.outspecial_spherical_bessel_j0_outTrueN/AN/A
_fused_adam_fused_adam_kernel_mps_TrueN/AN/A
_fused_adam.tensor_lr_fused_adam_kernel_mps_TrueN/AN/A
_fused_adamw_fused_adamw_kernel_mps_TrueN/AN/A
_fused_adamw.tensor_lr_fused_adamw_kernel_mps_TrueN/AN/A
_fused_sgd_fused_sgd_kernel_mps_TrueN/AN/A
_fused_sgd.tensor_lr_fused_sgd_kernel_mps_TrueN/AN/A