Searched refs:kArgMax (Results 1 – 9 of 9) sorted by relevance
464 static const int kArgMax = (1 << kArgBits) - 1; variable469 CHECK_LE(argc, kArgMax); in CallApiCallbackStub()
110 case OperatorType::kArgMax: { in Run()
63 type == OperatorType::kArgMax || type == OperatorType::kRelu || in SupportsQuantization()
2304 case OperatorType::kArgMax: in Run()
98 kArgMax, enumerator1757 ArgMaxOperator() : Operator(OperatorType::kArgMax) {}
2274 } else if (src_op.type == OperatorType::kArgMax) { in ConvertOperator()
486 GetOperator("ARG_MAX", OperatorType::kArgMax), op); in TEST_F()
2432 OperatorType::kArgMax)); in BuildOperatorList()
2888 if (argc > CallApiCallbackStub::kArgMax) return NoChange(); in ReduceCallApiFunction()