Searched refs:GenerateTensorWithSetRandom (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | remapper_test.cc | 642 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 32, 3}); in RunTest()
643 auto filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 3, 128}); in RunTest()
644 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest()
653 input_t = GenerateTensorWithSetRandom<DTYPE>({8, 4, 32, 32, 3}); in RunTest()
654 filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 1, 3, 128}); in RunTest()
655 bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest()
749 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 32, 3}); in RunTest()
750 auto filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 3, 128}); in RunTest()
751 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest()
760 input_t = GenerateTensorWithSetRandom<DTYPE>({8, 4, 32, 32, 3}); in RunTest()
[all …]
|
| H A D | mkl_remapper_test.cc | 687 auto lhs_t = GenerateTensorWithSetRandom<DTYPE>({8, 32}); in RunTest()
688 auto rhs_t = GenerateTensorWithSetRandom<DTYPE>({32, 64}); in RunTest()
689 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({64}); in RunTest()
868 auto mul_t = GenerateTensorWithSetRandom<DTYPE>({64, 64}); in RunTest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | grappler_test.h | 99 Tensor GenerateTensorWithSetRandom(const TensorShape& shape) const { in GenerateTensorWithSetRandom() function
|