/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | Embedding.cpp | 40 auto indices_arg = TensorArg(indices, "indices", 1); in embedding_symint() local 79 auto indices_arg = TensorArg(indices_, "indices", 2); in embedding_sparse_backward() local 116 auto indices_arg = TensorArg(indices, "indices", 2); in embedding_dense_backward_cpu() local 184 auto indices_arg = TensorArg(indices, "indices", 2); in embedding_renorm_cpu_() local
|
H A D | EmbeddingBag.cpp | 877 auto indices_arg = TensorArg(indices, "indices", 1); in check_arguments() local 1405 auto indices_arg = TensorArg(indices, "indices", 1); in _embedding_bag_backward_symint() local 1671 auto indices_arg = TensorArg(indices, "indices", 1); in _embedding_bag_per_sample_weights_backward_cpu_template() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | MaxUnpooling.cu | 132 indices_arg{indices_, "indices_", 3}; in max_unpooling2d_forward_out_cuda() local 310 indices_arg{indices_, "indices_", 3}; in max_unpooling3d_forward_out_cuda() local 413 self_arg{self_, "self_", 3}, indices_arg{indices_, "indices_", 4}; in max_unpooling2d_backward_out_cuda() local 521 TensorArg self_arg{self_, "self_", 1}, indices_arg{indices_, "indices_", 2}, in max_unpooling3d_backward_out_cuda() local
|
H A D | ScanKernels.cpp | 30 TensorArg indices_arg{ indices, "indices", 2 }; in cummax_helper_cuda() local 47 TensorArg indices_arg{ indices, "indices", 2 }; in cummin_helper_cuda() local
|
H A D | Embedding.cu | 252 auto indices_arg = TensorArg(indices_, "indices", 1); in embedding_dense_backward_cuda() local 350 auto indices_arg = TensorArg(indices, "indices", 1); in embedding_renorm_cuda_() local
|
H A D | AdaptiveMaxPooling2d.cu | 210 TensorArg indices_arg{indices, "indices", 2}; in TORCH_IMPL_FUNC() local 334 TensorArg indices_arg{indices, "indices", 4}; in TORCH_IMPL_FUNC() local
|
H A D | AdaptiveMaxPooling3d.cu | 311 TensorArg indices_arg{indices, "indices", 2}; in TORCH_IMPL_FUNC() local 393 TensorArg indices_arg{indices, "indices", 4}; in TORCH_IMPL_FUNC() local
|
H A D | TensorTopK.cpp | 52 …TensorArg topK_arg{values, "topK", 1}, indices_arg{indices, "indices", 2}, input_arg{self, "self",… in TORCH_IMPL_FUNC() local
|
H A D | EmbeddingBag.cu | 354 auto indices_arg = TensorArg(indices, "indices", 1); in _embedding_bag_cuda() local 445 auto indices_arg = TensorArg(indices, "indices", 1); in _embedding_bag_dense_backward_cuda() local
|
H A D | DilatedMaxPool3d.cu | 294 TensorArg indices_arg{ indices, "indices", 2 }; in max_pool3d_with_indices_out_cuda_template() local 425 TensorArg indices_arg{ indices, "indices", 4 }; in max_pool3d_with_indices_backward_out_cuda_template() local
|
H A D | DilatedMaxPool2d.cu | 306 TensorArg indices_arg{ indices, "indices", 2 }; in TORCH_IMPL_FUNC() local 432 TensorArg indices_arg{ indices_, "indices", 4 }; in TORCH_IMPL_FUNC() local
|