Home
last modified time | relevance | path

Searched refs:softmaxOperatorCode (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/armnn/delegate/test/
H A DSoftmaxTestHelper.hpp24 std::vector<char> CreateSoftmaxTfLiteModel(tflite::BuiltinOperator softmaxOperatorCode, in CreateSoftmaxTfLiteModel() argument
56 switch (softmaxOperatorCode) in CreateSoftmaxTfLiteModel()
105 void SoftmaxTest(tflite::BuiltinOperator softmaxOperatorCode, in SoftmaxTest() argument
114 std::vector<char> modelBuffer = CreateSoftmaxTfLiteModel(softmaxOperatorCode, in SoftmaxTest()
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DSoftmax.hpp61 int32_t softmaxOperatorCode) in VisitSoftmaxOperator() argument
92 switch(softmaxOperatorCode) in VisitSoftmaxOperator()
121 switch(softmaxOperatorCode) in VisitSoftmaxOperator()