Searched refs:batch_norm_backward_reduce_cuda_template (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cu | 748 …return batch_norm_backward_reduce_cuda_template<scalar_t, accscalar_t, int32_t>(grad_output, input… in batch_norm_backward_reduce_cuda()
750 …return batch_norm_backward_reduce_cuda_template<scalar_t, scalar_t, int32_t>(grad_output, input, m… in batch_norm_backward_reduce_cuda()
754 …return batch_norm_backward_reduce_cuda_template<scalar_t, accscalar_t, int64_t>(grad_output, input… in batch_norm_backward_reduce_cuda()
756 …return batch_norm_backward_reduce_cuda_template<scalar_t, scalar_t, int64_t>(grad_output, input, m… in batch_norm_backward_reduce_cuda()
|
| H A D | Normalization.cuh | 778 std::tuple<Tensor, Tensor, Tensor, Tensor> batch_norm_backward_reduce_cuda_template(const Tensor& g… in batch_norm_backward_reduce_cuda_template() function
|