Searched defs:ArgMaxWithValue (Results 1 – 7 of 7) sorted by relevance
17 from mindspore.ops.function.array_func import arg_max_with_value_ as ArgMaxWithValue unknown
25 class ArgMaxWithValue(nn.Cell): class
24 class ArgMaxWithValue(nn.Cell): class
147 class ArgMaxWithValue(Cell): class
179 ArgMaxWithValue() : BaseOperator(kNameArgMaxWithValue) {} in ArgMaxWithValue() function
197 "ArgMaxWithValue": { object
4332 [ArgMaxWithValue] section4333 opInfo.subTypeOfInferShape=14334 opInfo.opsFlag=OPS_FLAG_CLOSE4335 opInfo.engine=DNN_VM_AICPU4336 opInfo.flagPartial=False4337 opInfo.computeCost=1004338 opInfo.flagAsync=False4339 opInfo.opKernelLib=CUSTAICPUKernel4340 opInfo.formatAgnostic=False4341 opInfo.userDefined=True[all …]