Searched defs:mean_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | BatchNormalizationLayerFixture.h | 60 void fill(U &&src_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U &&gamma_tensor) in fill()
|
| D | NormalizePlanarYUVLayerFixture.h | 57 void fill(U &&src_tensor, U &&mean_tensor, U &&std_tensor) in fill()
|
| D | BatchNormalizationLayerFusionFixture.h | 66 …void fill(U &&src, U &&w_tensor, U &&b_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U… in fill()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | DistributionTemplates.h | 232 auto mean_tensor = at::full({}, mean, output.options()); in normal_out_impl() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | metrics_impl.py | 1513 def mean_tensor(values, function
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 4954 const Tensor mean_tensor = mean.reshape_symint({N, G, 1, 1}); in infinitely_differentiable_native_group_norm_backward() local
|