Searched refs:batch_norm_cpu_transform_input_template (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 135 std::tuple<Tensor,Tensor,Tensor> batch_norm_cpu_transform_input_template( in batch_norm_cpu_transform_input_template() function
780 …return batch_norm_cpu_transform_input_template<scalar_t, opmath_t>(self, weight, bias, save_mean, … in batch_norm_cpu_out()
786 …return batch_norm_cpu_transform_input_template<scalar_t, opmath_t>(self, weight, bias, std::get<0>… in batch_norm_cpu_out()
790 …return batch_norm_cpu_transform_input_template<scalar_t, scalar_t>(self, weight, bias, save_mean, … in batch_norm_cpu_out()
796 …return batch_norm_cpu_transform_input_template<scalar_t, scalar_t>(self, weight, bias, std::get<0>… in batch_norm_cpu_out()
|