Searched refs:argMinMaxOperatorCode (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | ArgMinMax.hpp | 21 int32_t argMinMaxOperatorCode) in VisitArgMinMaxOperator() argument
28 if (!IsValid(tfLiteContext, tfLiteInputTensor, argMinMaxOperatorCode, nodeIndex)) in VisitArgMinMaxOperator()
34 if (!IsValid(tfLiteContext, tfLiteOutputTensor, argMinMaxOperatorCode, nodeIndex)) in VisitArgMinMaxOperator()
44 if (!IsValid(tfLiteContext, tfLiteAxisTensor, argMinMaxOperatorCode, nodeIndex)) in VisitArgMinMaxOperator()
61 argMinMaxOperatorCode, nodeIndex); in VisitArgMinMaxOperator()
66 if (argMinMaxOperatorCode == kTfLiteBuiltinArgMax) in VisitArgMinMaxOperator()
75 argMinMaxOperatorCode, nodeIndex); in VisitArgMinMaxOperator()
88 argMinMaxOperatorCode, nodeIndex); in VisitArgMinMaxOperator()
|
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | ArgMinMaxTestHelper.hpp | 25 std::vector<char> CreateArgMinMaxTfLiteModel(tflite::BuiltinOperator argMinMaxOperatorCode, in CreateArgMinMaxTfLiteModel() argument
85 if (argMinMaxOperatorCode == tflite::BuiltinOperator_ARG_MIN) in CreateArgMinMaxTfLiteModel()
112 argMinMaxOperatorCode); in CreateArgMinMaxTfLiteModel()
129 void ArgMinMaxTest(tflite::BuiltinOperator argMinMaxOperatorCode, in ArgMinMaxTest() argument
143 … std::vector<char> modelBuffer = CreateArgMinMaxTfLiteModel<InputT, OutputT>(argMinMaxOperatorCode, in ArgMinMaxTest()
|