Searched refs:kSoftmax (Results 1 – 8 of 8) sorted by relevance
46 type == OperatorType::kLogistic || type == OperatorType::kSoftmax || in SupportsQuantization()296 if (op.type == OperatorType::kLogistic || op.type == OperatorType::kSoftmax) { in ChooseHardcodedQuantizationForOperatorOutput()
469 case OperatorType::kSoftmax: in Run()
2089 case OperatorType::kSoftmax: in Run()
78 kSoftmax, enumerator1662 SoftmaxOperator() : Operator(OperatorType::kSoftmax) {}
1918 case OperatorType::kSoftmax: in EstimateArithmeticOpsCount()
2152 } else if (src_op.type == OperatorType::kSoftmax) { in ConvertOperator()
319 GetOperator("SOFTMAX", OperatorType::kSoftmax), op); in TEST_F()
2392 OperatorType::kSoftmax)); in BuildOperatorList()