mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[functorch] Added all of the upsample no overload functions too (not sure if these ever even show up...)
This commit is contained in:
@ -268,6 +268,10 @@ struct UpsampleBackwardBatchRuleHelper<F, Func, typelist<A, B, C, T...>> {
|
|||||||
&ATEN_FN2(op, overload),\
|
&ATEN_FN2(op, overload),\
|
||||||
c10::guts::function_traits<decltype(ATEN_FN2(op, overload))>::parameter_types>::apply))
|
c10::guts::function_traits<decltype(ATEN_FN2(op, overload))>::parameter_types>::apply))
|
||||||
|
|
||||||
|
#define UPSAMPLE_BATCH(op) \
|
||||||
|
VMAP_SUPPORT(#op".vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(op, vec))); \
|
||||||
|
VMAP_SUPPORT(#op, EXISTING_BDIM_BATCH_RULE(ATEN_FN(op)));
|
||||||
|
|
||||||
TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
|
TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
|
||||||
VMAP_SUPPORT("convolution", convolution_batching_rule);
|
VMAP_SUPPORT("convolution", convolution_batching_rule);
|
||||||
m.impl("conv1d", convNd_decomp);
|
m.impl("conv1d", convNd_decomp);
|
||||||
@ -286,13 +290,14 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
|
|||||||
VMAP_SUPPORT("replication_pad2d", EXISTING_BDIM_BATCH_RULE(at::replication_pad2d));
|
VMAP_SUPPORT("replication_pad2d", EXISTING_BDIM_BATCH_RULE(at::replication_pad2d));
|
||||||
VMAP_SUPPORT("replication_pad3d", EXISTING_BDIM_BATCH_RULE(at::replication_pad3d));
|
VMAP_SUPPORT("replication_pad3d", EXISTING_BDIM_BATCH_RULE(at::replication_pad3d));
|
||||||
|
|
||||||
VMAP_SUPPORT("upsample_bicubic2d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_bicubic2d, vec)));
|
UPSAMPLE_BATCH(upsample_bicubic2d);
|
||||||
VMAP_SUPPORT("upsample_bilinear2d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_bilinear2d, vec)));
|
UPSAMPLE_BATCH(upsample_bilinear2d);
|
||||||
VMAP_SUPPORT("upsample_linear1d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_linear1d, vec)));
|
UPSAMPLE_BATCH(upsample_linear1d);
|
||||||
VMAP_SUPPORT("upsample_nearest1d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_nearest1d, vec)));
|
UPSAMPLE_BATCH(unsample_nearest1d);
|
||||||
VMAP_SUPPORT("upsample_nearest2d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_nearest2d, vec)));
|
UPSAMPLE_BATCH(unsample_nearest2d);
|
||||||
VMAP_SUPPORT("upsample_nearest3d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_nearest3d, vec)));
|
UPSAMPLE_BATCH(unsample_nearest3d);
|
||||||
VMAP_SUPPORT("upsample_trilinear3d.vec", EXISTING_BDIM_BATCH_RULE(ATEN_FN2(upsample_trilinear3d, vec)));
|
UPSAMPLE_BATCH(unsample_trilinear3d);
|
||||||
|
|
||||||
UPSAMPLE_BACKWARD(upsample_bicubic2d_backward, vec);
|
UPSAMPLE_BACKWARD(upsample_bicubic2d_backward, vec);
|
||||||
UPSAMPLE_BACKWARD(upsample_bilinear2d_backward, vec);
|
UPSAMPLE_BACKWARD(upsample_bilinear2d_backward, vec);
|
||||||
UPSAMPLE_BACKWARD(upsample_linear1d_backward, vec);
|
UPSAMPLE_BACKWARD(upsample_linear1d_backward, vec);
|
||||||
|
@ -365,7 +365,7 @@ bitwise_or, primitive pointwise
|
|||||||
bitwise_xor, primitive pointwise
|
bitwise_xor, primitive pointwise
|
||||||
addbmm, composite matmul
|
addbmm, composite matmul
|
||||||
diag, view/reshape
|
diag, view/reshape
|
||||||
cross, reduction
|
cross, misc
|
||||||
triu, view/reshape
|
triu, view/reshape
|
||||||
tril, view/reshape
|
tril, view/reshape
|
||||||
tril_indices, view/reshape
|
tril_indices, view/reshape
|
||||||
|
793
functorch/op_analysis/noncomposite_dedup.txt
Normal file
793
functorch/op_analysis/noncomposite_dedup.txt
Normal file
@ -0,0 +1,793 @@
|
|||||||
|
aten::__ilshift__
|
||||||
|
aten::__irshift__
|
||||||
|
aten::__lshift__
|
||||||
|
aten::__rshift__
|
||||||
|
aten::_adaptive_avg_pool2d
|
||||||
|
aten::_adaptive_avg_pool2d_backward
|
||||||
|
aten::_adaptive_avg_pool3d
|
||||||
|
aten::_adaptive_avg_pool3d_backward
|
||||||
|
aten::_add_relu
|
||||||
|
aten::_add_relu_
|
||||||
|
aten::_aminmax
|
||||||
|
aten::_amp_foreach_non_finite_check_and_unscale_
|
||||||
|
aten::_amp_update_scale_
|
||||||
|
aten::_assert_async
|
||||||
|
aten::_bmm
|
||||||
|
aten::_cat
|
||||||
|
aten::_cdist_backward
|
||||||
|
aten::_cdist_forward
|
||||||
|
aten::_cholesky_solve_helper
|
||||||
|
aten::_coalesce
|
||||||
|
aten::_coalesced_
|
||||||
|
aten::_compute_linear_combination
|
||||||
|
aten::_conj
|
||||||
|
aten::_conj_physical
|
||||||
|
aten::_copy_from
|
||||||
|
aten::_copy_from_and_resize
|
||||||
|
aten::_ctc_loss
|
||||||
|
aten::_ctc_loss_backward
|
||||||
|
aten::_cudnn_ctc_loss
|
||||||
|
aten::_cudnn_init_dropout_state
|
||||||
|
aten::_cudnn_rnn
|
||||||
|
aten::_cudnn_rnn_backward
|
||||||
|
aten::_cudnn_rnn_flatten_weight
|
||||||
|
aten::_cummax_helper
|
||||||
|
aten::_cummin_helper
|
||||||
|
aten::_cumprod
|
||||||
|
aten::_cumsum
|
||||||
|
aten::_dimI
|
||||||
|
aten::_dimV
|
||||||
|
aten::_dirichlet_grad
|
||||||
|
aten::_embedding_bag
|
||||||
|
aten::_embedding_bag_dense_backward
|
||||||
|
aten::_embedding_bag_forward_only
|
||||||
|
aten::_embedding_bag_per_sample_weights_backward
|
||||||
|
aten::_empty_affine_quantized
|
||||||
|
aten::_empty_per_channel_affine_quantized
|
||||||
|
aten::_euclidean_dist
|
||||||
|
aten::_fake_quantize_learnable_per_channel_affine
|
||||||
|
aten::_fake_quantize_learnable_per_tensor_affine
|
||||||
|
aten::_fft_c2c
|
||||||
|
aten::_fft_c2r
|
||||||
|
aten::_fft_r2c
|
||||||
|
aten::_foreach_abs
|
||||||
|
aten::_foreach_abs_
|
||||||
|
aten::_foreach_acos
|
||||||
|
aten::_foreach_acos_
|
||||||
|
aten::_foreach_add
|
||||||
|
aten::_foreach_add_
|
||||||
|
aten::_foreach_addcdiv
|
||||||
|
aten::_foreach_addcdiv_
|
||||||
|
aten::_foreach_addcmul
|
||||||
|
aten::_foreach_addcmul_
|
||||||
|
aten::_foreach_asin
|
||||||
|
aten::_foreach_asin_
|
||||||
|
aten::_foreach_atan
|
||||||
|
aten::_foreach_atan_
|
||||||
|
aten::_foreach_ceil
|
||||||
|
aten::_foreach_ceil_
|
||||||
|
aten::_foreach_cos
|
||||||
|
aten::_foreach_cos_
|
||||||
|
aten::_foreach_cosh
|
||||||
|
aten::_foreach_cosh_
|
||||||
|
aten::_foreach_div
|
||||||
|
aten::_foreach_div_
|
||||||
|
aten::_foreach_erf
|
||||||
|
aten::_foreach_erf_
|
||||||
|
aten::_foreach_erfc
|
||||||
|
aten::_foreach_erfc_
|
||||||
|
aten::_foreach_exp
|
||||||
|
aten::_foreach_exp_
|
||||||
|
aten::_foreach_expm1
|
||||||
|
aten::_foreach_expm1_
|
||||||
|
aten::_foreach_floor
|
||||||
|
aten::_foreach_floor_
|
||||||
|
aten::_foreach_frac
|
||||||
|
aten::_foreach_frac_
|
||||||
|
aten::_foreach_lgamma
|
||||||
|
aten::_foreach_lgamma_
|
||||||
|
aten::_foreach_log
|
||||||
|
aten::_foreach_log10
|
||||||
|
aten::_foreach_log10_
|
||||||
|
aten::_foreach_log1p
|
||||||
|
aten::_foreach_log1p_
|
||||||
|
aten::_foreach_log2
|
||||||
|
aten::_foreach_log2_
|
||||||
|
aten::_foreach_log_
|
||||||
|
aten::_foreach_maximum
|
||||||
|
aten::_foreach_minimum
|
||||||
|
aten::_foreach_mul
|
||||||
|
aten::_foreach_mul_
|
||||||
|
aten::_foreach_neg
|
||||||
|
aten::_foreach_neg_
|
||||||
|
aten::_foreach_reciprocal
|
||||||
|
aten::_foreach_reciprocal_
|
||||||
|
aten::_foreach_round
|
||||||
|
aten::_foreach_round_
|
||||||
|
aten::_foreach_sigmoid
|
||||||
|
aten::_foreach_sigmoid_
|
||||||
|
aten::_foreach_sin
|
||||||
|
aten::_foreach_sin_
|
||||||
|
aten::_foreach_sinh
|
||||||
|
aten::_foreach_sinh_
|
||||||
|
aten::_foreach_sqrt
|
||||||
|
aten::_foreach_sqrt_
|
||||||
|
aten::_foreach_sub
|
||||||
|
aten::_foreach_sub_
|
||||||
|
aten::_foreach_tan
|
||||||
|
aten::_foreach_tan_
|
||||||
|
aten::_foreach_tanh
|
||||||
|
aten::_foreach_tanh_
|
||||||
|
aten::_foreach_trunc
|
||||||
|
aten::_foreach_trunc_
|
||||||
|
aten::_foreach_zero_
|
||||||
|
aten::_fused_dropout
|
||||||
|
aten::_fw_primal
|
||||||
|
aten::_grid_sampler_2d_cpu_fallback
|
||||||
|
aten::_index_copy_
|
||||||
|
aten::_index_put_impl_
|
||||||
|
aten::_indices
|
||||||
|
aten::_inverse_helper
|
||||||
|
aten::_linalg_inv_out_helper_
|
||||||
|
aten::_linalg_qr_helper
|
||||||
|
aten::_linalg_solve_out_helper_
|
||||||
|
aten::_local_scalar_dense
|
||||||
|
aten::_log_softmax
|
||||||
|
aten::_log_softmax_backward_data
|
||||||
|
aten::_logcumsumexp
|
||||||
|
aten::_lu_with_info
|
||||||
|
aten::_make_per_channel_quantized_tensor
|
||||||
|
aten::_make_per_tensor_quantized_tensor
|
||||||
|
aten::_masked_scale
|
||||||
|
aten::_mkldnn_reshape
|
||||||
|
aten::_mkldnn_transpose
|
||||||
|
aten::_mkldnn_transpose_
|
||||||
|
aten::_nnpack_spatial_convolution
|
||||||
|
aten::_nnz
|
||||||
|
aten::_pack_padded_sequence
|
||||||
|
aten::_pdist_backward
|
||||||
|
aten::_pdist_forward
|
||||||
|
aten::_s_where
|
||||||
|
aten::_sample_dirichlet
|
||||||
|
aten::_segment_reduce_backward
|
||||||
|
aten::_softmax
|
||||||
|
aten::_softmax_backward_data
|
||||||
|
aten::_solve_helper
|
||||||
|
aten::_sparse_addmm
|
||||||
|
aten::_sparse_coo_tensor_with_dims
|
||||||
|
aten::_sparse_coo_tensor_with_dims_and_tensors
|
||||||
|
aten::_sparse_log_softmax
|
||||||
|
aten::_sparse_log_softmax_backward_data
|
||||||
|
aten::_sparse_mask_helper
|
||||||
|
aten::_sparse_softmax
|
||||||
|
aten::_sparse_softmax_backward_data
|
||||||
|
aten::_sparse_sparse_matmul
|
||||||
|
aten::_sparse_sum
|
||||||
|
aten::_sparse_sum_backward
|
||||||
|
aten::_stack
|
||||||
|
aten::_standard_gamma
|
||||||
|
aten::_standard_gamma_grad
|
||||||
|
aten::_svd_helper
|
||||||
|
aten::_symeig_helper
|
||||||
|
aten::_test_optional_filled_intlist
|
||||||
|
aten::_test_optional_floatlist
|
||||||
|
aten::_test_optional_intlist
|
||||||
|
aten::_thnn_fused_gru_cell
|
||||||
|
aten::_thnn_fused_gru_cell_backward
|
||||||
|
aten::_thnn_fused_lstm_cell
|
||||||
|
aten::_thnn_fused_lstm_cell_backward
|
||||||
|
aten::_trilinear
|
||||||
|
aten::_unique
|
||||||
|
aten::_unique2
|
||||||
|
aten::_unsafe_view
|
||||||
|
aten::_use_cudnn_ctc_loss
|
||||||
|
aten::_values
|
||||||
|
aten::_view_as_real_physical
|
||||||
|
aten::_weight_norm_cuda_interface
|
||||||
|
aten::_weight_norm_cuda_interface_backward
|
||||||
|
aten::abs
|
||||||
|
aten::abs_
|
||||||
|
aten::acos
|
||||||
|
aten::acos_
|
||||||
|
aten::acosh
|
||||||
|
aten::acosh_
|
||||||
|
aten::adaptive_avg_pool2d
|
||||||
|
aten::adaptive_avg_pool3d
|
||||||
|
aten::adaptive_avg_pool3d_backward
|
||||||
|
aten::adaptive_max_pool2d
|
||||||
|
aten::adaptive_max_pool2d_backward
|
||||||
|
aten::adaptive_max_pool3d
|
||||||
|
aten::adaptive_max_pool3d_backward
|
||||||
|
aten::add
|
||||||
|
aten::add_
|
||||||
|
aten::addbmm
|
||||||
|
aten::addbmm_
|
||||||
|
aten::addcdiv
|
||||||
|
aten::addcdiv_
|
||||||
|
aten::addcmul
|
||||||
|
aten::addcmul_
|
||||||
|
aten::addmm
|
||||||
|
aten::addmm_
|
||||||
|
aten::addmv
|
||||||
|
aten::addmv_
|
||||||
|
aten::addr_
|
||||||
|
aten::affine_grid_generator
|
||||||
|
aten::alias
|
||||||
|
aten::all
|
||||||
|
aten::amax
|
||||||
|
aten::amin
|
||||||
|
aten::angle
|
||||||
|
aten::any
|
||||||
|
aten::arange
|
||||||
|
aten::argmax
|
||||||
|
aten::argmin
|
||||||
|
aten::as_strided
|
||||||
|
aten::as_strided_
|
||||||
|
aten::asin
|
||||||
|
aten::asin_
|
||||||
|
aten::asinh
|
||||||
|
aten::asinh_
|
||||||
|
aten::atan
|
||||||
|
aten::atan2
|
||||||
|
aten::atan2_
|
||||||
|
aten::atan_
|
||||||
|
aten::atanh
|
||||||
|
aten::atanh_
|
||||||
|
aten::avg_pool2d
|
||||||
|
aten::avg_pool2d_backward
|
||||||
|
aten::avg_pool3d
|
||||||
|
aten::avg_pool3d_backward
|
||||||
|
aten::baddbmm
|
||||||
|
aten::baddbmm_
|
||||||
|
aten::batch_norm_backward_elemt
|
||||||
|
aten::batch_norm_backward_reduce
|
||||||
|
aten::batch_norm_elemt
|
||||||
|
aten::batch_norm_gather_stats
|
||||||
|
aten::batch_norm_gather_stats_with_counts
|
||||||
|
aten::batch_norm_stats
|
||||||
|
aten::batch_norm_update_stats
|
||||||
|
aten::bernoulli
|
||||||
|
aten::bernoulli_
|
||||||
|
aten::binary_cross_entropy
|
||||||
|
aten::binary_cross_entropy_backward
|
||||||
|
aten::binary_cross_entropy_with_logits
|
||||||
|
aten::bincount
|
||||||
|
aten::binomial
|
||||||
|
aten::bitwise_and
|
||||||
|
aten::bitwise_not
|
||||||
|
aten::bitwise_not_
|
||||||
|
aten::bitwise_or
|
||||||
|
aten::bitwise_xor
|
||||||
|
aten::bmm
|
||||||
|
aten::bucketize
|
||||||
|
aten::cat
|
||||||
|
aten::cauchy_
|
||||||
|
aten::ceil
|
||||||
|
aten::ceil_
|
||||||
|
aten::celu
|
||||||
|
aten::celu_
|
||||||
|
aten::channel_shuffle
|
||||||
|
aten::cholesky
|
||||||
|
aten::cholesky_inverse
|
||||||
|
aten::cholesky_solve
|
||||||
|
aten::clamp
|
||||||
|
aten::clamp_
|
||||||
|
aten::clamp_max
|
||||||
|
aten::clamp_max_
|
||||||
|
aten::clamp_min
|
||||||
|
aten::clamp_min_
|
||||||
|
aten::clone
|
||||||
|
aten::col2im
|
||||||
|
aten::col2im_backward
|
||||||
|
aten::col_indices
|
||||||
|
aten::complex
|
||||||
|
aten::conj_physical
|
||||||
|
aten::conj_physical_
|
||||||
|
aten::constant_pad_nd
|
||||||
|
aten::conv_depthwise3d
|
||||||
|
aten::conv_depthwise3d_backward
|
||||||
|
aten::conv_tbc
|
||||||
|
aten::convolution_backward_overrideable
|
||||||
|
aten::convolution_overrideable
|
||||||
|
aten::copy_
|
||||||
|
aten::copy_sparse_to_sparse_
|
||||||
|
aten::copysign
|
||||||
|
aten::copysign_
|
||||||
|
aten::cos
|
||||||
|
aten::cos_
|
||||||
|
aten::cosh
|
||||||
|
aten::cosh_
|
||||||
|
aten::count_nonzero
|
||||||
|
aten::cross
|
||||||
|
aten::crow_indices
|
||||||
|
aten::cudnn_affine_grid_generator
|
||||||
|
aten::cudnn_affine_grid_generator_backward
|
||||||
|
aten::cudnn_batch_norm
|
||||||
|
aten::cudnn_batch_norm_backward
|
||||||
|
aten::cudnn_convolution
|
||||||
|
aten::cudnn_convolution_add_relu
|
||||||
|
aten::cudnn_convolution_backward
|
||||||
|
aten::cudnn_convolution_backward_input
|
||||||
|
aten::cudnn_convolution_backward_weight
|
||||||
|
aten::cudnn_convolution_relu
|
||||||
|
aten::cudnn_convolution_transpose
|
||||||
|
aten::cudnn_convolution_transpose_backward
|
||||||
|
aten::cudnn_convolution_transpose_backward_input
|
||||||
|
aten::cudnn_convolution_transpose_backward_weight
|
||||||
|
aten::cudnn_grid_sampler
|
||||||
|
aten::cudnn_grid_sampler_backward
|
||||||
|
aten::cummax
|
||||||
|
aten::cummin
|
||||||
|
aten::cumprod
|
||||||
|
aten::cumprod_
|
||||||
|
aten::cumsum
|
||||||
|
aten::cumsum_
|
||||||
|
aten::deg2rad
|
||||||
|
aten::deg2rad_
|
||||||
|
aten::dense_dim
|
||||||
|
aten::dequantize
|
||||||
|
aten::detach
|
||||||
|
aten::detach_
|
||||||
|
aten::diag
|
||||||
|
aten::diagonal
|
||||||
|
aten::digamma
|
||||||
|
aten::digamma_
|
||||||
|
aten::dist
|
||||||
|
aten::div
|
||||||
|
aten::div_
|
||||||
|
aten::dot
|
||||||
|
aten::eig
|
||||||
|
aten::elu
|
||||||
|
aten::elu_
|
||||||
|
aten::elu_backward
|
||||||
|
aten::embedding
|
||||||
|
aten::embedding_dense_backward
|
||||||
|
aten::embedding_renorm_
|
||||||
|
aten::empty
|
||||||
|
aten::empty_quantized
|
||||||
|
aten::empty_strided
|
||||||
|
aten::eq
|
||||||
|
aten::eq_
|
||||||
|
aten::equal
|
||||||
|
aten::erf
|
||||||
|
aten::erf_
|
||||||
|
aten::erfc
|
||||||
|
aten::erfc_
|
||||||
|
aten::erfinv
|
||||||
|
aten::erfinv_
|
||||||
|
aten::exp
|
||||||
|
aten::exp2
|
||||||
|
aten::exp2_
|
||||||
|
aten::exp_
|
||||||
|
aten::expand
|
||||||
|
aten::expm1
|
||||||
|
aten::expm1_
|
||||||
|
aten::exponential_
|
||||||
|
aten::eye
|
||||||
|
aten::fake_quantize_per_channel_affine_cachemask
|
||||||
|
aten::fake_quantize_per_tensor_affine_cachemask
|
||||||
|
aten::fill_
|
||||||
|
aten::flip
|
||||||
|
aten::floor
|
||||||
|
aten::floor_
|
||||||
|
aten::floor_divide
|
||||||
|
aten::floor_divide_
|
||||||
|
aten::fmax
|
||||||
|
aten::fmin
|
||||||
|
aten::fmod
|
||||||
|
aten::fmod_
|
||||||
|
aten::frac
|
||||||
|
aten::frac_
|
||||||
|
aten::fractional_max_pool2d
|
||||||
|
aten::fractional_max_pool2d_backward
|
||||||
|
aten::fractional_max_pool3d
|
||||||
|
aten::fractional_max_pool3d_backward
|
||||||
|
aten::frexp
|
||||||
|
aten::from_file
|
||||||
|
aten::gather
|
||||||
|
aten::gcd
|
||||||
|
aten::gcd_
|
||||||
|
aten::ge
|
||||||
|
aten::ge_
|
||||||
|
aten::gelu
|
||||||
|
aten::gelu_backward
|
||||||
|
aten::geometric_
|
||||||
|
aten::geqrf
|
||||||
|
aten::get_gradients
|
||||||
|
aten::glu
|
||||||
|
aten::glu_backward
|
||||||
|
aten::grid_sampler_2d
|
||||||
|
aten::grid_sampler_2d_backward
|
||||||
|
aten::grid_sampler_3d
|
||||||
|
aten::grid_sampler_3d_backward
|
||||||
|
aten::gt
|
||||||
|
aten::gt_
|
||||||
|
aten::hardshrink
|
||||||
|
aten::hardshrink_backward
|
||||||
|
aten::hardsigmoid
|
||||||
|
aten::hardsigmoid_
|
||||||
|
aten::hardsigmoid_backward
|
||||||
|
aten::hardswish
|
||||||
|
aten::hardswish_
|
||||||
|
aten::hardswish_backward
|
||||||
|
aten::hardtanh
|
||||||
|
aten::hardtanh_
|
||||||
|
aten::hardtanh_backward
|
||||||
|
aten::heaviside
|
||||||
|
aten::heaviside_
|
||||||
|
aten::histc
|
||||||
|
aten::hspmm
|
||||||
|
aten::huber_loss
|
||||||
|
aten::huber_loss_backward
|
||||||
|
aten::hypot
|
||||||
|
aten::hypot_
|
||||||
|
aten::i0
|
||||||
|
aten::i0_
|
||||||
|
aten::igamma
|
||||||
|
aten::igamma_
|
||||||
|
aten::igammac
|
||||||
|
aten::igammac_
|
||||||
|
aten::im2col
|
||||||
|
aten::im2col_backward
|
||||||
|
aten::index
|
||||||
|
aten::index_add_
|
||||||
|
aten::index_copy_
|
||||||
|
aten::index_fill_
|
||||||
|
aten::index_put_
|
||||||
|
aten::index_select
|
||||||
|
aten::indices
|
||||||
|
aten::int_repr
|
||||||
|
aten::inverse
|
||||||
|
aten::is_coalesced
|
||||||
|
aten::is_set_to
|
||||||
|
aten::isin
|
||||||
|
aten::isnan
|
||||||
|
aten::isneginf
|
||||||
|
aten::isposinf
|
||||||
|
aten::kl_div
|
||||||
|
aten::kl_div_backward
|
||||||
|
aten::kthvalue
|
||||||
|
aten::l1_loss
|
||||||
|
aten::l1_loss_backward
|
||||||
|
aten::lcm
|
||||||
|
aten::lcm_
|
||||||
|
aten::le
|
||||||
|
aten::le_
|
||||||
|
aten::leaky_relu
|
||||||
|
aten::leaky_relu_
|
||||||
|
aten::leaky_relu_backward
|
||||||
|
aten::lerp
|
||||||
|
aten::lerp_
|
||||||
|
aten::lgamma
|
||||||
|
aten::lgamma_
|
||||||
|
aten::linalg_cholesky_ex
|
||||||
|
aten::linalg_det
|
||||||
|
aten::linalg_eig
|
||||||
|
aten::linalg_eigh
|
||||||
|
aten::linalg_householder_product
|
||||||
|
aten::linalg_inv_ex
|
||||||
|
aten::linalg_lstsq
|
||||||
|
aten::linalg_qr
|
||||||
|
aten::linalg_slogdet
|
||||||
|
aten::linalg_solve
|
||||||
|
aten::linalg_vector_norm
|
||||||
|
aten::linspace
|
||||||
|
aten::log
|
||||||
|
aten::log10
|
||||||
|
aten::log10_
|
||||||
|
aten::log1p
|
||||||
|
aten::log1p_
|
||||||
|
aten::log2
|
||||||
|
aten::log2_
|
||||||
|
aten::log_
|
||||||
|
aten::log_normal_
|
||||||
|
aten::log_sigmoid_backward
|
||||||
|
aten::log_sigmoid_forward
|
||||||
|
aten::logaddexp
|
||||||
|
aten::logaddexp2
|
||||||
|
aten::logcumsumexp
|
||||||
|
aten::logdet
|
||||||
|
aten::logical_and
|
||||||
|
aten::logical_not
|
||||||
|
aten::logical_or
|
||||||
|
aten::logical_xor
|
||||||
|
aten::logit
|
||||||
|
aten::logit_
|
||||||
|
aten::logit_backward
|
||||||
|
aten::logspace
|
||||||
|
aten::logsumexp
|
||||||
|
aten::lstsq
|
||||||
|
aten::lt
|
||||||
|
aten::lt_
|
||||||
|
aten::lu_solve
|
||||||
|
aten::lu_unpack
|
||||||
|
aten::masked_fill_
|
||||||
|
aten::masked_scatter_
|
||||||
|
aten::masked_select
|
||||||
|
aten::matrix_exp
|
||||||
|
aten::max
|
||||||
|
aten::max_pool2d_with_indices
|
||||||
|
aten::max_pool2d_with_indices_backward
|
||||||
|
aten::max_pool3d_with_indices
|
||||||
|
aten::max_pool3d_with_indices_backward
|
||||||
|
aten::max_unpool2d
|
||||||
|
aten::max_unpool2d_backward
|
||||||
|
aten::max_unpool3d
|
||||||
|
aten::max_unpool3d_backward
|
||||||
|
aten::maximum
|
||||||
|
aten::mean
|
||||||
|
aten::median
|
||||||
|
aten::min
|
||||||
|
aten::minimum
|
||||||
|
aten::miopen_batch_norm
|
||||||
|
aten::miopen_batch_norm_backward
|
||||||
|
aten::miopen_convolution
|
||||||
|
aten::miopen_convolution_backward
|
||||||
|
aten::miopen_convolution_backward_bias
|
||||||
|
aten::miopen_convolution_backward_input
|
||||||
|
aten::miopen_convolution_backward_weight
|
||||||
|
aten::miopen_convolution_transpose
|
||||||
|
aten::miopen_convolution_transpose_backward
|
||||||
|
aten::miopen_convolution_transpose_backward_input
|
||||||
|
aten::miopen_convolution_transpose_backward_weight
|
||||||
|
aten::miopen_depthwise_convolution
|
||||||
|
aten::miopen_depthwise_convolution_backward
|
||||||
|
aten::miopen_depthwise_convolution_backward_input
|
||||||
|
aten::miopen_depthwise_convolution_backward_weight
|
||||||
|
aten::miopen_rnn
|
||||||
|
aten::miopen_rnn_backward
|
||||||
|
aten::mish
|
||||||
|
aten::mish_
|
||||||
|
aten::mkldnn_adaptive_avg_pool2d
|
||||||
|
aten::mkldnn_adaptive_avg_pool2d_backward
|
||||||
|
aten::mkldnn_convolution
|
||||||
|
aten::mkldnn_convolution_backward
|
||||||
|
aten::mkldnn_linear
|
||||||
|
aten::mkldnn_linear_backward
|
||||||
|
aten::mkldnn_linear_backward_input
|
||||||
|
aten::mkldnn_linear_backward_weights
|
||||||
|
aten::mkldnn_max_pool2d
|
||||||
|
aten::mkldnn_max_pool2d_backward
|
||||||
|
aten::mkldnn_max_pool3d
|
||||||
|
aten::mkldnn_max_pool3d_backward
|
||||||
|
aten::mkldnn_reorder_conv2d_weight
|
||||||
|
aten::mkldnn_reorder_conv3d_weight
|
||||||
|
aten::mm
|
||||||
|
aten::mode
|
||||||
|
aten::mse_loss
|
||||||
|
aten::mse_loss_backward
|
||||||
|
aten::mul
|
||||||
|
aten::mul_
|
||||||
|
aten::multi_margin_loss
|
||||||
|
aten::multi_margin_loss_backward
|
||||||
|
aten::multilabel_margin_loss_backward
|
||||||
|
aten::multilabel_margin_loss_forward
|
||||||
|
aten::multinomial
|
||||||
|
aten::mv
|
||||||
|
aten::mvlgamma
|
||||||
|
aten::mvlgamma_
|
||||||
|
aten::nan_to_num
|
||||||
|
aten::nan_to_num_
|
||||||
|
aten::nanmedian
|
||||||
|
aten::nansum
|
||||||
|
aten::narrow_copy
|
||||||
|
aten::native_batch_norm
|
||||||
|
aten::native_batch_norm_backward
|
||||||
|
aten::native_group_norm_backward
|
||||||
|
aten::native_layer_norm_backward
|
||||||
|
aten::native_norm
|
||||||
|
aten::ne
|
||||||
|
aten::ne_
|
||||||
|
aten::neg
|
||||||
|
aten::neg_
|
||||||
|
aten::nextafter
|
||||||
|
aten::nextafter_
|
||||||
|
aten::nll_loss2d_backward
|
||||||
|
aten::nll_loss2d_forward
|
||||||
|
aten::nll_loss_backward
|
||||||
|
aten::nll_loss_forward
|
||||||
|
aten::nonzero
|
||||||
|
aten::norm
|
||||||
|
aten::normal
|
||||||
|
aten::normal_
|
||||||
|
aten::ormqr
|
||||||
|
aten::permute
|
||||||
|
aten::poisson
|
||||||
|
aten::polar
|
||||||
|
aten::polygamma
|
||||||
|
aten::polygamma_
|
||||||
|
aten::pow
|
||||||
|
aten::pow_
|
||||||
|
aten::prelu
|
||||||
|
aten::prelu_backward
|
||||||
|
aten::prod
|
||||||
|
aten::put_
|
||||||
|
aten::q_per_channel_axis
|
||||||
|
aten::q_per_channel_scales
|
||||||
|
aten::q_per_channel_zero_points
|
||||||
|
aten::q_scale
|
||||||
|
aten::q_zero_point
|
||||||
|
aten::qscheme
|
||||||
|
aten::quantize_per_channel
|
||||||
|
aten::quantize_per_tensor
|
||||||
|
aten::quantized_batch_norm
|
||||||
|
aten::quantized_gru
|
||||||
|
aten::quantized_lstm
|
||||||
|
aten::quantized_max_pool1d
|
||||||
|
aten::quantized_max_pool2d
|
||||||
|
aten::rad2deg
|
||||||
|
aten::rad2deg_
|
||||||
|
aten::random_
|
||||||
|
aten::randperm
|
||||||
|
aten::range
|
||||||
|
aten::reciprocal
|
||||||
|
aten::reciprocal_
|
||||||
|
aten::record_stream
|
||||||
|
aten::reflection_pad1d
|
||||||
|
aten::reflection_pad1d_backward
|
||||||
|
aten::reflection_pad2d
|
||||||
|
aten::reflection_pad2d_backward
|
||||||
|
aten::relu
|
||||||
|
aten::relu_
|
||||||
|
aten::remainder
|
||||||
|
aten::remainder_
|
||||||
|
aten::renorm
|
||||||
|
aten::renorm_
|
||||||
|
aten::repeat
|
||||||
|
aten::repeat_interleave
|
||||||
|
aten::replication_pad1d
|
||||||
|
aten::replication_pad1d_backward
|
||||||
|
aten::replication_pad2d
|
||||||
|
aten::replication_pad2d_backward
|
||||||
|
aten::replication_pad3d
|
||||||
|
aten::replication_pad3d_backward
|
||||||
|
aten::resize_
|
||||||
|
aten::resize_as_
|
||||||
|
aten::resize_as_sparse_
|
||||||
|
aten::roll
|
||||||
|
aten::rot90
|
||||||
|
aten::round
|
||||||
|
aten::round_
|
||||||
|
aten::rrelu_with_noise
|
||||||
|
aten::rrelu_with_noise_
|
||||||
|
aten::rrelu_with_noise_backward
|
||||||
|
aten::rsqrt
|
||||||
|
aten::rsqrt_
|
||||||
|
aten::rsub
|
||||||
|
aten::scatter
|
||||||
|
aten::scatter_
|
||||||
|
aten::scatter_add
|
||||||
|
aten::scatter_add_
|
||||||
|
aten::searchsorted
|
||||||
|
aten::segment_reduce
|
||||||
|
aten::select
|
||||||
|
aten::set_
|
||||||
|
aten::sgn
|
||||||
|
aten::sgn_
|
||||||
|
aten::sigmoid
|
||||||
|
aten::sigmoid_
|
||||||
|
aten::sigmoid_backward
|
||||||
|
aten::sign
|
||||||
|
aten::sign_
|
||||||
|
aten::signbit
|
||||||
|
aten::silu
|
||||||
|
aten::silu_
|
||||||
|
aten::sin
|
||||||
|
aten::sin_
|
||||||
|
aten::sinc
|
||||||
|
aten::sinc_
|
||||||
|
aten::sinh
|
||||||
|
aten::sinh_
|
||||||
|
aten::slice
|
||||||
|
aten::slogdet
|
||||||
|
aten::slow_conv3d_backward
|
||||||
|
aten::slow_conv3d_forward
|
||||||
|
aten::slow_conv_dilated2d
|
||||||
|
aten::slow_conv_dilated2d_backward
|
||||||
|
aten::slow_conv_dilated3d
|
||||||
|
aten::slow_conv_dilated3d_backward
|
||||||
|
aten::slow_conv_transpose2d
|
||||||
|
aten::slow_conv_transpose2d_backward
|
||||||
|
aten::slow_conv_transpose3d
|
||||||
|
aten::slow_conv_transpose3d_backward
|
||||||
|
aten::smooth_l1_loss
|
||||||
|
aten::smooth_l1_loss_backward
|
||||||
|
aten::soft_margin_loss
|
||||||
|
aten::soft_margin_loss_backward
|
||||||
|
aten::softplus
|
||||||
|
aten::softplus_backward
|
||||||
|
aten::softshrink
|
||||||
|
aten::softshrink_backward
|
||||||
|
aten::solve
|
||||||
|
aten::sort
|
||||||
|
aten::sparse_dim
|
||||||
|
aten::sparse_mask
|
||||||
|
aten::sparse_resize_
|
||||||
|
aten::sparse_resize_and_clear_
|
||||||
|
aten::special_entr
|
||||||
|
aten::special_i0e
|
||||||
|
aten::special_i1
|
||||||
|
aten::special_i1e
|
||||||
|
aten::special_xlog1py
|
||||||
|
aten::split
|
||||||
|
aten::split_with_sizes
|
||||||
|
aten::sqrt
|
||||||
|
aten::sqrt_
|
||||||
|
aten::square
|
||||||
|
aten::squeeze
|
||||||
|
aten::squeeze_
|
||||||
|
aten::sspaddmm
|
||||||
|
aten::stack
|
||||||
|
aten::std
|
||||||
|
aten::std_mean
|
||||||
|
aten::sub
|
||||||
|
aten::sub_
|
||||||
|
aten::sum
|
||||||
|
aten::symeig
|
||||||
|
aten::t
|
||||||
|
aten::t_
|
||||||
|
aten::take
|
||||||
|
aten::tan
|
||||||
|
aten::tan_
|
||||||
|
aten::tanh
|
||||||
|
aten::tanh_
|
||||||
|
aten::tanh_backward
|
||||||
|
aten::tensordot
|
||||||
|
aten::thnn_conv2d_backward
|
||||||
|
aten::thnn_conv2d_forward
|
||||||
|
aten::thnn_conv_depthwise2d_backward
|
||||||
|
aten::thnn_conv_depthwise2d_forward
|
||||||
|
aten::threshold
|
||||||
|
aten::threshold_
|
||||||
|
aten::threshold_backward
|
||||||
|
aten::to_dense
|
||||||
|
aten::to_mkldnn
|
||||||
|
aten::to_sparse
|
||||||
|
aten::topk
|
||||||
|
aten::trace
|
||||||
|
aten::transpose
|
||||||
|
aten::transpose_
|
||||||
|
aten::triangular_solve
|
||||||
|
aten::tril
|
||||||
|
aten::tril_
|
||||||
|
aten::tril_indices
|
||||||
|
aten::triu
|
||||||
|
aten::triu_
|
||||||
|
aten::triu_indices
|
||||||
|
aten::trunc
|
||||||
|
aten::trunc_
|
||||||
|
aten::unbind
|
||||||
|
aten::unfold
|
||||||
|
aten::unfold_backward
|
||||||
|
aten::uniform_
|
||||||
|
aten::unique_consecutive
|
||||||
|
aten::unique_dim
|
||||||
|
aten::unique_dim_consecutive
|
||||||
|
aten::unsafe_split
|
||||||
|
aten::unsafe_split_with_sizes
|
||||||
|
aten::unsqueeze
|
||||||
|
aten::unsqueeze_
|
||||||
|
aten::upsample_bicubic2d
|
||||||
|
aten::upsample_bicubic2d_backward
|
||||||
|
aten::upsample_bilinear2d
|
||||||
|
aten::upsample_bilinear2d_backward
|
||||||
|
aten::upsample_linear1d
|
||||||
|
aten::upsample_linear1d_backward
|
||||||
|
aten::upsample_nearest1d
|
||||||
|
aten::upsample_nearest1d_backward
|
||||||
|
aten::upsample_nearest2d
|
||||||
|
aten::upsample_nearest2d_backward
|
||||||
|
aten::upsample_nearest3d
|
||||||
|
aten::upsample_nearest3d_backward
|
||||||
|
aten::upsample_trilinear3d
|
||||||
|
aten::upsample_trilinear3d_backward
|
||||||
|
aten::values
|
||||||
|
aten::var
|
||||||
|
aten::var_mean
|
||||||
|
aten::vdot
|
||||||
|
aten::view
|
||||||
|
aten::view_as_complex
|
||||||
|
aten::view_as_real
|
||||||
|
aten::xlogy
|
||||||
|
aten::xlogy_
|
||||||
|
aten::zero_
|
Reference in New Issue
Block a user