Searched refs:OpType_ArgMax (Results 1 – 13 of 13) sorted by relevance
19 return MNN::OpType_ArgMax; in opType()
15 return MNN::OpType_ArgMax; in opType()
34 CoreMLCreatorRegister<TypedCreator<CoreMLArgMax>> __argmax_op(OpType_ArgMax);
46 NPUCreatorRegister<TypedCreator<NPUArgMax>> __argmax_op(OpType_ArgMax);
27 ArgMaxOp->type = OpType_ArgMax; in onExecute()
97 REGISTER_SHAPE(ArgMaxComputer, OpType_ArgMax);
23 op->type = OpType_ArgMax; in run()
84 static CUDACreatorRegister<ArgMaxCreator> __init(OpType_ArgMax);
106 {"ArgMax", MNN::OpType_ArgMax},
281 REGISTER_CPU_OP_CREATOR(CPUArgMaxCreator, OpType_ArgMax);
365 if (MNN::OpType_ArgMax == op->type || MNN::OpType_ArgMin == op->type) { in onExecute()
1012 op->type = OpType_ArgMax; in _ArgMax()
91 OpType_ArgMax = 2, enumerator258 OpType_ArgMax, in EnumValuesOpType()