| /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
|