| /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 …ensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_backward_contiguous_impl()
530 …ensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save_invstd, 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 …ensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_backward_contiguous_internal()
1108 …ensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_backward_contiguous_impl()
1123 …ensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_cpu_backward_channels_last_internal()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cu | 435 …l train, double momentum, double epsilon, Tensor& output, Tensor& save_mean, Tensor& save_invstd) { in batch_norm_cuda_out()
467 auto save_invstd = at::empty({n_input}, options); in batch_norm_cuda() local
540 …l train, double momentum, double epsilon, Tensor& output, Tensor& save_mean, Tensor& save_invstd) { in _batch_norm_legit_cuda_out()
544 …l train, double momentum, double epsilon, Tensor& output, Tensor& save_mean, Tensor& save_invstd) { in _batch_norm_legit_no_stats_cuda_out()
574 c10::MaybeOwned<Tensor> save_invstd = at::borrow_from_optional_tensor(save_invstd_opt); in batch_norm_backward_cuda() local
654 auto save_invstd = at::empty({n_channels}, options); in batch_norm_stats_cuda() local
|
| H A D | Normalization.cuh | 366 …onst GenericPackedTensorAccessor<const stat_accscalar_t, 1, DefaultPtrTraits, index_t> save_invstd, in batch_norm_backward_kernel()
644 auto save_invstd = packed_accessor_or_dummy< in batch_norm_backward_cuda_template() local
764 auto save_invstd = 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()
305 …ensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save_invstd, in batch_norm_backward_cpu_template()
557 auto save_invstd = at::empty_symint(c10::SymIntArrayRef({std::move(num_features)}), options); in _batch_norm_impl_index() local
927 const Tensor& save_invstd = c10::value_or_else(save_invstd_opt, [] {return Tensor();}); in batch_norm_backward_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Normalization.cpp | 257 const Tensor& save_invstd = c10::value_or_else(save_invstd_opt, [] {return Tensor();}); in mkldnn_batch_norm_backward() 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 | 4662 const std::optional<Tensor>& save_invstd, in batchnorm_double_backward()
4813 auto save_invstd = save_invstd_t.reshape({M, 1}); in layer_norm_double_backward() local
|