Home
last modified time | relevance | path

Searched defs:RandomPoisson (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Drandom_poisson.h33 RandomPoisson() : BaseOperator(kRandomPoisson) { InitIOName({"shape", "rate"}, {"output"}); } in RandomPoisson() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Drandom_op_impl.cu131 cudaError_t RandomPoisson(uint64_t seed, uint64_t seed_offset, curandState *globalState, R *rate, i… in RandomPoisson() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Drandom_ops.py591 class RandomPoisson(Primitive): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Drandom_ops_proto.cc336 CUST_IMPLEMT_INFERFUNC(RandomPoisson, RandomPoissonInfer) { in CUST_IMPLEMT_INFERFUNC() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini5114 [RandomPoisson] section
5115 opInfo.subTypeOfInferShape = 1
5116 opInfo.opsFlag = OPS_FLAG_CLOSE
5117 opInfo.engine = DNN_VM_AICPU
5118 opInfo.flagPartial = False
5119 opInfo.computeCost = 100
5120 opInfo.flagAsync = False
5121 opInfo.opKernelLib = CUSTAICPUKernel
5122 opInfo.formatAgnostic = False
5123 opInfo.userDefined = True
[all …]