Searched refs:CheckIntTensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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()
|
| D | while_test.cc | 26 using subgraph_test_util::CheckIntTensor;
59 CheckIntTensor(output1, {1}, {i + 1}); in TEST_F()
61 CheckIntTensor(output2, {1}, {expected[i]}); in TEST_F()
81 CheckIntTensor(output1, {1}, {4}); in TEST_F()
83 CheckIntTensor(output2, {11}, {0, 0, 0, 5, 7, 0, 0, 0, 0, 0, 0}); in TEST_F()
118 CheckIntTensor(integer_output, {1}, {4}); in TEST_F()
|
| 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()
95 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
104 CheckIntTensor(output, {5}, {0, 5, 7, 0, 0}); in TEST_F()
|
| D | subgraph_test_util.h | 161 void CheckIntTensor(const TfLiteTensor* tensor, const std::vector<int>& shape,
|
| D | subgraph_test_util.cc | 642 void CheckIntTensor(const TfLiteTensor* tensor, const std::vector<int>& shape, in CheckIntTensor() function
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib_test.cc | 36 using subgraph_test_util::CheckIntTensor;
457 CheckIntTensor(output1, {1}, {kSeqNumber + 1}); in TEST_F()
459 CheckIntTensor(output2, {1}, {kExpectedValue}); in TEST_F()
480 CheckIntTensor(output1, {1}, {kSeqNumber + 1}); in TEST_F()
482 CheckIntTensor(output2, {1}, {kExpectedValue}); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer_test.cc | 182 subgraph_test_util::CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
209 subgraph_test_util::CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
|