Searched defs:RandomChoiceWithMask (Results 1 – 4 of 4) sorted by relevance
32 …RandomChoiceWithMask() : BaseOperator(kRandomChoiceWithMask) { InitIOName({"input_x"}, {"y", "mask… in RandomChoiceWithMask() function
779 class RandomChoiceWithMask(Primitive): class
374 CUST_IMPLEMT_INFERFUNC(RandomChoiceWithMask, RandomChoiceWithMaskInfer) { in CUST_IMPLEMT_INFERFUNC() argument
5855 [RandomChoiceWithMask] section5856 opInfo.subTypeOfInferShape=15857 opInfo.opsFlag=OPS_FLAG_CLOSE5858 opInfo.engine=DNN_VM_AICPU5859 opInfo.flagPartial=False5860 opInfo.computeCost=1005861 opInfo.flagAsync=False5862 opInfo.opKernelLib=CUSTAICPUKernel5863 opInfo.formatAgnostic=False5864 opInfo.userDefined=True[all …]