Searched defs:true_classes (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_uniform_candidate_sampler.py | 27 def __init__(self, true_classes, num_true, num_sampled, unique, range_max, seed): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | uniform_candidate_sampler.cc | 87 T *true_classes = reinterpret_cast<T *>(ctx.Input(0)->GetData()); in DoCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | log_uniform_candidate_sampler_cpu_kernel.cc | 113 int64_t *true_classes = static_cast<int64_t *>(inputs.at(0)->device_ptr()); in Launch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | random_func.py | 409 def uniform_candidate_sampler(true_classes, argument
579 def log_uniform_candidate_sampler(true_classes, num_true=1, num_sampled=5, unique=True, range_max=5… argument
|