Home
last modified time | relevance | path

Searched refs:GenerateRandomTensor (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc161 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F()
162 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
163 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
164 auto mean_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
166 auto var_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
269 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F()
270 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
271 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
272 auto mean_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
273 auto var_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
[all …]
H A Dmkl_remapper_test.cc104 auto input_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd()
106 auto input_addn_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd()
108 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd()
110 auto bias_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd()
255 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); \
256 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 1}); \
257 auto bias_t = GenerateRandomTensor<DT_FLOAT>({3}); \
353 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F()
354 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
355 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
[all …]
H A Dconstant_folding_test.cc267 auto in1_t = GenerateRandomTensor<DT_INT32>(TensorShape({4, 6})); in PaddingWithZeroSize()
268 auto in2_t = GenerateRandomTensor<DT_INT32>(TensorShape({2, 2})); in PaddingWithZeroSize()
388 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
389 auto y_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
445 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
516 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
592 auto x_mat_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
593 auto x_vec_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2})); in TEST_F()
970 auto a_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 2})); in TEST_F()
971 auto b_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 3})); in TEST_F()
[all …]
H A Darithmetic_optimizer_test.cc126 GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 44, 1, 96, 1, 64})); in TEST_F()
169 auto tensor = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 1, 1})); in TEST_F()
202 auto tensor = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 2, 1})); in TEST_F()
232 auto tensor1 = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 1, 1})); in TEST_F()
233 auto tensor2 = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 2, 1})); in TEST_F()
263 auto tensor = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 1, 1})); in TEST_F()
299 auto tensor = GenerateRandomTensor<DT_FLOAT>(TensorShape({1, 22, 48, 64})); in TEST_F()
404 auto a_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 5, 7, 11})); in TEST_F()
473 auto a_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 5, 7, 11})); in TEST_F()
532 auto a_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 5, 7, 11})); in TEST_F()
[all …]
H A Dmodel_pruner_test.cc302 auto v_in_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3})); in TEST_F()
351 auto a_t = GenerateRandomTensor<DT_INT64>(TensorShape({})); in TEST_F()
H A Dgeneric_layout_optimizer_test.cc636 Tensor x = GenerateRandomTensor<DT_FLOAT>({2, 6, 6, 8}); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/kernel_test/
H A Dinput_generator.cc38 std::vector<T> GenerateRandomTensor(TfLiteIntArray* dims, in GenerateRandomTensor() function
63 std::vector<T> data = GenerateRandomTensor(dims, random_t); in GenerateUniform()
86 std::vector<T> data = GenerateRandomTensor(dims, random_t); in GenerateGaussian()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dgenerate_testspec_test.cc26 TEST(GenerateRandomTensor, FloatValue) { in TEST() argument
36 const auto& data = GenerateRandomTensor<float>({1, 3, 4}, random_func); in TEST()
H A Dgenerate_testspec.h59 std::vector<T> GenerateRandomTensor(const std::vector<int>& shape, in GenerateRandomTensor() function
H A Dgenerate_testspec.cc41 GenerateRandomTensor<T>(shape, [&]() { return distribution(*engine); }); in GenerateCsv()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgrappler_test.h89 Tensor GenerateRandomTensor(const TensorShape& shape) const { in GenerateRandomTensor() function