Searched refs:CheckIntTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util_test.cc | 54 CheckIntTensor(output_tensor1, {1}, {output1}); in TestAccumelateLoopBody()
57 CheckIntTensor(output_tensor2, {1}, {output2}); in TestAccumelateLoopBody()
76 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
91 CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
106 CheckIntTensor(output, {5}, {0, 5, 7, 0, 0}); in TEST_F()
144 CheckIntTensor(output1, {1}, {2}); in TEST_F()
146 CheckIntTensor(output2, {8}, {0, 0, 5, 7, 0, 0, 0, 0}); in TEST_F()
|
| D | while_test.cc | 26 using subgraph_test_util::CheckIntTensor;
56 CheckIntTensor(output1, {1}, {i + 1}); in TEST_F()
58 CheckIntTensor(output2, {1}, {expected[i]}); in TEST_F()
82 CheckIntTensor(output1, {1}, {4}); in TEST_F()
84 CheckIntTensor(output2, {11}, {0, 0, 0, 5, 7, 0, 0, 0, 0, 0, 0}); in TEST_F()
|
| D | if_test.cc | 27 using subgraph_test_util::CheckIntTensor;
57 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
64 CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
94 CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
103 CheckIntTensor(output, {5}, {0, 5, 7, 0, 0}); in TEST_F()
|
| D | subgraph_test_util.h | 114 void CheckIntTensor(const TfLiteTensor* tensor, const std::vector<int>& shape,
|
| D | subgraph_test_util.cc | 380 void CheckIntTensor(const TfLiteTensor* tensor, const std::vector<int>& shape, in CheckIntTensor() function
|