Searched defs:ArgMax (Results 1 – 3 of 3) sorted by relevance
24 class ArgMax(nn.Cell): class
158 class ArgMax(Cell): class
246 [ArgMax] section247 opInfo.subTypeOfInferShape=1248 opInfo.opsFlag=OPS_FLAG_CLOSE249 opInfo.engine=DNN_VM_AICPU250 opInfo.flagPartial=False251 opInfo.computeCost=100252 opInfo.flagAsync=False253 opInfo.opKernelLib=CUSTAICPUKernel254 opInfo.formatAgnostic=False255 opInfo.kernelSo=libcust_cpu_kernels.so[all …]