| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | if_test.cc | 28 using subgraph_test_util::CheckIntTensor;
58 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
65 CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
83 CheckIntTensor(output, {kNumLargeTensors}, expected); in TEST_F()
91 CheckIntTensor(output, {kNumLargeTensors}, expected2); in TEST_F()
109 CheckIntTensor(output, {kNumLargeTensors}, expected); in TEST_F()
116 CheckIntTensor(output, {kNumLargeTensors}, expected2); in TEST_F()
146 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
155 CheckIntTensor(output, {5}, {0, 5, 7, 0, 0}); in TEST_F()
|
| H A D | while_test.cc | 27 using subgraph_test_util::CheckIntTensor;
63 CheckIntTensor(output1, {1}, {i + 1}); in TEST_F()
65 CheckIntTensor(output2, {1}, {expected[i]}); in TEST_F()
106 CheckIntTensor(output1, {1}, {i + 1}); in TEST_F()
109 CheckIntTensor(output2, {1000000}, expected2); in TEST_F()
129 CheckIntTensor(output1, {1}, {4}); in TEST_F()
131 CheckIntTensor(output2, {11}, {0, 0, 0, 5, 7, 0, 0, 0, 0, 0, 0}); in TEST_F()
159 CheckIntTensor(output1, {1}, {4}); in TEST_F()
164 CheckIntTensor(output2, {1000009}, output_vector); in TEST_F()
199 CheckIntTensor(integer_output, {1}, {4}); in TEST_F()
|
| H A D | subgraph_test_util_test.cc | 61 CheckIntTensor(output_tensor1, {1}, {output1}); in TestAccumulateLoopBody()
64 CheckIntTensor(output_tensor2, {1}, {output2}); in TestAccumulateLoopBody()
83 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
98 CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
113 CheckIntTensor(output, {5}, {0, 5, 7, 0, 0}); in TEST_F()
129 CheckIntTensor(output, {5}, {0, 5, 7, 0, 0}); in TEST_F()
167 CheckIntTensor(output1, {1}, {2}); in TEST_F()
169 CheckIntTensor(output2, {8}, {0, 0, 5, 7, 0, 0, 0, 0}); in TEST_F()
|
| H A D | subgraph_test_util.h | 161 void CheckIntTensor(const TfLiteTensor* tensor, const std::vector<int>& shape,
|
| H A D | subgraph_test_util.cc | 681 void CheckIntTensor(const TfLiteTensor* tensor, const std::vector<int>& shape, in CheckIntTensor() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | call_test.cc | 207 subgraph_test_util::CheckIntTensor(output, test_shapes[i], in TEST_F()
223 subgraph_test_util::CheckIntTensor(output, {0, 3}, {}); in TEST_F()
253 subgraph_test_util::CheckIntTensor(output, kCallOpInputShape, in TEST_F()
278 subgraph_test_util::CheckIntTensor(output_mul, test_shapes[i], in TEST_F()
281 subgraph_test_util::CheckIntTensor(output_add, test_shapes[i], in TEST_F()
330 subgraph_test_util::CheckIntTensor( in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/ |
| H A D | writer_lib_test.cc | 41 using subgraph_test_util::CheckIntTensor;
545 CheckIntTensor(output1, {1}, {kSeqNumber + 1}); in TEST_F()
547 CheckIntTensor(output2, {1}, {kExpectedValue}); in TEST_F()
568 CheckIntTensor(output1, {1}, {kSeqNumber + 1}); in TEST_F()
570 CheckIntTensor(output2, {1}, {kExpectedValue}); in TEST_F()
601 CheckIntTensor(output1, {1}, {kSeqNumber + 1}); in TEST_F()
603 CheckIntTensor(output2, {1}, {kExpectedValue}); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/profiling/ |
| H A D | profile_summarizer_test.cc | 183 subgraph_test_util::CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
210 subgraph_test_util::CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
|