Searched defs:mean_out (Results 1 – 6 of 6) sorted by relevance
| /external/executorch/kernels/portable/cpu/util/ |
| D | normalization_ops_util.cpp | 27 Tensor& mean_out, in check_batch_norm_args()
81 Tensor& mean_out, in check_layer_norm_args()
142 Tensor& mean_out, in check_group_norm_args()
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_native_batch_norm.cpp | 34 Tensor& mean_out, in _native_batch_norm_legit_no_training_out()
150 Tensor& mean_out, in _native_batch_norm_legit_out()
186 Tensor& mean_out, in _native_batch_norm_legit_no_stats_out()
|
| D | op_native_layer_norm.cpp | 107 Tensor& mean_out, in native_layer_norm_out()
|
| D | op_native_group_norm.cpp | 126 Tensor& mean_out, in native_group_norm_out()
|
| /external/executorch/kernels/optimized/cpu/ |
| D | op_native_layer_norm.cpp | 122 Tensor& mean_out, in opt_native_layer_norm_out()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReduceOps.cpp | 1348 TORCH_IMPL_FUNC(mean_out) in TORCH_IMPL_FUNC() argument
1412 Tensor& mean_out(const Tensor& self, DimnameList dim, in mean_out() function
|