| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 138 const Tensor& running_mean /* optional */, const Tensor& running_var /* optional */, in batch_norm_cpu_transform_input_template()
201 const Tensor& input, const Tensor& running_mean, const Tensor& running_var, in batch_norm_cpu_update_stats_template()
285 const Tensor& input, const Tensor& running_mean, const Tensor& running_var, in batch_norm_cpu_update_stats_template()
305 …const Tensor& running_mean, const Tensor& running_var, const Tensor& save_mean, const Tensor& save… in batch_norm_backward_cpu_template()
489 const Tensor& running_var, bool training, double eps) { in _select_batch_norm_backend()
548 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in _batch_norm_impl_index() local
629 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in _batch_norm_impl_index_backward() local
675 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in batch_norm() local
712 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in instance_norm() local
747 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in batch_norm_update_stats_cpu() local
[all …]
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_native_batch_norm_test.cpp | 28 const exec_aten::Tensor& running_var, in op_native_batch_norm_legit_no_training_out()
57 exec_aten::Tensor& running_var, in op_native_batch_norm_legit_out()
151 exec_aten::Tensor running_var = tfFloat.make( in TEST_F() local
274 exec_aten::Tensor running_var = tfFloat.make( in TEST_F() local
446 exec_aten::Tensor running_var = tfFloat.make( in TEST_F() local
606 exec_aten::Tensor running_var = tfDouble.make( in TEST_F() local
744 exec_aten::Tensor running_var = tfFloat.make( in TEST_F() local
850 exec_aten::Tensor running_var = tfFloat.make( in TEST_F() local
935 exec_aten::Tensor running_var = tfFloat.make( in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cu | 348 const Tensor& running_mean, const Tensor& running_var, in batch_norm_update_stats()
382 const Tensor& running_mean, const Tensor& running_var, in batch_norm_update_stats_and_invert()
417 void batch_norm_calc_invstd(const Tensor& out_invstd, const Tensor& running_var, double epsilon) { in batch_norm_calc_invstd()
486 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_cuda()
511 Tensor& running_mean, Tensor& running_var, double momentum, double eps, in _batch_norm_with_update_cuda_out()
532 …_opt, const std::optional<Tensor>& bias_opt, Tensor& running_mean, Tensor& running_var, bool train… in _batch_norm_legit_cuda()
540 …_opt, const std::optional<Tensor>& bias_opt, Tensor& running_mean, Tensor& running_var, bool train… in _batch_norm_legit_cuda_out()
555 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in _new_batch_norm_backward_cuda() local
576 c10::MaybeOwned<Tensor> running_var = at::borrow_from_optional_tensor(running_var_opt); in batch_norm_backward_cuda() local
697 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in batch_norm_gather_stats_cuda() local
[all …]
|
| H A D | Normalization.cuh | 364 … const GenericPackedTensorAccessor<const stat_scalar_t, 1, DefaultPtrTraits, index_t> running_var, in batch_norm_backward_kernel()
433 GenericPackedTensorAccessor<scalar_t, 1, RestrictPtrTraits, index_t> running_var, in batch_norm_reduce_statistics_kernel()
640 auto running_var = packed_accessor_or_dummy< in batch_norm_backward_cuda_template() local
757 auto running_var = packed_accessor_or_dummy< in batch_norm_gather_stats_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Normalization.cpp | 49 …sor>& weight_opt, const std::optional<Tensor>& bias_opt, Tensor& running_mean, Tensor& running_var, in _mkldnn_batch_norm_legit()
67 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_mkldnn()
143 const Tensor& running_var = c10::value_or_else(running_var_opt, [] {return Tensor();}); in mkldnn_batch_norm() local
213 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_mkldnn()
222 …sor>& weight_opt, const std::optional<Tensor>& bias_opt, Tensor& running_mean, Tensor& running_var, in _mkldnn_batch_norm_legit()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | batch_norm_kernel.cpp | 35 const Tensor& running_mean, const Tensor& running_var, bool train, double eps) { in batch_norm_cpu_collect_linear_and_constant_terms()
77 const Tensor& running_mean, const Tensor& running_var, bool train, double eps) { in batch_norm_cpu_contiguous_impl()
129 const Tensor& running_mean, const Tensor& running_var, bool train, double eps) { 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()
712 const Tensor& running_mean, const Tensor& running_var, bool train, double eps) { in batch_norm_cpu_contiguous_impl()
776 const Tensor& running_mean, const Tensor& running_var, bool train, double eps) { 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/executorch/kernels/portable/cpu/ |
| H A D | op_native_batch_norm.cpp | 30 const Tensor& running_var, in _native_batch_norm_legit_no_training_out()
145 Tensor& running_var, in _native_batch_norm_legit_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | BatchNorm_miopen.cpp | 74 running_var{ running_var_t, "running_var", 5 }; in miopen_batch_norm() local
173 const Tensor& running_var = in miopen_batch_norm_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesNorm.cpp | 60 const auto& running_var = *running_var_maybe_owned; in batch_norm_batch_rule() local
138 const Tensor& running_var = *running_var_maybe_owned; in batch_norm_backward_no_weight_bias_batch_rule() local
216 const Tensor& running_var = *running_var_maybe_owned; in batch_norm_backward_plumbing() local
854 Tensor& running_mean, Tensor& running_var, bool train, double momentum, double eps) { in _native_batch_norm_legit_batch()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | instancenorm.h | 14 const Tensor& running_var, in instance_norm()
|
| H A D | batchnorm.h | 16 const Tensor& running_var, in batch_norm()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | batchnorm.h | 91 Tensor running_var; variable
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | _functions.py | 15 running_var, argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_functionalization.py | 1913 def f(x, running_mean, running_var): argument
2037 def f(x, running_mean, running_var): argument
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | normalization_ops_util.cpp | 23 const exec_aten::optional<Tensor>& running_var, in check_batch_norm_args()
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_inductor_freezing.py | 103 running_var=None, argument
151 running_var=None, argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset14.py | 75 running_var, argument
|
| H A D | symbolic_helper.py | 1366 g: jit_utils.GraphContext, input, weight, bias, running_mean, running_var argument
|
| H A D | symbolic_opset9.py | 2632 running_var, argument
2784 running_var, argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | decomposition_skip.py | 202 running_var, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | BatchNorm.cpp | 141 running_var{running_var_t, "running_var", 5}; in cudnn_batch_norm() local
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | common_utils.py | 172 num_tensors, _, running_mean=None, running_var=None, *args argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 531 std::vector<torch::lazy::Shape> compute_shape_native_batch_norm( in compute_shape_native_batch_norm()
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/ |
| H A D | Module.cpp | 2326 double eps) { in initModule()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 4658 const std::optional<Tensor>& running_var, in batchnorm_double_backward()
6197 const std::optional<Tensor>& running_var, in batch_norm_jvp()
|