Searched refs:softmaxOperatorCode (Results 1 – 2 of 2) sorted by relevance
59 int32_t softmaxOperatorCode) in VisitSoftmaxOperator() argument90 switch(softmaxOperatorCode) in VisitSoftmaxOperator()119 switch(softmaxOperatorCode) in VisitSoftmaxOperator()
22 std::vector<char> CreateSoftmaxTfLiteModel(tflite::BuiltinOperator softmaxOperatorCode, in CreateSoftmaxTfLiteModel() argument52 switch (softmaxOperatorCode) in CreateSoftmaxTfLiteModel()101 void SoftmaxTest(tflite::BuiltinOperator softmaxOperatorCode, in SoftmaxTest() argument110 std::vector<char> modelBuffer = CreateSoftmaxTfLiteModel(softmaxOperatorCode, in SoftmaxTest()