| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ForeachTernaryOp.cu | 39 std::vector<std::vector<at::Tensor>> tensor_lists{ in foreach_tensor_lerp_ternary_cuda() local
71 std::vector<std::vector<at::Tensor>> tensor_lists{ in foreach_tensor_lerp_ternary_cuda_() local
106 std::vector<std::vector<at::Tensor>> tensor_lists{ in foreach_tensor_lerp_list_cuda() local
139 std::vector<std::vector<at::Tensor>> tensor_lists{ in foreach_tensor_lerp_list_cuda_() local
|
| H A D | ForeachPointwiseOp.cu | 30 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_pointwise_op() local
69 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_pointwise_op_() local
100 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_pointwise_op_() local
132 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_pointwise_op() local
|
| H A D | FusedSgdKernel.cu | 167 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_sgd_with_momentum_kernel_cuda_() local
241 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_sgd_with_momentum_kernel_cuda_() local
308 std::vector<std::vector<at::Tensor>> tensor_lists{params.vec(), grads.vec()}; in _fused_sgd_kernel_cuda_() local
400 std::vector<std::vector<at::Tensor>> tensor_lists{params.vec(), grads.vec()}; in _fused_sgd_kernel_cuda_() local
|
| H A D | ForeachBinaryOpList.cu | 32 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_tensor_list_op() local
62 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_tensor_list_op_() local
411 std::vector<std::vector<at::Tensor>> tensor_lists{src.vec(), self.vec()}; in foreach_tensor_copy_list_kernel_cuda_() local
|
| H A D | ForeachUnaryOp.cu | 48 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_unary_op() local
73 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_unary_op_() local
388 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_tensor_zero_cuda_() local
|
| H A D | MultiTensorApply.cuh | 127 std::vector<std::vector<at::Tensor>>& tensor_lists, in multi_tensor_apply()
219 std::vector<std::vector<at::Tensor>>& tensor_lists, in multi_tensor_apply()
299 std::vector<std::vector<at::Tensor>>& tensor_lists, in multi_tensor_apply_for_fused_optimizer()
|
| H A D | ForeachBinaryOpScalarTensor.cu | 38 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_binary_op() local
80 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_binary_op_() local
|
| H A D | fused_adam_impl.cu | 25 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adam_cuda_impl_() local
71 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adam_cuda_impl_() local
|
| H A D | fused_adamw_impl.cu | 26 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adamw_cuda_impl_() local
72 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adamw_cuda_impl_() local
|
| H A D | fused_adam_amsgrad_impl.cu | 26 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adam_amsgrad_cuda_impl_() local
77 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adam_amsgrad_cuda_impl_() local
|
| H A D | fused_adamw_amsgrad_impl.cu | 27 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adamw_amsgrad_cuda_impl_() local
78 std::vector<std::vector<at::Tensor>> tensor_lists{ in _fused_adamw_amsgrad_cuda_impl_() local
|
| H A D | ForeachBinaryOpScalarList.cu | 28 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_binary_op() local
54 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_binary_op_() local
|
| H A D | ForeachBinaryOpScalar.cu | 28 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_binary_op() local
53 std::vector<std::vector<at::Tensor>> tensor_lists; in foreach_binary_op_() local
|
| H A D | AmpKernels.cu | 106 std::vector<std::vector<at::Tensor>> tensor_lists; in _amp_foreach_non_finite_check_and_unscale_cuda_() local
|
| /aosp_15_r20/external/pytorch/test/cpp/c10d/ |
| H A D | ProcessGroupNCCLTest.cpp | 148 std::vector<std::vector<at::Tensor>>& tensor_lists) { in getTensorLists()
|