Searched refs:batch_norm_backward_cpu_template (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 303 std::tuple<Tensor, Tensor, Tensor> batch_norm_backward_cpu_template( in batch_norm_backward_cpu_template() function
934 …return batch_norm_backward_cpu_template<scalar_t, opmath_t>(grad_out, self, weight, running_mean, … in batch_norm_backward_cpu()
936 …return batch_norm_backward_cpu_template<scalar_t, scalar_t>(grad_out, self, weight, running_mean, … in batch_norm_backward_cpu()
|