pt_deps.py

# Generated for selective build without using static dispatch.
# Manually run the script to update (under fbsource):
# xplat/caffe2/fb/update_pt_deps.sh \
# --fbcode --base_ops xplat/caffe2/fb/pt_mobile_base_ops.txt
TORCH_DEPS = {
    "__BASE__": [
        "aten::_coalesced_",
        "aten::_copy_from",
        "aten::_empty_affine_quantized",
        "aten::_empty_per_channel_affine_quantized",
        "aten::_indices",
        "aten::_nnz",
        "aten::_values",
        "aten::add",
        "aten::add_",
        "aten::arange",
        "aten::as_strided",
        "aten::as_strided_",
        "aten::cat",
        "aten::clone",
        "aten::coalesce",
        "aten::contiguous",
        "aten::copy_",
        "aten::copy_sparse_to_sparse_",
        "aten::dense_dim",
        "aten::dequantize",
        "aten::div",
        "aten::div_",
        "aten::empty",
        "aten::empty_like",
        "aten::empty_strided",
        "aten::eq",
        "aten::equal",
        "aten::expand",
        "aten::fill_",
        "aten::is_coalesced",
        "aten::is_complex",
        "aten::is_floating_point",
        "aten::is_leaf",
        "aten::is_nonzero",
        "aten::item",
        "aten::max",
        "aten::min",
        "aten::mul",
        "aten::mul_",
        "aten::narrow",
        "aten::ne",
        "aten::permute",
        "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_tensor",
        "aten::reshape",
        "aten::resize_",
        "aten::resize_as_",
        "aten::scalar_tensor",
        "aten::select",
        "aten::set_",
        "aten::set_quantizer_",
        "aten::size",
        "aten::slice",
        "aten::sparse_dim",
        "aten::sparse_resize_and_clear_",
        "aten::squeeze",
        "aten::squeeze_",
        "aten::stride",
        "aten::sub",
        "aten::sub_",
        "aten::sum",
        "aten::t",
        "aten::to",
        "aten::unsqueeze",
        "aten::view",
        "aten::zero_",
        "aten::zeros",
        "aten::zeros_like",
    ],
    "__ROOT__": [
        "aten::_sparse_coo_tensor_unsafe",
        "aten::_version",
        "aten::any",
        "aten::chunk",
        "aten::detach",
        "aten::isnan",
        "aten::lt",
        "aten::mm",
        "aten::ones_like",
        "aten::output_nr",
        "aten::rsqrt",
        "aten::set_data",
    ],
    "_test::leaky_relu": [
        "aten::leaky_relu",
    ],
    "aten::__and__": [
        "aten::bitwise_and",
    ],
    "aten::__iand__": [
        "aten::bitwise_and_",
    ],
    "aten::__ior__": [
        "aten::bitwise_or_",
    ],
    "aten::__ixor__": [
        "aten::bitwise_xor_",
    ],
    "aten::__or__": [
        "aten::bitwise_or",
    ],
    "aten::__xor__": [
        "aten::bitwise_xor",
    ],
    "aten::_batch_norm_impl_index": [
        "aten::cudnn_batch_norm",
        "aten::miopen_batch_norm",
        "aten::native_batch_norm",
    ],
    "aten::_batch_norm_impl_index_backward": [
        "aten::cudnn_batch_norm_backward",
        "aten::miopen_batch_norm_backward",
        "aten::native_batch_norm_backward",
    ],
    "aten::_cdist_forward": [
        "aten::_euclidean_dist",
    ],
    "aten::_cholesky_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_cholesky_solve_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_convolution": [
        "aten::_convolution_nogroup",
        "aten::_unsafe_view",
        "aten::convolution_overrideable",
        "aten::cudnn_convolution",
        "aten::cudnn_convolution_transpose",
        "aten::miopen_convolution",
        "aten::miopen_convolution_transpose",
        "aten::miopen_depthwise_convolution",
        "aten::slow_conv3d",
        "aten::thnn_conv_depthwise2d",
    ],
    "aten::_convolution_double_backward": [
        "aten::_convolution",
        "aten::transpose",
    ],
    "aten::_convolution_nogroup": [
        "aten::_nnpack_available",
        "aten::_nnpack_spatial_convolution",
        "aten::slow_conv3d",
        "aten::slow_conv_dilated2d",
        "aten::slow_conv_dilated3d",
        "aten::slow_conv_transpose2d",
        "aten::slow_conv_transpose3d",
        "aten::thnn_conv2d",
    ],
    "aten::_ctc_loss_backward": [
        "aten::full_like",
    ],
    "aten::_embedding_bag": [
        "aten::cumsum",
        "aten::expand_as",
        "aten::index_add_",
        "aten::ones_like",
    ],
    "aten::_embedding_bag_backward": [
        "aten::_embedding_bag_dense_backward",
        "aten::_embedding_bag_sparse_backward",
        "aten::cumsum",
        "aten::index_add_",
        "aten::ones_like",
    ],
    "aten::_embedding_bag_dense_backward": [
        "aten::index_add_",
        "aten::index_select",
        "aten::nonzero",
        "aten::sort",
    ],
    "aten::_embedding_bag_forward_only": [
        "aten::cumsum",
        "aten::expand_as",
        "aten::index_add_",
        "aten::ones_like",
    ],
    "aten::_embedding_bag_per_sample_weights_backward": [
        "aten::cumsum",
        "aten::index_add_",
        "aten::ones_like",
    ],
    "aten::_embedding_bag_sparse_backward": [
        "aten::embedding_dense_backward",
        "aten::embedding_sparse_backward",
        "aten::index_select",
    ],
    "aten::_euclidean_dist": [
        "aten::clamp_min_",
        "aten::matmul",
        "aten::ones_like",
        "aten::pow",
        "aten::sqrt_",
        "aten::transpose",
    ],
    "aten::_fake_quantize_learnable_per_channel_affine_backward": [
        "aten::clamp",
        "aten::unbind",
    ],
    "aten::_gather_sparse_backward": [
        "aten::_sparse_coo_tensor_unsafe",
        "aten::repeat",
    ],
    "aten::_index_put_impl_": [
        "aten::nonzero",
    ],
    "aten::_inverse_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_lu_solve_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_lu_with_info": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_nnpack_spatial_convolution_backward": [
        "aten::_nnpack_spatial_convolution_backward_input",
        "aten::_nnpack_spatial_convolution_backward_weight",
    ],
    "aten::_pack_padded_sequence": [
        "aten::transpose",
    ],
    "aten::_pack_padded_sequence_backward": [
        "aten::transpose",
    ],
    "aten::_pad_packed_sequence": [
        "aten::full",
        "aten::transpose",
    ],
    "aten::_qr_helper": [
        "aten::expand_as",
        "aten::eye",
    ],
    "aten::_sobol_engine_initialize_state_": [
        "aten::pow",
    ],
    "aten::_sobol_engine_scramble_": [
        "aten::diagonal",
        "aten::expand_as",
        "aten::pow",
    ],
    "aten::_solve_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_sparse_addmm": [
        "aten::addmm",
    ],
    "aten::_sparse_coo_tensor_unsafe": [
        "aten::_sparse_coo_tensor_with_dims_and_tensors",
    ],
    "aten::_sparse_mm": [
        "aten::_sparse_addmm",
    ],
    "aten::_sparse_sum": [
        "aten::_sparse_coo_tensor_with_dims_and_tensors",
        "aten::values",
    ],
    "aten::_std": [
        "aten::mean",
    ],
    "aten::_svd_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_symeig_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_thnn_differentiable_gru_cell_backward": [
        "aten::sigmoid",
        "aten::sigmoid_backward",
        "aten::tanh",
        "aten::tanh_backward",
        "aten::unsafe_chunk",
    ],
    "aten::_thnn_differentiable_lstm_cell_backward": [
        "aten::sigmoid",
        "aten::sigmoid_backward",
        "aten::tanh",
        "aten::tanh_backward",
        "aten::unsafe_chunk",
    ],
    "aten::_triangular_solve_helper": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::_trilinear": [
        "aten::bmm",
    ],
    "aten::_var": [
        "aten::mean",
    ],
    "aten::_weight_norm": [
        "aten::_weight_norm_cuda_interface",
        "aten::norm_except_dim",
    ],
    "aten::abs": [
        "aten::real",
    ],
    "aten::abs_": [
        "aten::abs",
    ],
    "aten::absolute": [
        "aten::abs",
        "aten::real",
    ],
    "aten::absolute_": [
        "aten::abs",
    ],
    "aten::acos_": [
        "aten::acos",
    ],
    "aten::acosh_": [
        "aten::acosh",
    ],
    "aten::adaptive_avg_pool1d": [
        "aten::adaptive_avg_pool2d",
    ],
    "aten::adaptive_avg_pool2d": [
        "aten::_adaptive_avg_pool2d",
        "aten::mean",
        "aten::mkldnn_adaptive_avg_pool2d",
    ],
    "aten::adaptive_max_pool1d": [
        "aten::adaptive_max_pool2d",
    ],
    "aten::add": [
        "aten::empty_meta",
    ],
    "aten::addbmm": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::addbmm_": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::addcdiv_": [
        "aten::addcdiv",
    ],
    "aten::addcmul_": [
        "aten::addcmul",
    ],
    "aten::addmm": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::addmm_": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::addmv": [
        "aten::_addmv_impl_",
    ],
    "aten::addmv_": [
        "aten::_addmv_impl_",
    ],
    "aten::addr": [
        "aten::_addr",
    ],
    "aten::addr_": [
        "aten::_addr_",
    ],
    "aten::affine_grid_generator": [
        "aten::bmm",
        "aten::linspace",
        "aten::transpose",
        "aten::unsqueeze_",
    ],
    "aten::affine_grid_generator_backward": [
        "aten::bmm",
        "aten::linspace",
        "aten::transpose",
        "aten::unsqueeze_",
    ],
    "aten::align_tensors": [
        "aten::rename",
    ],
    "aten::allclose": [
        "aten::all",
        "aten::isclose",
    ],
    "aten::alpha_dropout": [
        "aten::bernoulli_",
    ],
    "aten::alpha_dropout_": [
        "aten::bernoulli_",
    ],
    "aten::angle": [
        "aten::real",
    ],
    "aten::argsort": [
        "aten::sort",
    ],
    "aten::asin_": [
        "aten::asin",
    ],
    "aten::asinh_": [
        "aten::asinh",
    ],
    "aten::atan_": [
        "aten::atan",
    ],
    "aten::atanh_": [
        "aten::atanh",
    ],
    "aten::avg_pool1d": [
        "aten::avg_pool2d",
    ],
    "aten::baddbmm": [
        "aten::addmm_",
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::baddbmm_": [
        "aten::addmm_",
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::batch_norm": [
        "aten::_batch_norm_impl_index",
    ],
    "aten::bernoulli": [
        "aten::bernoulli_",
    ],
    "aten::bilinear": [
        "aten::_trilinear",
    ],
    "aten::binary_cross_entropy": [
        "aten::mean",
    ],
    "aten::binary_cross_entropy_with_logits": [
        "aten::clamp_min_",
        "aten::exp_",
        "aten::log_",
        "aten::mean",
        "aten::neg",
    ],
    "aten::binary_cross_entropy_with_logits_backward": [
        "aten::sigmoid",
    ],
    "aten::bitwise_and_": [
        "aten::bitwise_and",
    ],
    "aten::bitwise_not_": [
        "aten::bitwise_not",
    ],
    "aten::bitwise_or_": [
        "aten::bitwise_or",
    ],
    "aten::bitwise_xor_": [
        "aten::bitwise_xor",
    ],
    "aten::blackman_window": [
        "aten::cos_",
    ],
    "aten::bmm": [
        "aten::addmm_",
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::cartesian_prod": [
        "aten::flatten",
        "aten::meshgrid",
        "aten::stack",
    ],
    "aten::cat": [
        "aten::_cat",
        "aten::_sparse_coo_tensor_with_dims_and_tensors",
    ],
    "aten::cdist": [
        "aten::_cdist_forward",
        "aten::_euclidean_dist",
    ],
    "aten::ceil_": [
        "aten::ceil",
    ],
    "aten::celu": [
        "aten::elu",
    ],
    "aten::celu_": [
        "aten::elu_",
    ],
    "aten::chain_matmul": [
        "aten::mm",
    ],
    "aten::cholesky": [
        "aten::_cholesky_helper",
        "aten::tril_",
        "aten::triu_",
    ],
    "aten::cholesky_solve": [
        "aten::_cholesky_solve_helper",
    ],
    "aten::chunk": [
        "aten::split",
        "aten::split_with_sizes",
    ],
    "aten::clamp": [
        "aten::clamp_max",
        "aten::clamp_min",
    ],
    "aten::clamp_": [
        "aten::clamp",
    ],
    "aten::clamp_max_": [
        "aten::clamp_max",
    ],
    "aten::clamp_min_": [
        "aten::clamp_min",
    ],
    "aten::combinations": [
        "aten::full",
        "aten::le",
        "aten::lt",
        "aten::masked_select",
        "aten::meshgrid",
        "aten::stack",
    ],
    "aten::conv1d": [
        "aten::convolution",
    ],
    "aten::conv2d": [
        "aten::convolution",
    ],
    "aten::conv3d": [
        "aten::convolution",
    ],
    "aten::conv_tbc": [
        "aten::addmm_",
    ],
    "aten::conv_tbc_backward": [
        "aten::addmm_",
    ],
    "aten::conv_transpose1d": [
        "aten::convolution",
    ],
    "aten::conv_transpose2d": [
        "aten::convolution",
    ],
    "aten::conv_transpose3d": [
        "aten::convolution",
    ],
    "aten::convolution": [
        "aten::_convolution",
    ],
    "aten::cos_": [
        "aten::cos",
    ],
    "aten::cosh_": [
        "aten::cosh",
    ],
    "aten::cosine_embedding_loss": [
        "aten::clamp_min_",
        "aten::mean",
        "aten::sqrt_",
        "aten::where",
    ],
    "aten::cosine_similarity": [
        "aten::clamp_min_",
        "aten::sqrt_",
    ],
    "aten::ctc_loss": [
        "aten::_ctc_loss",
        "aten::_cudnn_ctc_loss",
        "aten::_use_cudnn_ctc_loss",
        "aten::clamp_min",
        "aten::mean",
        "aten::where",
    ],
    "aten::cummax": [
        "aten::_cummax_helper",
    ],
    "aten::cummin": [
        "aten::_cummin_helper",
    ],
    "aten::cumprod": [
        "aten::_cumprod",
    ],
    "aten::cumsum": [
        "aten::_cumsum",
    ],
    "aten::deg2rad_": [
        "aten::deg2rad",
    ],
    "aten::det": [
        "aten::_lu_with_info",
        "aten::all",
        "aten::diagonal",
        "aten::fmod_",
        "aten::ge",
        "aten::prod",
    ],
    "aten::diag_embed": [
        "aten::diagonal",
    ],
    "aten::diagflat": [
        "aten::diag",
    ],
    "aten::diagonal": [
        "aten::refine_names",
    ],
    "aten::dist": [
        "aten::norm",
    ],
    "aten::dropout": [
        "aten::_fused_dropout",
        "aten::bernoulli_",
    ],
    "aten::dropout_": [
        "aten::bernoulli_",
    ],
    "aten::einsum": [
        "aten::bmm",
        "aten::diagonal",
    ],
    "aten::elu_": [
        "aten::elu",
    ],
    "aten::embedding": [
        "aten::index_select",
    ],
    "aten::embedding_backward": [
        "aten::embedding_dense_backward",
        "aten::embedding_sparse_backward",
    ],
    "aten::embedding_bag": [
        "aten::_embedding_bag",
        "aten::_embedding_bag_forward_only",
    ],
    "aten::embedding_renorm_": [
        "aten::norm",
    ],
    "aten::embedding_sparse_backward": [
        "aten::_sparse_coo_tensor_unsafe",
        "aten::index",
    ],
    "aten::equal": [
        "aten::is_same_size",
    ],
    "aten::erf_": [
        "aten::erf",
    ],
    "aten::erfc_": [
        "aten::erfc",
    ],
    "aten::erfinv_": [
        "aten::erfinv",
    ],
    "aten::exp_": [
        "aten::exp",
    ],
    "aten::expm1_": [
        "aten::expm1",
    ],
    "aten::feature_alpha_dropout": [
        "aten::bernoulli_",
    ],
    "aten::feature_alpha_dropout_": [
        "aten::bernoulli_",
    ],
    "aten::feature_dropout": [
        "aten::bernoulli_",
    ],
    "aten::feature_dropout_": [
        "aten::bernoulli_",
    ],
    "aten::fft": [
        "aten::_fft_with_size",
    ],
    "aten::fliplr": [
        "aten::flip",
    ],
    "aten::flipud": [
        "aten::flip",
    ],
    "aten::floor_": [
        "aten::floor",
    ],
    "aten::floor_divide": [
        "aten::trunc_",
    ],
    "aten::floor_divide_": [
        "aten::floor_divide",
        "aten::trunc_",
    ],
    "aten::fmod_": [
        "aten::fmod",
    ],
    "aten::frac_": [
        "aten::frac",
    ],
    "aten::frobenius_norm": [
        "aten::conj",
        "aten::norm",
        "aten::real",
        "aten::sqrt",
    ],
    "aten::gcd_": [
        "aten::gcd",
    ],
    "aten::ge_": [
        "aten::ge",
    ],
    "aten::ger": [
        "aten::_addr",
    ],
    "aten::glu_backward": [
        "aten::sigmoid",
    ],
    "aten::grid_sampler": [
        "aten::cudnn_grid_sampler",
        "aten::grid_sampler_2d",
        "aten::grid_sampler_3d",
    ],
    "aten::group_norm": [
        "aten::native_group_norm",
    ],
    "aten::gru": [
        "aten::_thnn_fused_gru_cell",
        "aten::cudnn_is_acceptable",
        "aten::dropout",
        "aten::linear",
        "aten::matmul",
        "aten::sigmoid_",
        "aten::stack",
        "aten::tanh_",
        "aten::transpose",
        "aten::transpose_",
        "aten::unbind",
        "aten::unsafe_chunk",
    ],
    "aten::gru_cell": [
        "aten::_thnn_fused_gru_cell",
        "aten::linear",
        "aten::matmul",
        "aten::sigmoid_",
        "aten::tanh_",
        "aten::unsafe_chunk",
    ],
    "aten::gt_": [
        "aten::gt",
    ],
    "aten::hamming_window": [
        "aten::cos_",
    ],
    "aten::hann_window": [
        "aten::cos_",
    ],
    "aten::hardtanh": [
        "aten::clamp",
    ],
    "aten::hardtanh_": [
        "aten::clamp_",
    ],
    "aten::hinge_embedding_loss": [
        "aten::clamp_min_",
        "aten::mean",
        "aten::where",
    ],
    "aten::ifft": [
        "aten::_fft_with_size",
    ],
    "aten::imag": [
        "aten::view_as_real",
    ],
    "aten::index": [
        "aten::nonzero",
    ],
    "aten::index_add": [
        "aten::index_add_",
    ],
    "aten::index_copy": [
        "aten::index_copy_",
    ],
    "aten::index_copy_": [
        "aten::_index_copy_",
    ],
    "aten::index_fill": [
        "aten::index_fill_",
    ],
    "aten::index_put": [
        "aten::index_put_",
    ],
    "aten::index_put_": [
        "aten::_index_put_impl_",
    ],
    "aten::instance_norm": [
        "aten::alias",
        "aten::batch_norm",
        "aten::mean",
        "aten::repeat",
    ],
    "aten::inverse": [
        "aten::_inverse_helper",
    ],
    "aten::irfft": [
        "aten::_fft_with_size",
    ],
    "aten::isclose": [
        "aten::__iand__",
        "aten::__ior__",
        "aten::abs",
        "aten::isfinite",
        "aten::le",
    ],
    "aten::isfinite": [
        "aten::abs",
        "aten::ones_like",
    ],
    "aten::isinf": [
        "aten::__ior__",
        "aten::abs",
        "aten::imag",
        "aten::real",
    ],
    "aten::isreal": [
        "aten::imag",
        "aten::ones_like",
    ],
    "aten::istft": [
        "aten::_fft_with_size",
        "aten::abs",
        "aten::constant_pad_nd",
        "aten::conv_transpose1d",
        "aten::eye",
        "aten::ones",
        "aten::pow",
        "aten::repeat",
        "aten::transpose",
    ],
    "aten::item": [
        "aten::_local_scalar_dense",
    ],
    "aten::kl_div": [
        "aten::exp",
        "aten::gt",
        "aten::log",
        "aten::mean",
        "aten::where",
    ],
    "aten::kl_div_backward": [
        "aten::exp",
        "aten::expand_as",
        "aten::neg",
    ],
    "aten::kthvalue": [
        "aten::unsqueeze_",
    ],
    "aten::l1_loss": [
        "aten::abs_",
        "aten::mean",
    ],
    "aten::l1_loss_backward": [
        "aten::sign_",
    ],
    "aten::layer_norm": [
        "aten::native_layer_norm",
    ],
    "aten::lcm_": [
        "aten::lcm",
    ],
    "aten::le_": [
        "aten::le",
    ],
    "aten::leaky_relu_": [
        "aten::leaky_relu",
    ],
    "aten::lgamma_": [
        "aten::lgamma",
    ],
    "aten::linear": [
        "aten::addmm",
        "aten::matmul",
        "aten::mkldnn_linear",
    ],
    "aten::log10_": [
        "aten::log10",
    ],
    "aten::log1p_": [
        "aten::log1p",
    ],
    "aten::log2_": [
        "aten::log2",
    ],
    "aten::log_": [
        "aten::log",
    ],
    "aten::log_sigmoid": [
        "aten::log_sigmoid_forward",
    ],
    "aten::log_softmax": [
        "aten::_log_softmax",
    ],
    "aten::logcumsumexp": [
        "aten::_logcumsumexp",
    ],
    "aten::logdet": [
        "aten::_lu_with_info",
        "aten::abs_",
        "aten::all",
        "aten::diagonal",
        "aten::fmod_",
        "aten::full",
        "aten::ge",
        "aten::index_put_",
        "aten::log_",
        "aten::lt",
        "aten::nonzero_numpy",
        "aten::prod",
        "aten::sign",
    ],
    "aten::logical_and_": [
        "aten::logical_and",
    ],
    "aten::logical_not_": [
        "aten::logical_not",
    ],
    "aten::logical_or_": [
        "aten::logical_or",
    ],
    "aten::logical_xor_": [
        "aten::logical_xor",
    ],
    "aten::logit_": [
        "aten::logit",
    ],
    "aten::logsumexp": [
        "aten::abs",
        "aten::exp",
        "aten::log_",
        "aten::masked_fill_",
        "aten::max_values",
    ],
    "aten::lstm": [
        "aten::_thnn_fused_lstm_cell",
        "aten::cudnn_is_acceptable",
        "aten::dropout",
        "aten::linear",
        "aten::matmul",
        "aten::sigmoid_",
        "aten::stack",
        "aten::tanh",
        "aten::tanh_",
        "aten::transpose",
        "aten::unbind",
        "aten::unsafe_chunk",
    ],
    "aten::lstm_cell": [
        "aten::_thnn_fused_lstm_cell",
        "aten::linear",
        "aten::matmul",
        "aten::sigmoid_",
        "aten::tanh",
        "aten::tanh_",
        "aten::unsafe_chunk",
    ],
    "aten::lt_": [
        "aten::lt",
    ],
    "aten::lu_solve": [
        "aten::_lu_solve_helper",
    ],
    "aten::margin_ranking_loss": [
        "aten::clamp_min_",
        "aten::mean",
        "aten::neg",
    ],
    "aten::masked_fill": [
        "aten::masked_fill_",
    ],
    "aten::masked_scatter": [
        "aten::masked_scatter_",
    ],
    "aten::matmul": [
        "aten::_unsafe_view",
        "aten::bmm",
        "aten::dot",
        "aten::mm",
        "aten::mv",
        "aten::transpose",
    ],
    "aten::matrix_power": [
        "aten::_unsafe_view",
        "aten::bmm",
        "aten::dot",
        "aten::expand_as",
        "aten::eye",
        "aten::inverse",
        "aten::mm",
        "aten::mv",
        "aten::transpose",
    ],
    "aten::matrix_rank": [
        "aten::abs",
        "aten::gt",
        "aten::svd",
        "aten::symeig",
    ],
    "aten::max": [
        "aten::_make_per_tensor_quantized_tensor",
        "aten::int_repr",
        "aten::unsqueeze_",
    ],
    "aten::max_pool1d": [
        "aten::max_pool1d_with_indices",
    ],
    "aten::max_pool1d_with_indices": [
        "aten::max_pool2d_with_indices",
    ],
    "aten::max_pool2d": [
        "aten::max_pool2d_with_indices",
        "aten::mkldnn_max_pool2d",
        "aten::quantized_max_pool2d",
    ],
    "aten::max_pool3d": [
        "aten::max_pool3d_with_indices",
        "aten::mkldnn_max_pool3d",
    ],
    "aten::median": [
        "aten::kthvalue",
    ],
    "aten::min": [
        "aten::_make_per_tensor_quantized_tensor",
        "aten::int_repr",
        "aten::unsqueeze_",
    ],
    "aten::mm": [
        "aten::transpose",
        "aten::transpose_",
    ],
    "aten::mode": [
        "aten::_mode",
    ],
    "aten::mse_loss": [
        "aten::mean",
    ],
    "aten::multilabel_margin_loss": [
        "aten::multilabel_margin_loss_forward",
    ],
    "aten::multinomial": [
        "aten::bitwise_and",
        "aten::ge",
        "aten::log_",
        "aten::lt",
        "aten::topk",
        "aten::uniform_",
    ],
    "aten::mv": [
        "aten::_addmv_impl_",
    ],
    "aten::mvlgamma": [
        "aten::all",
        "aten::gt",
        "aten::lgamma_",
    ],
    "aten::mvlgamma_": [
        "aten::all",
        "aten::gt",
        "aten::lgamma_",
    ],
    "aten::neg_": [
        "aten::neg",
    ],
    "aten::new_full": [
        "aten::full",
    ],
    "aten::nll_loss": [
        "aten::nll_loss_forward",
    ],
    "aten::nll_loss2d": [
        "aten::nll_loss2d_forward",
    ],
    "aten::nonzero_numpy": [
        "aten::nonzero",
        "aten::unbind",
    ],
    "aten::norm": [
        "aten::native_norm",
    ],
    "aten::norm_except_dim": [
        "aten::norm",
        "aten::transpose",
    ],
    "aten::normal": [
        "aten::full",
        "aten::normal_",
        "aten::view_as_real",
    ],
    "aten::normal_": [
        "aten::view_as_real",
    ],
    "aten::nuclear_norm": [
        "aten::svd",
        "aten::unsqueeze_",
    ],
    "aten::one_hot": [
        "aten::scatter_",
    ],
    "aten::pairwise_distance": [
        "aten::norm",
    ],
    "aten::pdist": [
        "aten::_pdist_forward",
    ],
    "aten::pin_memory": [
        "aten::is_pinned",
    ],
    "aten::pinverse": [
        "aten::diag_embed",
        "aten::gt",
        "aten::matmul",
        "aten::reciprocal",
        "aten::svd",
        "aten::transpose",
        "aten::where",
    ],
    "aten::poisson_nll_loss": [
        "aten::exp",
        "aten::le",
        "aten::log",
        "aten::masked_fill",
        "aten::mean",
    ],
    "aten::polygamma_": [
        "aten::polygamma",
    ],
    "aten::pow": [
        "aten::can_cast",
        "aten::result_type",
    ],
    "aten::pow_": [
        "aten::can_cast",
        "aten::result_type",
    ],
    "aten::qr": [
        "aten::_qr_helper",
    ],
    "aten::quantized_gru": [
        "aten::_thnn_fused_gru_cell",
        "aten::dropout",
        "aten::fbgemm_linear_int8_weight_fp32_activation",
        "aten::sigmoid_",
        "aten::stack",
        "aten::tanh_",
        "aten::transpose",
        "aten::transpose_",
        "aten::unbind",
        "aten::unsafe_chunk",
    ],
    "aten::quantized_gru_cell": [
        "aten::_thnn_fused_gru_cell",
        "aten::fbgemm_linear_int8_weight_fp32_activation",
        "aten::sigmoid_",
        "aten::tanh_",
        "aten::unsafe_chunk",
    ],
    "aten::quantized_lstm": [
        "aten::_thnn_fused_lstm_cell",
        "aten::dropout",
        "aten::fbgemm_linear_int8_weight_fp32_activation",
        "aten::sigmoid_",
        "aten::stack",
        "aten::tanh",
        "aten::tanh_",
        "aten::transpose",
        "aten::unbind",
        "aten::unsafe_chunk",
    ],
    "aten::quantized_lstm_cell": [
        "aten::_thnn_fused_lstm_cell",
        "aten::fbgemm_linear_int8_weight_fp32_activation",
        "aten::sigmoid_",
        "aten::tanh",
        "aten::tanh_",
        "aten::unsafe_chunk",
    ],
    "aten::quantized_rnn_relu_cell": [
        "aten::fbgemm_linear_int8_weight_fp32_activation",
        "aten::relu",
    ],
    "aten::quantized_rnn_tanh_cell": [
        "aten::fbgemm_linear_int8_weight_fp32_activation",
        "aten::tanh",
    ],
    "aten::rad2deg_": [
        "aten::rad2deg",
    ],
    "aten::rand": [
        "aten::uniform_",
    ],
    "aten::rand_like": [
        "aten::uniform_",
    ],
    "aten::randint": [
        "aten::random_",
    ],
    "aten::randint_like": [
        "aten::random_",
    ],
    "aten::randn": [
        "aten::normal_",
    ],
    "aten::randn_like": [
        "aten::normal_",
    ],
    "aten::real": [
        "aten::view_as_real",
    ],
    "aten::reciprocal_": [
        "aten::reciprocal",
    ],
    "aten::refine_names": [
        "aten::alias",
    ],
    "aten::relu": [
        "aten::threshold",
    ],
    "aten::relu_": [
        "aten::threshold_",
    ],
    "aten::rename": [
        "aten::alias",
    ],
    "aten::repeat": [
        "aten::alias",
        "aten::empty_quantized",
        "aten::expand_as",
        "aten::unfold",
    ],
    "aten::repeat_interleave": [
        "aten::all",
        "aten::cumsum",
        "aten::flatten",
        "aten::ge",
        "aten::index_select",
    ],
    "aten::reshape": [
        "aten::_mkldnn_reshape",
        "aten::_unsafe_view",
    ],
    "aten::rfft": [
        "aten::_fft_with_size",
    ],
    "aten::rnn_relu": [
        "aten::cudnn_is_acceptable",
        "aten::dropout",
        "aten::linear",
        "aten::matmul",
        "aten::relu",
        "aten::stack",
        "aten::transpose",
        "aten::transpose_",
        "aten::unbind",
    ],
    "aten::rnn_relu_cell": [
        "aten::linear",
        "aten::matmul",
        "aten::relu",
    ],
    "aten::rnn_tanh": [
        "aten::cudnn_is_acceptable",
        "aten::dropout",
        "aten::linear",
        "aten::matmul",
        "aten::stack",
        "aten::tanh",
        "aten::transpose",
        "aten::transpose_",
        "aten::unbind",
    ],
    "aten::rnn_tanh_cell": [
        "aten::linear",
        "aten::matmul",
        "aten::tanh",
    ],
    "aten::rot90": [
        "aten::flip",
        "aten::transpose_",
    ],
    "aten::round_": [
        "aten::round",
    ],
    "aten::rrelu": [
        "aten::rrelu_with_noise",
    ],
    "aten::rrelu_": [
        "aten::rrelu_with_noise_",
    ],
    "aten::rrelu_with_noise": [
        "aten::leaky_relu",
    ],
    "aten::rrelu_with_noise_": [
        "aten::leaky_relu",
    ],
    "aten::rrelu_with_noise_backward": [
        "aten::leaky_relu_backward",
    ],
    "aten::rsqrt_": [
        "aten::rsqrt",
    ],
    "aten::scatter": [
        "aten::scatter_",
    ],
    "aten::scatter_add": [
        "aten::scatter_add_",
    ],
    "aten::select": [
        "aten::_sparse_coo_tensor_with_dims_and_tensors",
        "aten::index_select",
        "aten::nonzero",
    ],
    "aten::selu": [
        "aten::elu",
    ],
    "aten::selu_": [
        "aten::elu_",
    ],
    "aten::sigmoid_": [
        "aten::sigmoid",
    ],
    "aten::sign_": [
        "aten::sign",
    ],
    "aten::silu": [
        "aten::sigmoid",
    ],
    "aten::silu_": [
        "aten::sigmoid",
    ],
    "aten::silu_backward": [
        "aten::sigmoid",
    ],
    "aten::sin_": [
        "aten::sin",
    ],
    "aten::sinh_": [
        "aten::sinh",
    ],
    "aten::slogdet": [
        "aten::_lu_with_info",
        "aten::abs_",
        "aten::all",
        "aten::diagonal",
        "aten::fmod_",
        "aten::ge",
        "aten::log_",
        "aten::prod",
        "aten::sign",
    ],
    "aten::slow_conv3d": [
        "aten::slow_conv3d_forward",
    ],
    "aten::slow_conv3d_backward": [
        "aten::addmm_",
        "aten::baddbmm_",
        "aten::bmm",
        "aten::mm",
        "aten::transpose",
    ],
    "aten::slow_conv3d_forward": [
        "aten::addmm_",
        "aten::baddbmm_",
        "aten::bmm",
        "aten::mm",
    ],
    "aten::smm": [
        "aten::sspaddmm",
    ],
    "aten::smooth_l1_loss": [
        "aten::mean",
    ],
    "aten::soft_margin_loss": [
        "aten::exp_",
        "aten::log_",
        "aten::mean",
        "aten::neg",
    ],
    "aten::soft_margin_loss_backward": [
        "aten::exp",
        "aten::neg",
    ],
    "aten::softmax": [
        "aten::_softmax",
    ],
    "aten::solve": [
        "aten::_solve_helper",
    ],
    "aten::sort": [
        "aten::_make_per_tensor_quantized_tensor",
        "aten::int_repr",
    ],
    "aten::sparse_coo_tensor": [
        "aten::_sparse_coo_tensor_with_dims",
        "aten::_sparse_coo_tensor_with_dims_and_tensors",
    ],
    "aten::sqrt_": [
        "aten::sqrt",
    ],
    "aten::square": [
        "aten::pow",
    ],
    "aten::square_": [
        "aten::pow",
    ],
    "aten::std": [
        "aten::_std",
        "aten::imag",
        "aten::real",
        "aten::sqrt",
    ],
    "aten::std_mean": [
        "aten::imag",
        "aten::real",
        "aten::sqrt",
    ],
    "aten::stft": [
        "aten::rfft",
        "aten::transpose_",
    ],
    "aten::svd": [
        "aten::_svd_helper",
    ],
    "aten::symeig": [
        "aten::_symeig_helper",
    ],
    "aten::t": [
        "aten::transpose",
    ],
    "aten::t_": [
        "aten::transpose_",
    ],
    "aten::tan_": [
        "aten::tan",
    ],
    "aten::tanh_": [
        "aten::tanh",
    ],
    "aten::tensordot": [
        "aten::mm",
    ],
    "aten::thnn_conv2d": [
        "aten::thnn_conv2d_forward",
    ],
    "aten::thnn_conv2d_backward": [
        "aten::addmm_",
        "aten::transpose",
    ],
    "aten::thnn_conv2d_forward": [
        "aten::addmm_",
        "aten::mm",
    ],
    "aten::thnn_conv_depthwise2d": [
        "aten::thnn_conv_depthwise2d_forward",
    ],
    "aten::to_dense_backward": [
        "aten::sparse_mask",
        "aten::to_mkldnn",
    ],
    "aten::to_mkldnn_backward": [
        "aten::to_dense",
    ],
    "aten::to_sparse": [
        "aten::chunk",
        "aten::index",
        "aten::nonzero",
        "aten::sparse_coo_tensor",
        "aten::transpose",
        "aten::unique_dim",
    ],
    "aten::transpose": [
        "aten::_mkldnn_transpose",
    ],
    "aten::transpose_": [
        "aten::_mkldnn_transpose_",
    ],
    "aten::triangular_solve": [
        "aten::_triangular_solve_helper",
    ],
    "aten::triplet_margin_loss": [
        "aten::clamp_min",
        "aten::mean",
        "aten::pairwise_distance",
    ],
    "aten::trunc_": [
        "aten::trunc",
    ],
    "aten::uniform_": [
        "aten::view_as_real",
    ],
    "aten::unique_consecutive": [
        "aten::stack",
        "aten::transpose",
        "aten::unbind",
    ],
    "aten::unique_dim": [
        "aten::stack",
        "aten::transpose",
        "aten::unbind",
    ],
    "aten::unique_dim_consecutive": [
        "aten::stack",
        "aten::transpose",
        "aten::unbind",
    ],
    "aten::unsafe_chunk": [
        "aten::unsafe_split",
        "aten::unsafe_split_with_sizes",
    ],
    "aten::unsqueeze": [
        "aten::_cat",
        "aten::_sparse_coo_tensor_with_dims_and_tensors",
    ],
    "aten::vander": [
        "aten::cumprod",
        "aten::flip",
        "aten::promote_types",
    ],
    "aten::var": [
        "aten::_var",
        "aten::imag",
        "aten::real",
        "aten::sqrt",
        "aten::std",
    ],
    "aten::var_mean": [
        "aten::imag",
        "aten::real",
        "aten::sqrt",
    ],
    "aten::where": [
        "aten::_s_where",
        "aten::nonzero_numpy",
    ],
    "quantized::make_quantized_cell_params": [
        "aten::fbgemm_linear_int8_weight_fp32_activation",
    ],
    "quantized::max_pool2d": [
        "aten::max_pool2d",
    ],
    "quantized::quantized_gru_cell_dynamic": [
        "aten::_thnn_fused_gru_cell",
        "aten::sigmoid_",
        "aten::tanh_",
        "aten::unsafe_chunk",
    ],
    "quantized::quantized_lstm_cell_dynamic": [
        "aten::_thnn_fused_lstm_cell",
        "aten::sigmoid_",
        "aten::tanh",
        "aten::tanh_",
        "aten::unsafe_chunk",
    ],
    "quantized::quantized_rnn_relu_cell_dynamic": [
        "aten::relu",
    ],
    "quantized::quantized_rnn_tanh_cell_dynamic": [
        "aten::tanh",
    ],
}