Searched refs:reduceOperatorCode (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Reduce.hpp | 21 int32_t reduceOperatorCode) in VisitReduceOperator() argument
28 if (!IsValid(tfLiteContext, tfLiteInputTensor, reduceOperatorCode, nodeIndex)) in VisitReduceOperator()
34 if (!IsValid(tfLiteContext, tfLiteOutputTensor, reduceOperatorCode, nodeIndex)) in VisitReduceOperator()
44 if (!IsValid(tfLiteContext, tfLiteAxisTensor, reduceOperatorCode, nodeIndex)) in VisitReduceOperator()
82 if (reduceOperatorCode == kTfLiteBuiltinReduceMax) in VisitReduceOperator()
86 else if (reduceOperatorCode == kTfLiteBuiltinReduceMin) in VisitReduceOperator()
90 else if (reduceOperatorCode == kTfLiteBuiltinSum) in VisitReduceOperator()
94 else if (reduceOperatorCode == kTfLiteBuiltinReduceProd) in VisitReduceOperator()
103 reduceOperatorCode, nodeIndex); in VisitReduceOperator()
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | ReduceTest.cpp | 18 void ReduceUint8KeepDimsTest(tflite::BuiltinOperator reduceOperatorCode, in ReduceUint8KeepDimsTest() argument
30 ReduceTest<uint8_t>(reduceOperatorCode, in ReduceUint8KeepDimsTest()
42 void ReduceUint8Test(tflite::BuiltinOperator reduceOperatorCode, in ReduceUint8Test() argument
54 ReduceTest<uint8_t>(reduceOperatorCode, in ReduceUint8Test()
66 void ReduceFp32KeepDimsTest(tflite::BuiltinOperator reduceOperatorCode, in ReduceFp32KeepDimsTest() argument
78 ReduceTest<float>(reduceOperatorCode, in ReduceFp32KeepDimsTest()
90 void ReduceFp32Test(tflite::BuiltinOperator reduceOperatorCode, in ReduceFp32Test() argument
102 ReduceTest<float>(reduceOperatorCode, in ReduceFp32Test()
|
| H A D | ReduceTestHelper.hpp | 24 std::vector<char> CreateReduceTfLiteModel(tflite::BuiltinOperator reduceOperatorCode, in CreateReduceTfLiteModel() argument
131 …rs::Offset <OperatorCode> operatorCode = CreateOperatorCode(flatBufferBuilder, reduceOperatorCode); in CreateReduceTfLiteModel()
148 void ReduceTest(tflite::BuiltinOperator reduceOperatorCode, in ReduceTest() argument
162 std::vector<char> modelBufferArmNN = CreateReduceTfLiteModel(reduceOperatorCode, in ReduceTest()
172 std::vector<char> modelBufferTFLite = CreateReduceTfLiteModel(reduceOperatorCode, in ReduceTest()
|