Go to the source code of this file.
◆ TEST_SUITE()
TEST_SUITE |
( |
"TensorflowLiteParser_Cast" |
| ) |
|
Definition at line 9 of file Cast.cpp.
References ParserFlatbuffersFixture::SetupSingleInputSingleOutput(), and TEST_CASE_FIXTURE().
13 explicit CastFixture(
const std::string& inputShape,
14 const std::string& outputShape,
15 const std::string& inputDataType,
16 const std::string& outputDataType)
21 "operator_codes": [ { "builtin_code": "CAST" } ], 25 "shape": )" + inputShape + R"(, 26 "type": )" + inputDataType + R"(, 28 "name": "inputTensor", 37 "shape": )" + outputShape + R"(, 38 "type": )" + outputDataType + R"(, 40 "name": "outputTensor", 56 "custom_options_format": "FLEXBUFFERS" 60 "buffers" : [ {}, {} ] 67 struct SimpleCastFixture : CastFixture
69 SimpleCastFixture() : CastFixture(
"[ 1, 6 ]",
77 RunTest<2, armnn::DataType::Signed32 , armnn::DataType::Float32>(
79 {{
"inputTensor", { 0, -1, 5, -100, 200, -255 }}},
80 {{
"outputTensor", { 0.0f, -1.0f, 5.0f, -100.0f, 200.0f, -255.0f }}});
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)