13 explicit MeanNoReduceFixture(
const std::string & inputShape,
14 const std::string & outputShape,
15 const std::string & dimShape,
16 const std::string & dimData)
21 "operator_codes": [ { "builtin_code": "MEAN" } ], 25 "shape": )" + inputShape + R"(, 28 "name": "inputTensor", 37 "shape": )" + outputShape + R"( , 40 "name": "outputTensor", 49 "shape": )" + dimShape + R"( , 68 "custom_options_format": "FLEXBUFFERS" 75 { "data": )" + dimData + R"(, }, 83 struct SimpleMeanNoReduceFixture :
public MeanNoReduceFixture
85 SimpleMeanNoReduceFixture() : MeanNoReduceFixture(
"[ 2, 2 ]",
"[ 1, 1 ]",
"[ 0 ]",
"[ ]") {}
90 RunTest<2, armnn::DataType::Float32>(0, {{
"inputTensor", { 1.0f, 1.0f, 2.0f, 2.0f } } },
91 {{
"outputTensor", { 1.5f } } });
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
TEST_SUITE("TensorflowLiteParser_Mean")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)