• Home
  • Raw
  • Download

Lines Matching refs:subgraphTensors

157     TfLiteParser::TensorIdRawPtrVector subgraphTensors = TfLiteParser::GetSubgraphInputs(model, 0);  in BOOST_FIXTURE_TEST_CASE()  local
158 BOOST_CHECK_EQUAL(0, subgraphTensors.size()); in BOOST_FIXTURE_TEST_CASE()
164 TfLiteParser::TensorIdRawPtrVector subgraphTensors = TfLiteParser::GetSubgraphOutputs(model, 0); in BOOST_FIXTURE_TEST_CASE() local
165 BOOST_CHECK_EQUAL(0, subgraphTensors.size()); in BOOST_FIXTURE_TEST_CASE()
171 TfLiteParser::TensorIdRawPtrVector subgraphTensors = TfLiteParser::GetSubgraphInputs(model, 0); in BOOST_FIXTURE_TEST_CASE() local
172 BOOST_CHECK_EQUAL(1, subgraphTensors.size()); in BOOST_FIXTURE_TEST_CASE()
173 BOOST_CHECK_EQUAL(1, subgraphTensors[0].first); in BOOST_FIXTURE_TEST_CASE()
174 …CheckTensors(subgraphTensors[0].second, 4, { 1, 2, 2, 1 }, tflite::TensorType::TensorType_UINT8, 1, in BOOST_FIXTURE_TEST_CASE()
181 TfLiteParser::TensorIdRawPtrVector subgraphTensors = TfLiteParser::GetSubgraphOutputs(model, 0); in BOOST_FIXTURE_TEST_CASE() local
182 BOOST_CHECK_EQUAL(1, subgraphTensors.size()); in BOOST_FIXTURE_TEST_CASE()
183 BOOST_CHECK_EQUAL(0, subgraphTensors[0].first); in BOOST_FIXTURE_TEST_CASE()
184 …CheckTensors(subgraphTensors[0].second, 4, { 1, 1, 1, 1 }, tflite::TensorType::TensorType_UINT8, 0, in BOOST_FIXTURE_TEST_CASE()
191 TfLiteParser::TensorIdRawPtrVector subgraphTensors = TfLiteParser::GetSubgraphInputs(model, 1); in BOOST_FIXTURE_TEST_CASE() local
192 BOOST_CHECK_EQUAL(1, subgraphTensors.size()); in BOOST_FIXTURE_TEST_CASE()
193 BOOST_CHECK_EQUAL(0, subgraphTensors[0].first); in BOOST_FIXTURE_TEST_CASE()
194 …CheckTensors(subgraphTensors[0].second, 4, { 1, 3, 3, 1 }, tflite::TensorType::TensorType_UINT8, 0, in BOOST_FIXTURE_TEST_CASE()
201 TfLiteParser::TensorIdRawPtrVector subgraphTensors = TfLiteParser::GetSubgraphOutputs(model, 1); in BOOST_FIXTURE_TEST_CASE() local
202 BOOST_CHECK_EQUAL(1, subgraphTensors.size()); in BOOST_FIXTURE_TEST_CASE()
203 BOOST_CHECK_EQUAL(1, subgraphTensors[0].first); in BOOST_FIXTURE_TEST_CASE()
204 …CheckTensors(subgraphTensors[0].second, 4, { 1, 1, 1, 1 }, tflite::TensorType::TensorType_UINT8, 1, in BOOST_FIXTURE_TEST_CASE()