13 explicit DepthToSpaceFixture(
const std::string& inputShape,
14 const std::string& outputShape,
15 const std::string& dataType =
"FLOAT32",
16 const std::string& scale =
"1.0",
17 const std::string& offset =
"0")
22 "operator_codes": [ { "builtin_code": "DEPTH_TO_SPACE" } ], 26 "shape": )" + inputShape + R"(, 27 "type": )" + dataType + R"(, 29 "name": "inputTensor", 33 "scale": [ )" + scale + R"( ], 34 "zero_point": [ )" + offset + R"( ], 38 "shape": )" + outputShape + R"(, 39 "type": )" + dataType + R"(, 41 "name": "outputTensor", 45 "scale": [ )" + scale + R"( ], 46 "zero_point": [ )" + offset + R"( ], 57 "builtin_options_type": "DepthToSpaceOptions", 61 "custom_options_format": "FLEXBUFFERS" 75 struct SimpleDepthToSpaceFixture :
public DepthToSpaceFixture
77 SimpleDepthToSpaceFixture() : DepthToSpaceFixture(
"[ 1, 2, 2, 4 ]",
"[ 1, 4, 4, 1 ]") {}
82 RunTest<4, armnn::DataType::Float32>
84 {{
"inputTensor", { 1.f, 2.f, 3.f, 4.f,
86 9.f, 10.f, 11.f, 12.f,
87 13.f, 14.f, 15.f, 16.f }}},
88 {{
"outputTensor", { 1.f, 2.f, 5.f, 6.f,
90 9.f, 10.f, 13.f, 14.f,
91 11.f, 12.f, 15.f, 16.f }}});
TEST_SUITE("TensorflowLiteParser_DepthToSpace")
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)