| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 124 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F()
125 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F()
126 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F()
188 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F()
189 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F()
190 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F()
258 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F()
259 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F()
260 auto scale_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F()
261 auto offset_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 651 auto a_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 2})); in TEST_F()
652 auto b_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 3})); in TEST_F()
653 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
654 auto y_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
655 auto bias_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2})); in TEST_F()
802 auto x_known_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
804 GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 4})); in TEST_F()
805 auto x_unknown_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({5, 7})); in TEST_F()
873 GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
874 auto x_unknown_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F()
[all …]
|
| D | arithmetic_optimizer_test.cc | 911 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 3, 28, 28})); in TEST_F()
948 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3, 3, 28, 28})); in TEST_F()
977 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({4, 3, 28, 28})); in TEST_F()
1009 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({4, 3, 28, 28})); in TEST_F()
1041 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 3, 28, 28})); in TEST_F()
1098 auto x_t = GenerateRandomTensor<DT_INT8>(TensorShape({8, 3, 28, 28, 4})); in TEST_F()
1127 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F()
1173 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F()
1266 auto input_t = GenerateRandomTensor<DT_UINT8>(TensorShape({8, 28, 28, 3})); in TEST_F()
1409 auto x_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({8, 12, 28, 28})); in TEST_F()
[all …]
|
| D | model_pruner_test.cc | 328 auto v_in_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({3})); in TEST_F()
382 auto a_t = GenerateRandomTensor<DT_INT64>(TensorShape({})); in TEST_F()
|
| D | layout_optimizer_test.cc | 248 Tensor filter_data = GenerateRandomTensor<DT_FLOAT>(filter_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_testspec_test.cc | 24 TEST(GenerateRandomTensor, FloatValue) { in TEST() argument
34 const auto& data = GenerateRandomTensor<float>({1, 3, 4}, float_rand); in TEST()
|
| D | generate_testspec.h | 51 std::vector<T> GenerateRandomTensor(const std::vector<int>& shape, in GenerateRandomTensor() function
|
| D | generate_testspec.cc | 38 std::vector<T> data = GenerateRandomTensor(shape, random_t); 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 | 82 Tensor GenerateRandomTensor(const TensorShape& shape) const { in GenerateRandomTensor() function
|