Searched refs:std_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | NormalizePlanarYUVLayerFixture.h | 57 void fill(U &&src_tensor, U &&mean_tensor, U &&std_tensor) in fill() argument
69 library->fill(std_tensor, distribution_std, 2); in fill()
79 library->fill(std_tensor, distribution_std, 2); in fill()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | DistributionTemplates.h | 221 auto std_tensor = at::empty_like(output, MemoryFormat::Contiguous); in normal_out_impl() local
222 auto shape = at::infer_size(mean.sizes(), std_tensor.sizes()); in normal_out_impl()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 7872 std_tensor = cpu_std_tensor.detach().clone().to('mps')
7879 torch.normal(mean, std_tensor, out=mps_out)
7882 torch.normal(mean_tensor, std_tensor, out=mps_out)
7888 mps_out = torch.normal(mean, std_tensor)
7889 self.assertEqual(mps_out.size(), std_tensor.size())
7891 inferred_shape = torch.broadcast_shapes(mean_tensor.size(), std_tensor.size())
7892 mps_out = torch.normal(mean_tensor, std_tensor)
|