Searched refs:kSoftmax (Results 1 – 8 of 8) sorted by relevance
46 type == OperatorType::kLogistic || type == OperatorType::kSoftmax || in SupportsQuantization()280 (op.type == OperatorType::kSoftmax)) { in ChooseHardcodedQuantizationForOperatorOutput()
316 case OperatorType::kSoftmax: in Run()
1352 case OperatorType::kSoftmax: in Run()
66 kSoftmax, enumerator1268 SoftmaxOperator() : Operator(OperatorType::kSoftmax) {}
1524 case OperatorType::kSoftmax: in EstimateArithmeticOpsCount()
1648 } else if (src_op.type == OperatorType::kSoftmax) { in ConvertOperator()
244 GetOperator("SOFTMAX", OperatorType::kSoftmax), op); in TEST_F()
809 new Softmax(::tflite::BuiltinOperator_SOFTMAX, OperatorType::kSoftmax)); in BuildOperatorList()