Searched refs:tensor_assign (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ScatterGatherKernel.cu | 77 static TensorAssign tensor_assign; variable
474 "gather_out_cuda", tensor_assign); in gather_cuda_kernel()
482 "scatter_cuda_", tensor_assign); in scatter_cuda_kernel()
488 "scatter_fill_cuda_", tensor_assign); in scatter_fill_cuda_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ScatterGatherKernel.cpp | 94 static TensorAssign tensor_assign; variable
879 "gather_out_cpu", tensor_assign); in gather_cpu_kernel()
884 self, dim, index, src, "scatter_cpu_", tensor_assign); in scatter_cpu_kernel()
889 self, dim, index, value, "scatter_fill_cpu_", tensor_assign); in scatter_fill_cpu_kernel()
|