13 explicit ProdFixture(
const std::string& inputShape,
14 const std::string& outputShape,
15 const std::string& axisShape,
16 const std::string& axisData)
21 "operator_codes": [ { "builtin_code": "PROD" } ], 25 "shape": )" + inputShape + R"(, 28 "name": "inputTensor", 37 "shape": )" + outputShape + R"( , 40 "name": "outputTensor", 49 "shape": )" + axisShape + R"( , 68 "builtin_options_type": "ReducerOptions", 72 "custom_options_format": "FLEXBUFFERS" 79 { "data": )" + axisData + R"(, }, 87 struct SimpleProdFixture :
public ProdFixture
89 SimpleProdFixture() : ProdFixture(
"[ 1, 3, 2, 4 ]",
"[ 1, 1, 1, 4 ]",
"[ 2 ]",
"[ 1, 0, 0, 0, 2, 0, 0, 0 ]") {}
94 RunTest<4, armnn::DataType::Float32, armnn::DataType::Float32>
95 (0, {{
"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f,
96 5.0f, 6.0f, 7.0f, 8.0f,
98 10.0f, 20.0f, 30.0f, 40.0f,
99 50.0f, 60.0f, 70.0f, 80.0f,
101 11.0f, 22.0f, 33.0f, 44.0f,
102 55.0f, 66.0f, 77.0f, 88.0f } } },
103 {{
"outputTensor", { 1512500.f, 20908800.f, 112058100.f, 396492800.f } } });
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)