15 explicit GetTensorIdsFixture(
const std::string& inputs,
const std::string& outputs)
20 "operator_codes": [ { "builtin_code": "AVERAGE_POOL_2D" } ], 25 "shape": [ 1, 1, 1, 1 ] , 28 "name": "OutputTensor", 37 "shape": [ 1, 2, 2, 1 ] , 40 "name": "InputTensor", 59 "builtin_options_type": "Pool2DOptions", 67 "fused_activation_function": "NONE" 69 "custom_options_format": "FLEXBUFFERS" 73 "description": "Test loading a model", 74 "buffers" : [ {}, {} ] 81 struct GetEmptyTensorIdsFixture : GetTensorIdsFixture
83 GetEmptyTensorIdsFixture() : GetTensorIdsFixture(
"[ ]",
"[ ]") {}
86 struct GetInputOutputTensorIdsFixture : GetTensorIdsFixture
88 GetInputOutputTensorIdsFixture() : GetTensorIdsFixture(
"[ 0, 1, 2 ]",
"[ 3 ]") {}
94 m_GraphBinary.size());
95 std::vector<int32_t> expectedIds = { };
96 std::vector<int32_t> inputTensorIds = TfLiteParserImpl::GetInputTensorIds(model, 0, 0);
97 CHECK(std::equal(expectedIds.begin(), expectedIds.end(),
98 inputTensorIds.begin(), inputTensorIds.end()));
104 m_GraphBinary.size());
105 std::vector<int32_t> expectedIds = { };
106 std::vector<int32_t> outputTensorIds = TfLiteParserImpl::GetOutputTensorIds(model, 0, 0);
107 CHECK(std::equal(expectedIds.begin(), expectedIds.end(),
108 outputTensorIds.begin(), outputTensorIds.end()));
114 m_GraphBinary.size());
115 std::vector<int32_t> expectedInputIds = { 0, 1, 2 };
116 std::vector<int32_t> inputTensorIds = TfLiteParserImpl::GetInputTensorIds(model, 0, 0);
117 CHECK(std::equal(expectedInputIds.begin(), expectedInputIds.end(),
118 inputTensorIds.begin(), inputTensorIds.end()));
124 m_GraphBinary.size());
125 std::vector<int32_t> expectedOutputIds = { 3 };
126 std::vector<int32_t> outputTensorIds = TfLiteParserImpl::GetOutputTensorIds(model, 0, 0);
127 CHECK(std::equal(expectedOutputIds.begin(), expectedOutputIds.end(),
128 outputTensorIds.begin(), outputTensorIds.end()));
141 TEST_CASE_FIXTURE(GetInputOutputTensorIdsFixture,
"GetInputTensorIdsInvalidSubgraph")
144 m_GraphBinary.size());
148 TEST_CASE_FIXTURE( GetInputOutputTensorIdsFixture,
"GetOutputTensorIdsInvalidSubgraph")
151 m_GraphBinary.size());
155 TEST_CASE_FIXTURE(GetInputOutputTensorIdsFixture,
"GetInputTensorIdsInvalidOperator")
158 m_GraphBinary.size());
162 TEST_CASE_FIXTURE(GetInputOutputTensorIdsFixture,
"GetOutputTensorIdsInvalidOperator")
165 m_GraphBinary.size());
TfLiteParserImpl::ModelPtr ModelPtr
bool ReadStringToBinary()
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
TEST_SUITE("TensorflowLiteParser_GetTensorIds")