Searched defs:RandomCategorical (Results 1 – 4 of 4) sorted by relevance
35 RandomCategorical() : BaseOperator(kNameRandomCategorical) { in RandomCategorical() function
21 __global__ void RandomCategorical(const size_t num_samples, double **dev_rand, double **dev_cdf, in RandomCategorical() function
836 class RandomCategorical(PrimitiveWithInfer): class
5830 [RandomCategorical] section5831 opInfo.subTypeOfInferShape=15832 opInfo.opsFlag=OPS_FLAG_CLOSE5833 opInfo.engine=DNN_VM_AICPU5834 opInfo.flagPartial=False5835 opInfo.computeCost=1005836 opInfo.flagAsync=False5837 opInfo.opKernelLib=CUSTAICPUKernel5838 opInfo.formatAgnostic=False5839 opInfo.userDefined=True[all …]