Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dlog_uniform_candidate_sampler.h38 LogUniformCandidateSampler() : BaseOperator(kNameLogUniformCandidateSampler) { in LogUniformCandidateSampler() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Drandom_ops.py1081 class LogUniformCandidateSampler(Primitive): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Drandom_ops_proto.cc211 CUST_IMPLEMT_INFERFUNC(LogUniformCandidateSampler, LogUniformCandidateSamplerInfer) { 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.ini1809 [LogUniformCandidateSampler] section
1810 opInfo.subTypeOfInferShape = 1
1811 opInfo.opsFlag = OPS_FLAG_CLOSE
1812 opInfo.engine = DNN_VM_AICPU
1813 opInfo.flagPartial = False
1814 opInfo.computeCost = 100
1815 opInfo.flagAsync = False
1816 opInfo.opKernelLib = CUSTAICPUKernel
1817 opInfo.formatAgnostic = False
1818 opInfo.userDefined = True
[all …]