| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cu | 306 void batch_norm_mean_var(const Tensor& self, Tensor& save_mean, Tensor& save_var) { in batch_norm_mean_var()
347 const Tensor& save_mean, const Tensor& save_var, in batch_norm_update_stats()
381 const Tensor& save_mean, const Tensor& save_var, in batch_norm_update_stats_and_invert()
435 …r_opt, bool train, double momentum, double epsilon, Tensor& output, Tensor& save_mean, Tensor& sav… in batch_norm_cuda_out()
466 auto save_mean = at::empty({n_input}, options); in batch_norm_cuda() local
491 Tensor output, save_mean, save_var, reserve; in _batch_norm_with_update_cuda() local
512 Tensor& out, Tensor& save_mean, Tensor& save_var, Tensor& reserve) { in _batch_norm_with_update_cuda_out()
540 …g_var, bool train, double momentum, double epsilon, Tensor& output, Tensor& save_mean, Tensor& sav… in _batch_norm_legit_cuda_out()
544 …s_opt, bool train, double momentum, double epsilon, Tensor& output, Tensor& save_mean, Tensor& sav… in _batch_norm_legit_no_stats_cuda_out()
556 const Tensor& save_mean = c10::value_or_else(save_mean_opt, [] {return Tensor();}); in _new_batch_norm_backward_cuda() local
[all …]
|
| H A D | Normalization.cuh | 273 GenericPackedTensorAccessor<stat_accscalar_t, 1, RestrictPtrTraits, index_t> save_mean, in batch_norm_collect_statistics_kernel()
365 … const GenericPackedTensorAccessor<const stat_accscalar_t, 1, DefaultPtrTraits, index_t> save_mean, in batch_norm_backward_kernel()
642 auto save_mean = packed_accessor_or_dummy< in batch_norm_backward_cuda_template() local
762 auto save_mean = get_packed_accessor< in batch_norm_gather_stats_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 137 const Tensor& save_mean /* optional */, const Tensor& save_invstd /* optional */, in batch_norm_cpu_transform_input_template()
202 double momentum, double eps, Tensor& save_mean, Tensor& save_var_transform) { in batch_norm_cpu_update_stats_template()
297 …Tensor save_mean = is_contiguous(input) ? at::empty({n_input}, input.options().dtype(dtype)) : at:… in batch_norm_cpu_update_stats_template() local
305 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_backward_cpu_template()
556 auto save_mean = at::empty_symint(c10::SymIntArrayRef({num_features}), options); in _batch_norm_impl_index() local
630 const Tensor& save_mean = c10::value_or_else(save_mean_opt, [] {return Tensor();}); in _batch_norm_impl_index_backward() local
762 … bool train, double momentum, double eps, Tensor& out, Tensor& save_mean, Tensor& save_var) { in batch_norm_cpu_out()
825 Tensor save_mean; in batch_norm_cpu() local
865 Tensor& out, Tensor& save_mean, Tensor& save_var, Tensor& reserve) { in _batch_norm_with_update_cpu_out()
902 … running_var, bool train, double momentum, double eps, Tensor& out, Tensor& save_mean, Tensor& sav… in _batch_norm_legit_cpu_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | batch_norm_kernel.cpp | 34 const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_collect_linear_and_constant_terms()
76 const Tensor& weight, const Tensor& bias, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_contiguous_impl()
128 const Tensor& weight, const Tensor& bias, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_channels_last_impl()
407 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_cpu_backward_contiguous_impl()
530 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_cpu_backward_channels_last_impl()
711 const Tensor& weight, const Tensor& bias, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_contiguous_impl()
775 const Tensor& weight, const Tensor& bias, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_channels_last_impl()
999 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_cpu_backward_contiguous_internal()
1108 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_cpu_backward_contiguous_impl()
1123 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_cpu_backward_channels_last_internal()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | BatchNorm_miopen.cpp | 113 Tensor save_mean, save_var; in miopen_batch_norm() local
183 save_mean{ save_mean_t, "save_mean", 4 }, in miopen_batch_norm_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | BatchNorm.cpp | 185 Tensor save_mean, save_var; in cudnn_batch_norm() local
296 weight{weight_t, "weight", 3}, save_mean{save_mean_t, "save_mean", 4}, in cudnn_batch_norm_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Normalization.cpp | 256 const Tensor& save_mean = c10::value_or_else(save_mean_opt, [] {return Tensor();}); in mkldnn_batch_norm_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesNorm.cpp | 218 const Tensor& save_mean = *save_mean_opt; in batch_norm_backward_plumbing() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 570 std::vector<torch::lazy::Shape> compute_shape_native_batch_norm_backward( in compute_shape_native_batch_norm_backward()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 4661 const std::optional<Tensor>& save_mean, in batchnorm_double_backward()
4812 auto save_mean = save_mean_t.reshape({M, 1}); in layer_norm_double_backward() local
|