Home
last modified time | relevance | path

Searched defs:mean_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DBatchNormalizationLayerFixture.h60 void fill(U &&src_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U &&gamma_tensor) in fill()
H A DNormalizePlanarYUVLayerFixture.h57 void fill(U &&src_tensor, U &&mean_tensor, U &&std_tensor) in fill()
H A DBatchNormalizationLayerFusionFixture.h66 …void fill(U &&src, U &&w_tensor, U &&b_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U… in fill()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistributionTemplates.h232 auto mean_tensor = at::full({}, mean, output.options()); in normal_out_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dmetrics_impl.py1513 def mean_tensor(values, function
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp4954 const Tensor mean_tensor = mean.reshape_symint({N, G, 1, 1}); in infinitely_differentiable_native_group_norm_backward() local