Home
last modified time | relevance | path

Searched defs:RandomCategorical (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Drandom_categorical.h35 RandomCategorical() : BaseOperator(kNameRandomCategorical) { in RandomCategorical() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Drandom_categorical.cu21 __global__ void RandomCategorical(const size_t num_samples, double **dev_rand, double **dev_cdf, in RandomCategorical() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Drandom_ops.py836 class RandomCategorical(PrimitiveWithInfer): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini5830 [RandomCategorical] section
5831 opInfo.subTypeOfInferShape=1
5832 opInfo.opsFlag=OPS_FLAG_CLOSE
5833 opInfo.engine=DNN_VM_AICPU
5834 opInfo.flagPartial=False
5835 opInfo.computeCost=100
5836 opInfo.flagAsync=False
5837 opInfo.opKernelLib=CUSTAICPUKernel
5838 opInfo.formatAgnostic=False
5839 opInfo.userDefined=True
[all …]