Searched refs:softmaxOperatorCode (Results 1 – 2 of 2) sorted by relevance
24 std::vector<char> CreateSoftmaxTfLiteModel(tflite::BuiltinOperator softmaxOperatorCode, in CreateSoftmaxTfLiteModel() argument56 switch (softmaxOperatorCode) in CreateSoftmaxTfLiteModel()105 void SoftmaxTest(tflite::BuiltinOperator softmaxOperatorCode, in SoftmaxTest() argument114 std::vector<char> modelBuffer = CreateSoftmaxTfLiteModel(softmaxOperatorCode, in SoftmaxTest()
61 int32_t softmaxOperatorCode) in VisitSoftmaxOperator() argument92 switch(softmaxOperatorCode) in VisitSoftmaxOperator()121 switch(softmaxOperatorCode) in VisitSoftmaxOperator()