| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 158 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F()
159 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
160 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
161 auto mean_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
163 auto var_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
270 auto input_t = GenerateRandomTensor<DT_FLOAT>(input_shape); in TEST_F()
271 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
272 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
273 auto mean_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
275 auto var_t = GenerateRandomTensor<DT_FLOAT>(is_training ? empty_shape in TEST_F()
[all …]
|
| D | mkl_remapper_test.cc | 104 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}); \
354 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F()
355 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
356 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 267 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()
965 auto a_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 2})); in TEST_F()
966 auto b_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 3})); in TEST_F()
[all …]
|
| D | arithmetic_optimizer_test.cc | 856 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 3, 28, 28})); in TEST_F()
895 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 3, 28, 28})); in TEST_F()
924 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({4, 3, 28, 28})); in TEST_F()
956 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({4, 3, 28, 28})); in TEST_F()
988 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 3, 28, 28})); in TEST_F()
1046 auto x_t = GenerateRandomTensor<DT_INT8>(TensorShape({8, 3, 28, 28, 4})); in TEST_F()
1103 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F()
1150 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F()
1246 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F()
1423 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 12, 28, 28})); in TEST_F()
[all …]
|
| D | model_pruner_test.cc | 302 auto v_in_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3})); in TEST_F()
351 auto a_t = GenerateRandomTensor<DT_INT64>(TensorShape({})); in TEST_F()
|
| D | generic_layout_optimizer_test.cc | 666 Tensor x = GenerateRandomTensor<DT_FLOAT>({2, 6, 6, 8}); in TEST_F()
|
| D | layout_optimizer_test.cc | 254 Tensor filter_data = GenerateRandomTensor<DT_FLOAT>(filter_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_testspec_test.cc | 26 TEST(GenerateRandomTensor, FloatValue) { in TEST() argument
36 const auto& data = GenerateRandomTensor<float>({1, 3, 4}, random_func); in TEST()
|
| D | generate_testspec.h | 59 std::vector<T> GenerateRandomTensor(const std::vector<int>& shape, in GenerateRandomTensor() function
|
| D | generate_testspec.cc | 35 GenerateRandomTensor<T>(shape, [&]() { return distribution(*engine); }); in GenerateCsv()
|
| /external/tensorflow/tensorflow/lite/testing/kernel_test/ |
| D | input_generator.cc | 32 std::vector<T> GenerateRandomTensor(TfLiteIntArray* dims, in GenerateRandomTensor() function
57 std::vector<T> data = GenerateRandomTensor(dims, random_t); in GenerateUniform()
80 std::vector<T> data = GenerateRandomTensor(dims, random_t); in GenerateGaussian()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | grappler_test.h | 89 Tensor GenerateRandomTensor(const TensorShape& shape) const { in GenerateRandomTensor() function
|