mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Summary: * Add `torch._scaled_grouped_mm_v2` with more functionality and extensibility for future formats * Add `torch.nn.functional.scaled_grouped_mm` as public entrypoint * Test both original and v2 functionality Test Plan: ``` pytest -svv -k grouped test/test_scaled_matmul_cuda.py ``` Reviewers: Subscribers: Tasks: Tags: Signed-off-by: Simon Layton <simonlayton@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/165154 Approved by: https://github.com/drisspg, https://github.com/danielvegamyhre
1368 lines
38 KiB
Plaintext
1368 lines
38 KiB
Plaintext
aten::__lshift__.Scalar
|
|
aten::__lshift__.Scalar_out
|
|
aten::__lshift__.Tensor
|
|
aten::__lshift__.Tensor_out
|
|
aten::__rshift__.Scalar
|
|
aten::__rshift__.Scalar_out
|
|
aten::__rshift__.Tensor
|
|
aten::__rshift__.Tensor_out
|
|
aten::_adaptive_avg_pool2d_backward
|
|
aten::_adaptive_avg_pool2d_backward.out
|
|
aten::_adaptive_avg_pool3d
|
|
aten::_adaptive_avg_pool3d.out
|
|
aten::_adaptive_avg_pool3d_backward
|
|
aten::_adaptive_avg_pool3d_backward.out
|
|
aten::_add_relu.Scalar
|
|
aten::_add_relu.Scalar_out
|
|
aten::_add_relu.Tensor
|
|
aten::_add_relu.out
|
|
aten::_add_relu_.Scalar
|
|
aten::_add_relu_.Tensor
|
|
aten::_aminmax
|
|
aten::_aminmax.dim
|
|
aten::_aminmax.dim_out
|
|
aten::_aminmax.out
|
|
aten::_amp_foreach_non_finite_check_and_unscale
|
|
aten::_amp_foreach_non_finite_check_and_unscale.out
|
|
aten::_amp_foreach_non_finite_check_and_unscale_
|
|
aten::_amp_update_scale
|
|
aten::_amp_update_scale.out
|
|
aten::_amp_update_scale_
|
|
aten::_assert_async
|
|
aten::_assert_async.msg
|
|
aten::_assert_tensor_metadata
|
|
aten::_batch_norm_no_update.out
|
|
aten::_batch_norm_with_update.out
|
|
aten::_cdist_backward
|
|
aten::_cdist_backward.out
|
|
aten::_cdist_forward
|
|
aten::_cdist_forward.out
|
|
aten::_cholesky_solve_helper
|
|
aten::_cholesky_solve_helper.out
|
|
aten::_coalesce
|
|
aten::_coalesce.out
|
|
aten::_coalesced
|
|
aten::_coalesced.out
|
|
aten::_coalesced_
|
|
aten::_compute_linear_combination
|
|
aten::_compute_linear_combination.out
|
|
aten::_conj
|
|
aten::_conj_copy
|
|
aten::_conj_copy.out
|
|
aten::_conj_physical
|
|
aten::_conj_physical.out
|
|
aten::_conv_depthwise2d
|
|
aten::_conv_depthwise2d.out
|
|
aten::_convert_indices_from_coo_to_csr
|
|
aten::_convert_indices_from_coo_to_csr.out
|
|
aten::_convert_indices_from_csr_to_coo
|
|
aten::_convert_indices_from_csr_to_coo.out
|
|
aten::_convert_weight_to_int4pack
|
|
aten::_convert_weight_to_int4pack_for_cpu
|
|
aten::_convolution
|
|
aten::_convolution.out
|
|
aten::_copy_from
|
|
aten::_copy_from.out
|
|
aten::_copy_from_and_resize
|
|
aten::_copy_from_and_resize.out
|
|
aten::_cslt_compress
|
|
aten::_cslt_sparse_mm
|
|
aten::_cslt_sparse_mm_search
|
|
aten::_ctc_loss
|
|
aten::_ctc_loss.Tensor
|
|
aten::_ctc_loss.Tensor_out
|
|
aten::_ctc_loss.out
|
|
aten::_ctc_loss_backward
|
|
aten::_ctc_loss_backward.Tensor
|
|
aten::_ctc_loss_backward.out
|
|
aten::_cudnn_attention_backward
|
|
aten::_cudnn_attention_forward
|
|
aten::_cudnn_ctc_loss
|
|
aten::_cudnn_ctc_loss.Tensor
|
|
aten::_cudnn_ctc_loss.out
|
|
aten::_cudnn_init_dropout_state
|
|
aten::_cudnn_init_dropout_state.out
|
|
aten::_cudnn_rnn
|
|
aten::_cudnn_rnn.out
|
|
aten::_cudnn_rnn_backward
|
|
aten::_cudnn_rnn_backward.out
|
|
aten::_cudnn_rnn_flatten_weight
|
|
aten::_cudnn_rnn_flatten_weight.out
|
|
aten::_cummax_helper
|
|
aten::_cummin_helper
|
|
aten::_dimI
|
|
aten::_dimV
|
|
aten::_dirichlet_grad
|
|
aten::_dirichlet_grad.out
|
|
aten::_dyn_quant_matmul_4bit
|
|
aten::_dyn_quant_pack_4bit_weight
|
|
aten::_efficient_attention_backward
|
|
aten::_efficient_attention_forward
|
|
aten::_efficientzerotensor
|
|
aten::_efficientzerotensor.out
|
|
aten::_embedding_bag
|
|
aten::_embedding_bag.out
|
|
aten::_embedding_bag_backward
|
|
aten::_embedding_bag_dense_backward
|
|
aten::_embedding_bag_dense_backward.out
|
|
aten::_embedding_bag_forward_only
|
|
aten::_embedding_bag_forward_only.out
|
|
aten::_embedding_bag_per_sample_weights_backward
|
|
aten::_embedding_bag_per_sample_weights_backward.out
|
|
aten::_empty_affine_quantized
|
|
aten::_empty_affine_quantized.out
|
|
aten::_empty_per_channel_affine_quantized
|
|
aten::_empty_per_channel_affine_quantized.out
|
|
aten::_fake_quantize_learnable_per_channel_affine
|
|
aten::_fake_quantize_learnable_per_channel_affine.out
|
|
aten::_fake_quantize_learnable_per_channel_affine_backward
|
|
aten::_fake_quantize_learnable_per_tensor_affine
|
|
aten::_fake_quantize_learnable_per_tensor_affine.out
|
|
aten::_fake_quantize_learnable_per_tensor_affine_backward
|
|
aten::_fake_quantize_per_tensor_affine_cachemask_tensor_qparams
|
|
aten::_fake_quantize_per_tensor_affine_cachemask_tensor_qparams.out
|
|
aten::_fft_c2c
|
|
aten::_fft_c2c.out
|
|
aten::_fft_c2r
|
|
aten::_fft_c2r.out
|
|
aten::_fft_r2c
|
|
aten::_fft_r2c.out
|
|
aten::_fill_mem_eff_dropout_mask_
|
|
aten::_flash_attention_backward
|
|
aten::_flash_attention_forward
|
|
aten::_foobar
|
|
aten::_foobar.out
|
|
aten::_foreach_abs
|
|
aten::_foreach_abs.out
|
|
aten::_foreach_abs_
|
|
aten::_foreach_acos
|
|
aten::_foreach_acos.out
|
|
aten::_foreach_acos_
|
|
aten::_foreach_add.List
|
|
aten::_foreach_add.List_out
|
|
aten::_foreach_add.Scalar
|
|
aten::_foreach_add.ScalarList
|
|
aten::_foreach_add.ScalarList_out
|
|
aten::_foreach_add.Scalar_out
|
|
aten::_foreach_add.Tensor
|
|
aten::_foreach_add.Tensor_out
|
|
aten::_foreach_add_.List
|
|
aten::_foreach_add_.Scalar
|
|
aten::_foreach_add_.ScalarList
|
|
aten::_foreach_add_.Tensor
|
|
aten::_foreach_addcdiv.Scalar
|
|
aten::_foreach_addcdiv.ScalarList
|
|
aten::_foreach_addcdiv.ScalarList_out
|
|
aten::_foreach_addcdiv.Scalar_out
|
|
aten::_foreach_addcdiv.Tensor
|
|
aten::_foreach_addcdiv.Tensor_out
|
|
aten::_foreach_addcdiv_.Scalar
|
|
aten::_foreach_addcdiv_.ScalarList
|
|
aten::_foreach_addcdiv_.Tensor
|
|
aten::_foreach_addcmul.Scalar
|
|
aten::_foreach_addcmul.ScalarList
|
|
aten::_foreach_addcmul.ScalarList_out
|
|
aten::_foreach_addcmul.Scalar_out
|
|
aten::_foreach_addcmul.Tensor
|
|
aten::_foreach_addcmul.Tensor_out
|
|
aten::_foreach_addcmul_.Scalar
|
|
aten::_foreach_addcmul_.ScalarList
|
|
aten::_foreach_addcmul_.Tensor
|
|
aten::_foreach_asin
|
|
aten::_foreach_asin.out
|
|
aten::_foreach_asin_
|
|
aten::_foreach_atan
|
|
aten::_foreach_atan.out
|
|
aten::_foreach_atan_
|
|
aten::_foreach_ceil
|
|
aten::_foreach_ceil.out
|
|
aten::_foreach_ceil_
|
|
aten::_foreach_clamp_max.List
|
|
aten::_foreach_clamp_max.List_out
|
|
aten::_foreach_clamp_max.Scalar
|
|
aten::_foreach_clamp_max.ScalarList
|
|
aten::_foreach_clamp_max.ScalarList_out
|
|
aten::_foreach_clamp_max.Scalar_out
|
|
aten::_foreach_clamp_max_.List
|
|
aten::_foreach_clamp_max_.Scalar
|
|
aten::_foreach_clamp_max_.ScalarList
|
|
aten::_foreach_clamp_min.List
|
|
aten::_foreach_clamp_min.List_out
|
|
aten::_foreach_clamp_min.Scalar
|
|
aten::_foreach_clamp_min.ScalarList
|
|
aten::_foreach_clamp_min.ScalarList_out
|
|
aten::_foreach_clamp_min.Scalar_out
|
|
aten::_foreach_clamp_min_.List
|
|
aten::_foreach_clamp_min_.Scalar
|
|
aten::_foreach_clamp_min_.ScalarList
|
|
aten::_foreach_copy
|
|
aten::_foreach_copy.out
|
|
aten::_foreach_copy_
|
|
aten::_foreach_cos
|
|
aten::_foreach_cos.out
|
|
aten::_foreach_cos_
|
|
aten::_foreach_cosh
|
|
aten::_foreach_cosh.out
|
|
aten::_foreach_cosh_
|
|
aten::_foreach_div.List
|
|
aten::_foreach_div.List_out
|
|
aten::_foreach_div.Scalar
|
|
aten::_foreach_div.ScalarList
|
|
aten::_foreach_div.ScalarList_out
|
|
aten::_foreach_div.Scalar_out
|
|
aten::_foreach_div.Tensor
|
|
aten::_foreach_div.Tensor_out
|
|
aten::_foreach_div_.List
|
|
aten::_foreach_div_.Scalar
|
|
aten::_foreach_div_.ScalarList
|
|
aten::_foreach_div_.Tensor
|
|
aten::_foreach_erf
|
|
aten::_foreach_erf.out
|
|
aten::_foreach_erf_
|
|
aten::_foreach_erfc
|
|
aten::_foreach_erfc.out
|
|
aten::_foreach_erfc_
|
|
aten::_foreach_exp
|
|
aten::_foreach_exp.out
|
|
aten::_foreach_exp_
|
|
aten::_foreach_expm1
|
|
aten::_foreach_expm1.out
|
|
aten::_foreach_expm1_
|
|
aten::_foreach_floor
|
|
aten::_foreach_floor.out
|
|
aten::_foreach_floor_
|
|
aten::_foreach_frac
|
|
aten::_foreach_frac.out
|
|
aten::_foreach_frac_
|
|
aten::_foreach_lerp.List
|
|
aten::_foreach_lerp.List_out
|
|
aten::_foreach_lerp.Scalar
|
|
aten::_foreach_lerp.ScalarList
|
|
aten::_foreach_lerp.ScalarList_out
|
|
aten::_foreach_lerp.Scalar_out
|
|
aten::_foreach_lerp_.List
|
|
aten::_foreach_lerp_.Scalar
|
|
aten::_foreach_lerp_.ScalarList
|
|
aten::_foreach_lgamma
|
|
aten::_foreach_lgamma.out
|
|
aten::_foreach_lgamma_
|
|
aten::_foreach_log
|
|
aten::_foreach_log.out
|
|
aten::_foreach_log10
|
|
aten::_foreach_log10.out
|
|
aten::_foreach_log10_
|
|
aten::_foreach_log1p
|
|
aten::_foreach_log1p.out
|
|
aten::_foreach_log1p_
|
|
aten::_foreach_log2
|
|
aten::_foreach_log2.out
|
|
aten::_foreach_log2_
|
|
aten::_foreach_log_
|
|
aten::_foreach_max
|
|
aten::_foreach_max.out
|
|
aten::_foreach_maximum.List
|
|
aten::_foreach_maximum.List_out
|
|
aten::_foreach_maximum.Scalar
|
|
aten::_foreach_maximum.ScalarList
|
|
aten::_foreach_maximum.ScalarList_out
|
|
aten::_foreach_maximum.Scalar_out
|
|
aten::_foreach_maximum_.List
|
|
aten::_foreach_maximum_.Scalar
|
|
aten::_foreach_maximum_.ScalarList
|
|
aten::_foreach_minimum.List
|
|
aten::_foreach_minimum.List_out
|
|
aten::_foreach_minimum.Scalar
|
|
aten::_foreach_minimum.ScalarList
|
|
aten::_foreach_minimum.ScalarList_out
|
|
aten::_foreach_minimum.Scalar_out
|
|
aten::_foreach_minimum_.List
|
|
aten::_foreach_minimum_.Scalar
|
|
aten::_foreach_minimum_.ScalarList
|
|
aten::_foreach_mul.List
|
|
aten::_foreach_mul.List_out
|
|
aten::_foreach_mul.Scalar
|
|
aten::_foreach_mul.ScalarList
|
|
aten::_foreach_mul.ScalarList_out
|
|
aten::_foreach_mul.Scalar_out
|
|
aten::_foreach_mul.Tensor
|
|
aten::_foreach_mul.Tensor_out
|
|
aten::_foreach_mul_.List
|
|
aten::_foreach_mul_.Scalar
|
|
aten::_foreach_mul_.ScalarList
|
|
aten::_foreach_mul_.Tensor
|
|
aten::_foreach_neg
|
|
aten::_foreach_neg.out
|
|
aten::_foreach_neg_
|
|
aten::_foreach_norm.Scalar
|
|
aten::_foreach_norm.Scalar_out
|
|
aten::_foreach_pow.List
|
|
aten::_foreach_pow.List_out
|
|
aten::_foreach_pow.Scalar
|
|
aten::_foreach_pow.ScalarAndTensor
|
|
aten::_foreach_pow.ScalarList
|
|
aten::_foreach_pow.ScalarList_out
|
|
aten::_foreach_pow.Scalar_out
|
|
aten::_foreach_pow_.List
|
|
aten::_foreach_pow_.Scalar
|
|
aten::_foreach_pow_.ScalarList
|
|
aten::_foreach_reciprocal
|
|
aten::_foreach_reciprocal.out
|
|
aten::_foreach_reciprocal_
|
|
aten::_foreach_round
|
|
aten::_foreach_round.out
|
|
aten::_foreach_round_
|
|
aten::_foreach_rsqrt
|
|
aten::_foreach_rsqrt.out
|
|
aten::_foreach_rsqrt_
|
|
aten::_foreach_sigmoid
|
|
aten::_foreach_sigmoid.out
|
|
aten::_foreach_sigmoid_
|
|
aten::_foreach_sign
|
|
aten::_foreach_sign.out
|
|
aten::_foreach_sign_
|
|
aten::_foreach_sin
|
|
aten::_foreach_sin.out
|
|
aten::_foreach_sin_
|
|
aten::_foreach_sinh
|
|
aten::_foreach_sinh.out
|
|
aten::_foreach_sinh_
|
|
aten::_foreach_sqrt
|
|
aten::_foreach_sqrt.out
|
|
aten::_foreach_sqrt_
|
|
aten::_foreach_sub.List
|
|
aten::_foreach_sub.List_out
|
|
aten::_foreach_sub.Scalar
|
|
aten::_foreach_sub.ScalarList
|
|
aten::_foreach_sub.ScalarList_out
|
|
aten::_foreach_sub.Scalar_out
|
|
aten::_foreach_sub_.List
|
|
aten::_foreach_sub_.Scalar
|
|
aten::_foreach_sub_.ScalarList
|
|
aten::_foreach_tan
|
|
aten::_foreach_tan.out
|
|
aten::_foreach_tan_
|
|
aten::_foreach_tanh
|
|
aten::_foreach_tanh.out
|
|
aten::_foreach_tanh_
|
|
aten::_foreach_trunc
|
|
aten::_foreach_trunc.out
|
|
aten::_foreach_trunc_
|
|
aten::_foreach_zero
|
|
aten::_foreach_zero.out
|
|
aten::_foreach_zero_
|
|
aten::_functional_assert_async.msg
|
|
aten::_functional_assert_scalar
|
|
aten::_functional_sym_constrain_range
|
|
aten::_functional_sym_constrain_range_for_size
|
|
aten::_fused_adagrad
|
|
aten::_fused_adagrad.out
|
|
aten::_fused_adagrad.tensor_lr
|
|
aten::_fused_adagrad.tensor_lr_out
|
|
aten::_fused_adagrad_
|
|
aten::_fused_adagrad_.tensor_lr
|
|
aten::_fused_adam
|
|
aten::_fused_adam.out
|
|
aten::_fused_adam.tensor_lr
|
|
aten::_fused_adam.tensor_lr_out
|
|
aten::_fused_adam_
|
|
aten::_fused_adam_.tensor_lr
|
|
aten::_fused_adamw
|
|
aten::_fused_adamw.out
|
|
aten::_fused_adamw.tensor_lr
|
|
aten::_fused_adamw.tensor_lr_out
|
|
aten::_fused_adamw_
|
|
aten::_fused_adamw_.tensor_lr
|
|
aten::_fused_moving_avg_obs_fq_helper
|
|
aten::_fused_moving_avg_obs_fq_helper.out
|
|
aten::_fused_moving_avg_obs_fq_helper_functional
|
|
aten::_fused_sdp_choice
|
|
aten::_fused_sgd
|
|
aten::_fused_sgd.out
|
|
aten::_fused_sgd.tensor_lr
|
|
aten::_fused_sgd.tensor_lr_out
|
|
aten::_fused_sgd_
|
|
aten::_fused_sgd_.tensor_lr
|
|
aten::_fw_primal
|
|
aten::_fw_primal_copy
|
|
aten::_fw_primal_copy.out
|
|
aten::_grid_sampler_2d_cpu_fallback
|
|
aten::_grid_sampler_2d_cpu_fallback.out
|
|
aten::_grouped_mm
|
|
aten::_has_same_storage_numel
|
|
aten::_histogramdd_bin_edges
|
|
aten::_histogramdd_bin_edges.out
|
|
aten::_histogramdd_from_bin_cts
|
|
aten::_histogramdd_from_bin_cts.out
|
|
aten::_histogramdd_from_bin_tensors
|
|
aten::_histogramdd_from_bin_tensors.out
|
|
aten::_index_put_impl
|
|
aten::_index_put_impl.out
|
|
aten::_index_put_impl_
|
|
aten::_indices
|
|
aten::_indices_copy
|
|
aten::_indices_copy.out
|
|
aten::_int_mm
|
|
aten::_int_mm.out
|
|
aten::_is_all_true
|
|
aten::_is_any_true
|
|
aten::_jagged_to_padded_dense_forward
|
|
aten::_lazy_clone
|
|
aten::_linalg_check_errors
|
|
aten::_linalg_det
|
|
aten::_linalg_det.result
|
|
aten::_linalg_eigh
|
|
aten::_linalg_eigh.eigenvalues
|
|
aten::_linalg_eigvals
|
|
aten::_linalg_slogdet
|
|
aten::_linalg_slogdet.sign
|
|
aten::_linalg_solve_ex
|
|
aten::_linalg_solve_ex.result
|
|
aten::_linalg_svd
|
|
aten::_linalg_svd.U
|
|
aten::_local_scalar_dense
|
|
aten::_logcumsumexp
|
|
aten::_logcumsumexp.out
|
|
aten::_lstm_mps
|
|
aten::_lstm_mps.out
|
|
aten::_make_dep_token
|
|
aten::_make_dual
|
|
aten::_make_dual_copy
|
|
aten::_make_dual_copy.out
|
|
aten::_make_per_channel_quantized_tensor
|
|
aten::_make_per_channel_quantized_tensor.out
|
|
aten::_make_per_tensor_quantized_tensor
|
|
aten::_make_per_tensor_quantized_tensor.out
|
|
aten::_masked_scale
|
|
aten::_masked_scale.out
|
|
aten::_masked_softmax
|
|
aten::_masked_softmax.out
|
|
aten::_masked_softmax_backward
|
|
aten::_masked_softmax_backward.out
|
|
aten::_mixed_dtypes_linear
|
|
aten::_mkldnn_reshape
|
|
aten::_mkldnn_reshape.out
|
|
aten::_mkldnn_transpose
|
|
aten::_mkldnn_transpose.out
|
|
aten::_mkldnn_transpose_
|
|
aten::_mps_convolution
|
|
aten::_mps_convolution.out
|
|
aten::_mps_convolution_transpose
|
|
aten::_mps_convolution_transpose.out
|
|
aten::_native_batch_norm_legit.no_stats_out
|
|
aten::_native_batch_norm_legit.out
|
|
aten::_native_batch_norm_legit_no_training.out
|
|
aten::_native_multi_head_attention
|
|
aten::_native_multi_head_attention.out
|
|
aten::_neg_view
|
|
aten::_neg_view_copy
|
|
aten::_neg_view_copy.out
|
|
aten::_nested_compute_contiguous_strides_offsets
|
|
aten::_nested_from_padded
|
|
aten::_nested_from_padded.out
|
|
aten::_nested_from_padded_and_nested_example
|
|
aten::_nested_from_padded_and_nested_example.out
|
|
aten::_nested_from_padded_tensor
|
|
aten::_nested_get_jagged_dummy
|
|
aten::_nested_get_lengths
|
|
aten::_nested_get_max_seqlen
|
|
aten::_nested_get_min_seqlen
|
|
aten::_nested_get_offsets
|
|
aten::_nested_get_ragged_idx
|
|
aten::_nested_get_values
|
|
aten::_nested_get_values_copy
|
|
aten::_nested_get_values_copy.out
|
|
aten::_nested_select_backward
|
|
aten::_nested_sum_backward
|
|
aten::_nested_tensor_from_mask
|
|
aten::_nested_tensor_from_mask.out
|
|
aten::_nested_tensor_from_mask_left_aligned
|
|
aten::_nested_tensor_from_tensor_list
|
|
aten::_nested_tensor_from_tensor_list.out
|
|
aten::_nested_tensor_size
|
|
aten::_nested_tensor_size.out
|
|
aten::_nested_tensor_softmax_with_shape
|
|
aten::_nested_tensor_storage_offsets
|
|
aten::_nested_tensor_storage_offsets.out
|
|
aten::_nested_tensor_strides
|
|
aten::_nested_tensor_strides.out
|
|
aten::_nested_view_from_buffer
|
|
aten::_nested_view_from_buffer_copy
|
|
aten::_nested_view_from_buffer_copy.out
|
|
aten::_nested_view_from_jagged
|
|
aten::_nested_view_from_jagged_copy
|
|
aten::_nested_view_from_jagged_copy.out
|
|
aten::_new_zeros_with_same_feature_meta
|
|
aten::_new_zeros_with_same_feature_meta.out
|
|
aten::_nnpack_spatial_convolution
|
|
aten::_nnpack_spatial_convolution.out
|
|
aten::_nnz
|
|
aten::_pack_padded_sequence
|
|
aten::_pack_padded_sequence.out
|
|
aten::_padded_dense_to_jagged_forward
|
|
aten::_pdist_backward
|
|
aten::_pdist_backward.out
|
|
aten::_pdist_forward
|
|
aten::_pdist_forward.out
|
|
aten::_pin_memory
|
|
aten::_pin_memory.out
|
|
aten::_reshape_alias_copy
|
|
aten::_reshape_alias_copy.out
|
|
aten::_reshape_copy
|
|
aten::_resize_output
|
|
aten::_resize_output.out
|
|
aten::_resize_output_
|
|
aten::_sample_dirichlet
|
|
aten::_sample_dirichlet.out
|
|
aten::_scaled_dot_product_attention_math_for_mps
|
|
aten::_scaled_dot_product_cudnn_attention
|
|
aten::_scaled_dot_product_cudnn_attention_backward
|
|
aten::_scaled_dot_product_efficient_attention
|
|
aten::_scaled_dot_product_efficient_attention_backward
|
|
aten::_scaled_dot_product_flash_attention
|
|
aten::_scaled_dot_product_flash_attention_backward
|
|
aten::_scaled_dot_product_flash_attention_for_cpu_backward
|
|
aten::_scaled_dot_product_fused_attention_overrideable
|
|
aten::_scaled_dot_product_fused_attention_overrideable_backward
|
|
aten::_scaled_grouped_mm
|
|
aten::_scaled_grouped_mm_v2
|
|
aten::_scaled_mm
|
|
aten::_scaled_mm.out
|
|
aten::_scaled_mm_v2
|
|
aten::_scaled_mm_v2.out
|
|
aten::_segment_reduce_backward
|
|
aten::_segment_reduce_backward.out
|
|
aten::_slow_conv2d_backward.grad_input
|
|
aten::_slow_conv2d_backward.output_mask
|
|
aten::_slow_conv2d_backward.output_mask_out
|
|
aten::_slow_conv2d_forward
|
|
aten::_slow_conv2d_forward.output
|
|
aten::_sparse_addmm
|
|
aten::_sparse_addmm.out
|
|
aten::_sparse_broadcast_to
|
|
aten::_sparse_broadcast_to_copy
|
|
aten::_sparse_broadcast_to_copy.out
|
|
aten::_sparse_compressed_tensor_with_dims
|
|
aten::_sparse_coo_tensor_with_dims
|
|
aten::_sparse_coo_tensor_with_dims.out
|
|
aten::_sparse_coo_tensor_with_dims_and_tensors
|
|
aten::_sparse_coo_tensor_with_dims_and_tensors.out
|
|
aten::_sparse_csr_prod.dim_dtype
|
|
aten::_sparse_csr_prod.dim_dtype_out
|
|
aten::_sparse_csr_sum.dim_dtype
|
|
aten::_sparse_csr_sum.dim_dtype_out
|
|
aten::_sparse_log_softmax
|
|
aten::_sparse_log_softmax.out
|
|
aten::_sparse_log_softmax_backward_data
|
|
aten::_sparse_log_softmax_backward_data.out
|
|
aten::_sparse_mask_projection
|
|
aten::_sparse_mask_projection.out
|
|
aten::_sparse_mm_reduce_impl
|
|
aten::_sparse_mm_reduce_impl_backward
|
|
aten::_sparse_semi_structured_addmm
|
|
aten::_sparse_semi_structured_apply
|
|
aten::_sparse_semi_structured_apply_dense
|
|
aten::_sparse_semi_structured_linear
|
|
aten::_sparse_semi_structured_mm
|
|
aten::_sparse_semi_structured_tile
|
|
aten::_sparse_softmax
|
|
aten::_sparse_softmax.out
|
|
aten::_sparse_softmax_backward_data
|
|
aten::_sparse_softmax_backward_data.out
|
|
aten::_sparse_sparse_matmul
|
|
aten::_sparse_sparse_matmul.out
|
|
aten::_sparse_sum.dim
|
|
aten::_sparse_sum.dim_out
|
|
aten::_sparse_sum_backward
|
|
aten::_sparse_sum_backward.out
|
|
aten::_spdiags
|
|
aten::_spdiags.out
|
|
aten::_spsolve
|
|
aten::_stack
|
|
aten::_stack.out
|
|
aten::_standard_gamma
|
|
aten::_standard_gamma.out
|
|
aten::_standard_gamma_grad
|
|
aten::_standard_gamma_grad.out
|
|
aten::_test_autograd_multiple_dispatch.fullcoverage
|
|
aten::_test_autograd_multiple_dispatch.fullcoverage_out
|
|
aten::_test_autograd_multiple_dispatch_view
|
|
aten::_test_autograd_multiple_dispatch_view_copy
|
|
aten::_test_autograd_multiple_dispatch_view_copy.out
|
|
aten::_test_functorch_fallback
|
|
aten::_test_functorch_fallback.out
|
|
aten::_test_optional_filled_intlist
|
|
aten::_test_optional_filled_intlist.out
|
|
aten::_test_optional_floatlist
|
|
aten::_test_optional_floatlist.out
|
|
aten::_test_optional_intlist
|
|
aten::_test_optional_intlist.out
|
|
aten::_test_parallel_materialize
|
|
aten::_test_warn_in_autograd
|
|
aten::_test_warn_in_autograd.out
|
|
aten::_thnn_fused_gru_cell
|
|
aten::_thnn_fused_gru_cell.out
|
|
aten::_thnn_fused_gru_cell_backward
|
|
aten::_thnn_fused_gru_cell_backward.out
|
|
aten::_thnn_fused_lstm_cell
|
|
aten::_thnn_fused_lstm_cell.out
|
|
aten::_thnn_fused_lstm_cell_backward_impl
|
|
aten::_thnn_fused_lstm_cell_backward_impl.out
|
|
aten::_to_dense
|
|
aten::_to_dense.out
|
|
aten::_to_sparse
|
|
aten::_to_sparse.out
|
|
aten::_to_sparse.sparse_dim
|
|
aten::_to_sparse.sparse_dim_out
|
|
aten::_to_sparse_bsc
|
|
aten::_to_sparse_bsc.out
|
|
aten::_to_sparse_bsr
|
|
aten::_to_sparse_bsr.out
|
|
aten::_to_sparse_csc
|
|
aten::_to_sparse_csc.out
|
|
aten::_to_sparse_csr
|
|
aten::_to_sparse_csr.out
|
|
aten::_to_sparse_semi_structured
|
|
aten::_transform_bias_rescale_qkv
|
|
aten::_transform_bias_rescale_qkv.out
|
|
aten::_transformer_encoder_layer_fwd
|
|
aten::_transformer_encoder_layer_fwd.out
|
|
aten::_trilinear
|
|
aten::_trilinear.out
|
|
aten::_triton_multi_head_attention
|
|
aten::_triton_multi_head_attention.out
|
|
aten::_triton_scaled_dot_attention
|
|
aten::_triton_scaled_dot_attention.out
|
|
aten::_unique
|
|
aten::_unique.out
|
|
aten::_unique2
|
|
aten::_unique2.out
|
|
aten::_upsample_bicubic2d_aa
|
|
aten::_upsample_bicubic2d_aa.out
|
|
aten::_upsample_bicubic2d_aa_backward
|
|
aten::_upsample_bicubic2d_aa_backward.grad_input
|
|
aten::_upsample_bilinear2d_aa
|
|
aten::_upsample_bilinear2d_aa.out
|
|
aten::_upsample_bilinear2d_aa_backward
|
|
aten::_upsample_bilinear2d_aa_backward.grad_input
|
|
aten::_upsample_nearest_exact1d_backward
|
|
aten::_upsample_nearest_exact1d_backward.grad_input
|
|
aten::_upsample_nearest_exact2d_backward
|
|
aten::_upsample_nearest_exact2d_backward.grad_input
|
|
aten::_upsample_nearest_exact3d_backward
|
|
aten::_upsample_nearest_exact3d_backward.grad_input
|
|
aten::_use_cudnn_ctc_loss
|
|
aten::_use_cudnn_ctc_loss.Tensor
|
|
aten::_validate_compressed_sparse_indices
|
|
aten::_values
|
|
aten::_values_copy
|
|
aten::_values_copy.out
|
|
aten::_weight_int4pack_mm
|
|
aten::_weight_int4pack_mm_for_cpu
|
|
aten::_weight_int4pack_mm_with_scales_and_zeros
|
|
aten::_weight_int8pack_mm
|
|
aten::_weight_norm_interface_backward
|
|
aten::_weight_norm_interface_backward.out
|
|
aten::adaptive_avg_pool1d.out
|
|
aten::adaptive_avg_pool2d.out
|
|
aten::adaptive_avg_pool3d.out
|
|
aten::adaptive_avg_pool3d_backward.grad_input
|
|
aten::adaptive_max_pool2d
|
|
aten::adaptive_max_pool2d.out
|
|
aten::adaptive_max_pool2d_backward
|
|
aten::adaptive_max_pool2d_backward.grad_input
|
|
aten::adaptive_max_pool3d
|
|
aten::adaptive_max_pool3d.out
|
|
aten::adaptive_max_pool3d_backward
|
|
aten::adaptive_max_pool3d_backward.grad_input
|
|
aten::addbmm
|
|
aten::addbmm.out
|
|
aten::addr_
|
|
aten::allclose
|
|
aten::angle
|
|
aten::angle.out
|
|
aten::argmax
|
|
aten::argmax.out
|
|
aten::argmin
|
|
aten::argmin.out
|
|
aten::as_strided
|
|
aten::as_strided_
|
|
aten::avg_pool1d.out
|
|
aten::avg_pool2d
|
|
aten::avg_pool2d.out
|
|
aten::avg_pool2d_backward
|
|
aten::avg_pool2d_backward.grad_input
|
|
aten::avg_pool3d
|
|
aten::avg_pool3d.out
|
|
aten::avg_pool3d_backward
|
|
aten::avg_pool3d_backward.grad_input
|
|
aten::bartlett_window
|
|
aten::bartlett_window.out
|
|
aten::bartlett_window.periodic
|
|
aten::bartlett_window.periodic_out
|
|
aten::batch_norm_backward_elemt
|
|
aten::batch_norm_backward_elemt.out
|
|
aten::batch_norm_backward_reduce
|
|
aten::batch_norm_backward_reduce.out
|
|
aten::batch_norm_elemt
|
|
aten::batch_norm_elemt.out
|
|
aten::batch_norm_gather_stats
|
|
aten::batch_norm_gather_stats.out
|
|
aten::batch_norm_gather_stats_with_counts
|
|
aten::batch_norm_gather_stats_with_counts.out
|
|
aten::batch_norm_stats
|
|
aten::batch_norm_stats.out
|
|
aten::batch_norm_update_stats
|
|
aten::batch_norm_update_stats.out
|
|
aten::bernoulli.Tensor
|
|
aten::bernoulli.Tensor_out
|
|
aten::bernoulli.float_out
|
|
aten::bernoulli.out
|
|
aten::bernoulli.p
|
|
aten::bernoulli_.Tensor
|
|
aten::bernoulli_.float
|
|
aten::bincount
|
|
aten::bincount.out
|
|
aten::binomial
|
|
aten::binomial.out
|
|
aten::blackman_window
|
|
aten::blackman_window.out
|
|
aten::blackman_window.periodic
|
|
aten::blackman_window.periodic_out
|
|
aten::bmm
|
|
aten::bmm.dtype
|
|
aten::bmm.dtype_out
|
|
aten::bmm.out
|
|
aten::ccol_indices
|
|
aten::ccol_indices_copy
|
|
aten::ccol_indices_copy.out
|
|
aten::cholesky
|
|
aten::cholesky.out
|
|
aten::cholesky_inverse
|
|
aten::cholesky_inverse.out
|
|
aten::cholesky_solve
|
|
aten::cholesky_solve.out
|
|
aten::col_indices
|
|
aten::col_indices_copy
|
|
aten::col_indices_copy.out
|
|
aten::conv_depthwise3d
|
|
aten::conv_depthwise3d.out
|
|
aten::conv_tbc
|
|
aten::conv_tbc.out
|
|
aten::convolution
|
|
aten::convolution.out
|
|
aten::convolution_backward
|
|
aten::convolution_backward.out
|
|
aten::convolution_backward_overrideable
|
|
aten::convolution_backward_overrideable.out
|
|
aten::convolution_overrideable
|
|
aten::convolution_overrideable.out
|
|
aten::copy
|
|
aten::copy.out
|
|
aten::copy_
|
|
aten::copy_sparse_to_sparse
|
|
aten::copy_sparse_to_sparse.out
|
|
aten::copy_sparse_to_sparse_
|
|
aten::crow_indices
|
|
aten::crow_indices_copy
|
|
aten::crow_indices_copy.out
|
|
aten::cudnn_affine_grid_generator
|
|
aten::cudnn_affine_grid_generator.out
|
|
aten::cudnn_affine_grid_generator_backward
|
|
aten::cudnn_affine_grid_generator_backward.out
|
|
aten::cudnn_convolution
|
|
aten::cudnn_convolution.out
|
|
aten::cudnn_convolution_add_relu
|
|
aten::cudnn_convolution_add_relu.out
|
|
aten::cudnn_convolution_relu
|
|
aten::cudnn_convolution_relu.out
|
|
aten::cudnn_convolution_transpose
|
|
aten::cudnn_convolution_transpose.out
|
|
aten::cudnn_grid_sampler
|
|
aten::cudnn_grid_sampler.out
|
|
aten::cudnn_grid_sampler_backward
|
|
aten::cudnn_grid_sampler_backward.out
|
|
aten::cummax
|
|
aten::cummax.out
|
|
aten::cummin
|
|
aten::cummin.out
|
|
aten::dense_dim
|
|
aten::dequantize.self
|
|
aten::dequantize.self_out
|
|
aten::dequantize.tensors
|
|
aten::dequantize.tensors_out
|
|
aten::detach_
|
|
aten::detach_copy
|
|
aten::detach_copy.out
|
|
aten::embedding_renorm
|
|
aten::embedding_renorm.out
|
|
aten::embedding_renorm_
|
|
aten::empty.memory_format
|
|
aten::empty.names
|
|
aten::empty.names_out
|
|
aten::empty_permuted
|
|
aten::empty_permuted.out
|
|
aten::empty_quantized
|
|
aten::empty_quantized.out
|
|
aten::equal
|
|
aten::fake_quantize_per_channel_affine_cachemask
|
|
aten::fake_quantize_per_channel_affine_cachemask.out
|
|
aten::fake_quantize_per_tensor_affine_cachemask
|
|
aten::fake_quantize_per_tensor_affine_cachemask.out
|
|
aten::fft_fftfreq
|
|
aten::fft_fftfreq.out
|
|
aten::fft_rfftfreq
|
|
aten::fft_rfftfreq.out
|
|
aten::fill.Scalar_out
|
|
aten::fill.Tensor_out
|
|
aten::fractional_max_pool2d
|
|
aten::fractional_max_pool2d.output
|
|
aten::fractional_max_pool2d_backward
|
|
aten::fractional_max_pool2d_backward.grad_input
|
|
aten::fractional_max_pool3d
|
|
aten::fractional_max_pool3d.output
|
|
aten::fractional_max_pool3d_backward
|
|
aten::fractional_max_pool3d_backward.grad_input
|
|
aten::from_file
|
|
aten::from_file.out
|
|
aten::full.names
|
|
aten::full.names_out
|
|
aten::full_like
|
|
aten::full_like.out
|
|
aten::gather
|
|
aten::gather.out
|
|
aten::geqrf
|
|
aten::geqrf.a
|
|
aten::glu_backward_jvp
|
|
aten::glu_backward_jvp.out
|
|
aten::glu_jvp
|
|
aten::glu_jvp.out
|
|
aten::grid_sampler_2d_backward
|
|
aten::grid_sampler_2d_backward.out
|
|
aten::grid_sampler_3d
|
|
aten::grid_sampler_3d.out
|
|
aten::grid_sampler_3d_backward
|
|
aten::grid_sampler_3d_backward.out
|
|
aten::hamming_window
|
|
aten::hamming_window.out
|
|
aten::hamming_window.periodic
|
|
aten::hamming_window.periodic_alpha
|
|
aten::hamming_window.periodic_alpha_beta
|
|
aten::hamming_window.periodic_alpha_beta_out
|
|
aten::hamming_window.periodic_alpha_out
|
|
aten::hamming_window.periodic_out
|
|
aten::hann_window
|
|
aten::hann_window.out
|
|
aten::hann_window.periodic
|
|
aten::hann_window.periodic_out
|
|
aten::hardshrink_backward
|
|
aten::hardshrink_backward.grad_input
|
|
aten::hash_tensor
|
|
aten::hash_tensor.out
|
|
aten::histc
|
|
aten::histc.out
|
|
aten::histogram.bin_ct
|
|
aten::histogram.bin_ct_out
|
|
aten::histogram.bins_tensor
|
|
aten::histogram.bins_tensor_out
|
|
aten::hspmm
|
|
aten::hspmm.out
|
|
aten::index.Tensor
|
|
aten::index.Tensor_out
|
|
aten::index_put
|
|
aten::index_put.out
|
|
aten::index_reduce
|
|
aten::index_reduce.out
|
|
aten::indices
|
|
aten::indices_copy
|
|
aten::indices_copy.out
|
|
aten::int_repr
|
|
aten::int_repr.out
|
|
aten::is_coalesced
|
|
aten::is_pinned
|
|
aten::is_set_to
|
|
aten::kaiser_window
|
|
aten::kaiser_window.beta
|
|
aten::kaiser_window.beta_out
|
|
aten::kaiser_window.out
|
|
aten::kaiser_window.periodic
|
|
aten::kaiser_window.periodic_out
|
|
aten::kthvalue
|
|
aten::kthvalue.values
|
|
aten::lift_fresh_copy
|
|
aten::lift_fresh_copy.out
|
|
aten::linalg_cholesky_ex
|
|
aten::linalg_cholesky_ex.L
|
|
aten::linalg_eig
|
|
aten::linalg_eig.out
|
|
aten::linalg_eigvals.out
|
|
aten::linalg_householder_product
|
|
aten::linalg_householder_product.out
|
|
aten::linalg_inv_ex
|
|
aten::linalg_inv_ex.inverse
|
|
aten::linalg_ldl_factor_ex
|
|
aten::linalg_ldl_factor_ex.out
|
|
aten::linalg_ldl_solve
|
|
aten::linalg_ldl_solve.out
|
|
aten::linalg_lstsq
|
|
aten::linalg_lstsq.out
|
|
aten::linalg_lu
|
|
aten::linalg_lu.out
|
|
aten::linalg_lu_factor_ex
|
|
aten::linalg_lu_factor_ex.out
|
|
aten::linalg_lu_solve
|
|
aten::linalg_lu_solve.out
|
|
aten::linalg_matrix_exp
|
|
aten::linalg_matrix_exp.out
|
|
aten::linalg_pinv.atol_rtol_tensor
|
|
aten::linalg_pinv.atol_rtol_tensor_out
|
|
aten::linalg_qr
|
|
aten::linalg_qr.out
|
|
aten::linalg_solve_triangular
|
|
aten::linalg_solve_triangular.out
|
|
aten::linear.out
|
|
aten::linear_backward
|
|
aten::linear_backward.out
|
|
aten::log_softmax.int_out
|
|
aten::logcumsumexp
|
|
aten::logcumsumexp.out
|
|
aten::logit_backward.grad_input
|
|
aten::lstm_mps_backward
|
|
aten::lstm_mps_backward.out
|
|
aten::lu_unpack
|
|
aten::lu_unpack.out
|
|
aten::masked_scatter
|
|
aten::masked_scatter.out
|
|
aten::masked_scatter_
|
|
aten::masked_scatter_backward
|
|
aten::masked_select
|
|
aten::masked_select.out
|
|
aten::matmul_backward
|
|
aten::matmul_backward.out
|
|
aten::max
|
|
aten::max.dim
|
|
aten::max.dim_max
|
|
aten::max.unary_out
|
|
aten::max_pool2d_backward
|
|
aten::max_pool2d_backward.out
|
|
aten::max_pool2d_with_indices
|
|
aten::max_pool2d_with_indices.out
|
|
aten::max_pool2d_with_indices_backward
|
|
aten::max_pool2d_with_indices_backward.grad_input
|
|
aten::max_pool3d_with_indices
|
|
aten::max_pool3d_with_indices.out
|
|
aten::max_pool3d_with_indices_backward
|
|
aten::max_pool3d_with_indices_backward.grad_input
|
|
aten::median
|
|
aten::median.dim
|
|
aten::median.dim_values
|
|
aten::median.out
|
|
aten::min
|
|
aten::min.dim
|
|
aten::min.dim_min
|
|
aten::min.unary_out
|
|
aten::miopen_batch_norm
|
|
aten::miopen_batch_norm.out
|
|
aten::miopen_convolution
|
|
aten::miopen_convolution.out
|
|
aten::miopen_convolution_add_relu
|
|
aten::miopen_convolution_relu
|
|
aten::miopen_convolution_transpose
|
|
aten::miopen_convolution_transpose.out
|
|
aten::miopen_depthwise_convolution
|
|
aten::miopen_depthwise_convolution.out
|
|
aten::miopen_rnn
|
|
aten::miopen_rnn.out
|
|
aten::miopen_rnn_backward
|
|
aten::miopen_rnn_backward.out
|
|
aten::mkldnn_adaptive_avg_pool2d
|
|
aten::mkldnn_adaptive_avg_pool2d.out
|
|
aten::mkldnn_adaptive_avg_pool2d_backward
|
|
aten::mkldnn_adaptive_avg_pool2d_backward.out
|
|
aten::mkldnn_convolution
|
|
aten::mkldnn_convolution.out
|
|
aten::mkldnn_linear
|
|
aten::mkldnn_linear.out
|
|
aten::mkldnn_linear_backward
|
|
aten::mkldnn_linear_backward.out
|
|
aten::mkldnn_linear_backward_input
|
|
aten::mkldnn_linear_backward_input.out
|
|
aten::mkldnn_linear_backward_weights
|
|
aten::mkldnn_linear_backward_weights.out
|
|
aten::mkldnn_max_pool2d
|
|
aten::mkldnn_max_pool2d.out
|
|
aten::mkldnn_max_pool2d_backward
|
|
aten::mkldnn_max_pool2d_backward.out
|
|
aten::mkldnn_max_pool3d
|
|
aten::mkldnn_max_pool3d.out
|
|
aten::mkldnn_max_pool3d_backward
|
|
aten::mkldnn_max_pool3d_backward.out
|
|
aten::mkldnn_reorder_conv2d_weight
|
|
aten::mkldnn_reorder_conv2d_weight.out
|
|
aten::mkldnn_reorder_conv3d_weight
|
|
aten::mkldnn_reorder_conv3d_weight.out
|
|
aten::mkldnn_rnn_layer
|
|
aten::mkldnn_rnn_layer.out
|
|
aten::mkldnn_rnn_layer_backward
|
|
aten::mkldnn_rnn_layer_backward.out
|
|
aten::mm
|
|
aten::mm.dtype
|
|
aten::mm.dtype_out
|
|
aten::mm.out
|
|
aten::mode
|
|
aten::mode.values
|
|
aten::mps_convolution_backward
|
|
aten::mps_convolution_backward.out
|
|
aten::mps_convolution_transpose_backward
|
|
aten::mps_convolution_transpose_backward.out
|
|
aten::multi_margin_loss_backward
|
|
aten::multi_margin_loss_backward.grad_input
|
|
aten::multilabel_margin_loss_backward
|
|
aten::multilabel_margin_loss_backward.grad_input
|
|
aten::multinomial
|
|
aten::multinomial.out
|
|
aten::nanmedian
|
|
aten::nanmedian.dim
|
|
aten::nanmedian.dim_values
|
|
aten::nanmedian.out
|
|
aten::native_group_norm.out
|
|
aten::native_norm
|
|
aten::native_norm.ScalarOpt_dim_dtype
|
|
aten::native_norm.ScalarOpt_dim_dtype_out
|
|
aten::native_norm.out
|
|
aten::nonzero
|
|
aten::nonzero.out
|
|
aten::nonzero_static
|
|
aten::nonzero_static.out
|
|
aten::normal_functional
|
|
aten::ones.names
|
|
aten::ones.names_out
|
|
aten::ones.out
|
|
aten::ormqr
|
|
aten::ormqr.out
|
|
aten::poisson
|
|
aten::poisson.out
|
|
aten::polygamma
|
|
aten::polygamma.out
|
|
aten::polygamma_
|
|
aten::put
|
|
aten::put.out
|
|
aten::put_
|
|
aten::q_per_channel_axis
|
|
aten::q_per_channel_scales
|
|
aten::q_per_channel_scales.out
|
|
aten::q_per_channel_zero_points
|
|
aten::q_per_channel_zero_points.out
|
|
aten::q_scale
|
|
aten::q_zero_point
|
|
aten::qscheme
|
|
aten::quantize_per_channel
|
|
aten::quantize_per_channel.out
|
|
aten::quantize_per_tensor
|
|
aten::quantize_per_tensor.out
|
|
aten::quantize_per_tensor.tensor_qparams
|
|
aten::quantize_per_tensor.tensor_qparams_out
|
|
aten::quantize_per_tensor.tensors
|
|
aten::quantize_per_tensor.tensors_out
|
|
aten::quantize_per_tensor_dynamic
|
|
aten::quantize_per_tensor_dynamic.out
|
|
aten::quantized_batch_norm
|
|
aten::quantized_batch_norm.out
|
|
aten::quantized_gru.data
|
|
aten::quantized_gru.data_legacy
|
|
aten::quantized_gru.input
|
|
aten::quantized_gru.input_legacy
|
|
aten::quantized_lstm.data
|
|
aten::quantized_lstm.data_legacy
|
|
aten::quantized_lstm.input
|
|
aten::quantized_lstm.input_legacy
|
|
aten::quantized_max_pool1d
|
|
aten::quantized_max_pool1d.out
|
|
aten::quantized_max_pool2d
|
|
aten::quantized_max_pool2d.out
|
|
aten::quantized_max_pool3d
|
|
aten::quantized_max_pool3d.out
|
|
aten::rand
|
|
aten::rand.generator
|
|
aten::rand.generator_with_names
|
|
aten::rand.generator_with_names_out
|
|
aten::rand.names
|
|
aten::rand.names_out
|
|
aten::rand.out
|
|
aten::rand_like
|
|
aten::rand_like.out
|
|
aten::randint
|
|
aten::randint.generator
|
|
aten::randint.generator_out
|
|
aten::randint.low
|
|
aten::randint.low_generator
|
|
aten::randint.low_generator_out
|
|
aten::randint.low_out
|
|
aten::randint.out
|
|
aten::randint_like
|
|
aten::randint_like.Tensor
|
|
aten::randint_like.Tensor_out
|
|
aten::randint_like.low_dtype
|
|
aten::randint_like.low_dtype_out
|
|
aten::randint_like.out
|
|
aten::randn.generator
|
|
aten::randn.generator_with_names
|
|
aten::randn.generator_with_names_out
|
|
aten::randn.names
|
|
aten::randn.names_out
|
|
aten::randn_like
|
|
aten::randn_like.out
|
|
aten::random
|
|
aten::random.from
|
|
aten::random.from_out
|
|
aten::random.out
|
|
aten::random.to
|
|
aten::random.to_out
|
|
aten::random_
|
|
aten::random_.from
|
|
aten::random_.to
|
|
aten::randperm
|
|
aten::randperm.generator
|
|
aten::randperm.generator_out
|
|
aten::randperm.out
|
|
aten::range
|
|
aten::range.out
|
|
aten::range.out_
|
|
aten::range.step
|
|
aten::record_stream
|
|
aten::repeat_interleave.Tensor
|
|
aten::repeat_interleave.Tensor_out
|
|
aten::replication_pad1d_backward
|
|
aten::replication_pad1d_backward.grad_input
|
|
aten::replication_pad2d_backward
|
|
aten::replication_pad2d_backward.grad_input
|
|
aten::replication_pad3d_backward
|
|
aten::replication_pad3d_backward.grad_input
|
|
aten::resize
|
|
aten::resize.out
|
|
aten::resize_
|
|
aten::resize_as_
|
|
aten::resize_as_sparse
|
|
aten::resize_as_sparse.out
|
|
aten::resize_as_sparse_
|
|
aten::row_indices
|
|
aten::row_indices_copy
|
|
aten::row_indices_copy.out
|
|
aten::rrelu_with_noise
|
|
aten::rrelu_with_noise.out
|
|
aten::rrelu_with_noise_
|
|
aten::rrelu_with_noise_functional
|
|
aten::scalar_tensor
|
|
aten::scalar_tensor.out
|
|
aten::scatter.reduce
|
|
aten::scatter.reduce_out
|
|
aten::scatter.src
|
|
aten::scatter.src_out
|
|
aten::scatter.value
|
|
aten::scatter.value_out
|
|
aten::scatter.value_reduce
|
|
aten::scatter.value_reduce_out
|
|
aten::scatter_add
|
|
aten::scatter_add.out
|
|
aten::scatter_reduce.two
|
|
aten::scatter_reduce.two_out
|
|
aten::searchsorted.Scalar
|
|
aten::searchsorted.Scalar_out
|
|
aten::searchsorted.Tensor
|
|
aten::searchsorted.Tensor_out
|
|
aten::segment_reduce
|
|
aten::segment_reduce.out
|
|
aten::select.int
|
|
aten::select_copy.int
|
|
aten::select_copy.int_out
|
|
aten::set
|
|
aten::set.out
|
|
aten::set.source_Storage
|
|
aten::set.source_Storage_out
|
|
aten::set.source_Storage_storage_offset
|
|
aten::set.source_Storage_storage_offset_out
|
|
aten::set.source_Tensor
|
|
aten::set.source_Tensor_out
|
|
aten::set_
|
|
aten::set_.source_Storage
|
|
aten::set_.source_Storage_storage_offset
|
|
aten::set_.source_Tensor
|
|
aten::slice_copy.Tensor
|
|
aten::slice_copy.Tensor_out
|
|
aten::slice_inverse
|
|
aten::slow_conv3d_forward
|
|
aten::slow_conv3d_forward.output
|
|
aten::slow_conv_dilated2d
|
|
aten::slow_conv_dilated2d.out
|
|
aten::slow_conv_dilated3d
|
|
aten::slow_conv_dilated3d.out
|
|
aten::slow_conv_transpose2d
|
|
aten::slow_conv_transpose2d.out
|
|
aten::slow_conv_transpose3d
|
|
aten::slow_conv_transpose3d.out
|
|
aten::softmax.int_out
|
|
aten::softshrink_backward
|
|
aten::softshrink_backward.grad_input
|
|
aten::sort
|
|
aten::sort.stable
|
|
aten::sort.values
|
|
aten::sort.values_stable
|
|
aten::sparse_compressed_tensor.comp_plain_value
|
|
aten::sparse_compressed_tensor.comp_plain_value_size
|
|
aten::sparse_coo_tensor.size
|
|
aten::sparse_coo_tensor.size_out
|
|
aten::sparse_dim
|
|
aten::sparse_mask
|
|
aten::sparse_mask.out
|
|
aten::sparse_resize
|
|
aten::sparse_resize.out
|
|
aten::sparse_resize_
|
|
aten::sparse_resize_and_clear
|
|
aten::sparse_resize_and_clear.out
|
|
aten::sparse_resize_and_clear_
|
|
aten::sparse_sampled_addmm
|
|
aten::sparse_sampled_addmm.out
|
|
aten::special_airy_ai
|
|
aten::special_airy_ai.out
|
|
aten::special_bessel_y0
|
|
aten::special_bessel_y0.out
|
|
aten::special_bessel_y1
|
|
aten::special_bessel_y1.out
|
|
aten::special_chebyshev_polynomial_t
|
|
aten::special_chebyshev_polynomial_t.n_scalar
|
|
aten::special_chebyshev_polynomial_t.n_scalar_out
|
|
aten::special_chebyshev_polynomial_t.out
|
|
aten::special_chebyshev_polynomial_t.x_scalar
|
|
aten::special_chebyshev_polynomial_t.x_scalar_out
|
|
aten::special_chebyshev_polynomial_u
|
|
aten::special_chebyshev_polynomial_u.n_scalar
|
|
aten::special_chebyshev_polynomial_u.n_scalar_out
|
|
aten::special_chebyshev_polynomial_u.out
|
|
aten::special_chebyshev_polynomial_u.x_scalar
|
|
aten::special_chebyshev_polynomial_u.x_scalar_out
|
|
aten::special_chebyshev_polynomial_v
|
|
aten::special_chebyshev_polynomial_v.n_scalar
|
|
aten::special_chebyshev_polynomial_v.n_scalar_out
|
|
aten::special_chebyshev_polynomial_v.out
|
|
aten::special_chebyshev_polynomial_v.x_scalar
|
|
aten::special_chebyshev_polynomial_v.x_scalar_out
|
|
aten::special_chebyshev_polynomial_w
|
|
aten::special_chebyshev_polynomial_w.n_scalar
|
|
aten::special_chebyshev_polynomial_w.n_scalar_out
|
|
aten::special_chebyshev_polynomial_w.out
|
|
aten::special_chebyshev_polynomial_w.x_scalar
|
|
aten::special_chebyshev_polynomial_w.x_scalar_out
|
|
aten::special_hermite_polynomial_h
|
|
aten::special_hermite_polynomial_h.n_scalar
|
|
aten::special_hermite_polynomial_h.n_scalar_out
|
|
aten::special_hermite_polynomial_h.out
|
|
aten::special_hermite_polynomial_h.x_scalar
|
|
aten::special_hermite_polynomial_h.x_scalar_out
|
|
aten::special_hermite_polynomial_he
|
|
aten::special_hermite_polynomial_he.n_scalar
|
|
aten::special_hermite_polynomial_he.n_scalar_out
|
|
aten::special_hermite_polynomial_he.out
|
|
aten::special_hermite_polynomial_he.x_scalar
|
|
aten::special_hermite_polynomial_he.x_scalar_out
|
|
aten::special_laguerre_polynomial_l
|
|
aten::special_laguerre_polynomial_l.n_scalar
|
|
aten::special_laguerre_polynomial_l.n_scalar_out
|
|
aten::special_laguerre_polynomial_l.out
|
|
aten::special_laguerre_polynomial_l.x_scalar
|
|
aten::special_laguerre_polynomial_l.x_scalar_out
|
|
aten::special_legendre_polynomial_p
|
|
aten::special_legendre_polynomial_p.n_scalar
|
|
aten::special_legendre_polynomial_p.n_scalar_out
|
|
aten::special_legendre_polynomial_p.out
|
|
aten::special_legendre_polynomial_p.x_scalar
|
|
aten::special_legendre_polynomial_p.x_scalar_out
|
|
aten::special_modified_bessel_i0
|
|
aten::special_modified_bessel_i0.out
|
|
aten::special_modified_bessel_i1
|
|
aten::special_modified_bessel_i1.out
|
|
aten::special_modified_bessel_k0
|
|
aten::special_modified_bessel_k0.out
|
|
aten::special_modified_bessel_k1
|
|
aten::special_modified_bessel_k1.out
|
|
aten::special_scaled_modified_bessel_k0
|
|
aten::special_scaled_modified_bessel_k0.out
|
|
aten::special_scaled_modified_bessel_k1
|
|
aten::special_scaled_modified_bessel_k1.out
|
|
aten::special_shifted_chebyshev_polynomial_t
|
|
aten::special_shifted_chebyshev_polynomial_t.n_scalar
|
|
aten::special_shifted_chebyshev_polynomial_t.n_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_t.out
|
|
aten::special_shifted_chebyshev_polynomial_t.x_scalar
|
|
aten::special_shifted_chebyshev_polynomial_t.x_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_u
|
|
aten::special_shifted_chebyshev_polynomial_u.n_scalar
|
|
aten::special_shifted_chebyshev_polynomial_u.n_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_u.out
|
|
aten::special_shifted_chebyshev_polynomial_u.x_scalar
|
|
aten::special_shifted_chebyshev_polynomial_u.x_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_v
|
|
aten::special_shifted_chebyshev_polynomial_v.n_scalar
|
|
aten::special_shifted_chebyshev_polynomial_v.n_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_v.out
|
|
aten::special_shifted_chebyshev_polynomial_v.x_scalar
|
|
aten::special_shifted_chebyshev_polynomial_v.x_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_w
|
|
aten::special_shifted_chebyshev_polynomial_w.n_scalar
|
|
aten::special_shifted_chebyshev_polynomial_w.n_scalar_out
|
|
aten::special_shifted_chebyshev_polynomial_w.out
|
|
aten::special_shifted_chebyshev_polynomial_w.x_scalar
|
|
aten::special_shifted_chebyshev_polynomial_w.x_scalar_out
|
|
aten::split_copy.Tensor
|
|
aten::split_copy.Tensor_out
|
|
aten::squeeze_
|
|
aten::squeeze_.dim
|
|
aten::squeeze_.dims
|
|
aten::sspaddmm.out
|
|
aten::t_
|
|
aten::to_mkldnn
|
|
aten::to_mkldnn.out
|
|
aten::to_padded_tensor
|
|
aten::to_padded_tensor.out
|
|
aten::topk
|
|
aten::topk.values
|
|
aten::transpose_
|
|
aten::triangular_solve
|
|
aten::triangular_solve.X
|
|
aten::unique_consecutive
|
|
aten::unique_consecutive.out
|
|
aten::unique_dim
|
|
aten::unique_dim.out
|
|
aten::unique_dim_consecutive
|
|
aten::unique_dim_consecutive.out
|
|
aten::unsafe_split.Tensor_out
|
|
aten::unsafe_split_with_sizes.out
|
|
aten::unsqueeze_
|
|
aten::upsample_bicubic2d_backward
|
|
aten::upsample_bicubic2d_backward.grad_input
|
|
aten::upsample_bilinear2d.vec_out
|
|
aten::upsample_bilinear2d_backward
|
|
aten::upsample_bilinear2d_backward.grad_input
|
|
aten::upsample_linear1d_backward
|
|
aten::upsample_linear1d_backward.grad_input
|
|
aten::upsample_nearest1d_backward
|
|
aten::upsample_nearest1d_backward.grad_input
|
|
aten::upsample_nearest2d.vec_out
|
|
aten::upsample_nearest2d_backward
|
|
aten::upsample_nearest2d_backward.grad_input
|
|
aten::upsample_nearest3d_backward
|
|
aten::upsample_nearest3d_backward.grad_input
|
|
aten::upsample_trilinear3d_backward
|
|
aten::upsample_trilinear3d_backward.grad_input
|
|
aten::values
|
|
aten::values_copy
|
|
aten::values_copy.out
|
|
aten::view.dtype
|
|
aten::view_as_complex
|
|
aten::view_as_complex_copy
|
|
aten::view_as_complex_copy.out
|
|
aten::view_as_real
|
|
aten::view_as_real_copy
|
|
aten::view_as_real_copy.out
|
|
aten::zeros.names
|
|
aten::zeros.names_out
|
|
aten::zeros.out
|