| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_uniform_candidate_sampler_op.py | 25 def __init__(self, num_true, num_sampled, unique, range_max): argument
34 def uniform_candidate_sampler(x, num_true, num_sampled, unique, range_max): argument
42 def uniform_candidate_sampler_int64(x, num_true, num_sampled, unique, range_max): argument
52 def __init__(self, num_true, num_sampled, unique, range_max, seed, remove_accidental_hits): argument
62 def uniform_candidate_sampler_hit(x, num_true, num_sampled, unique, range_max, seed, argument
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_uniform_candidate_sampler.py | 27 …def __init__(self, embedding_weight, num_true, num_sampled, unique, range_max, seed, remove_accide… argument
52 … def __init__(self, mul_weight, num_true, num_sampled, unique, range_max, seed, remove_accidential, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | uniform_candidate_sampler_impl.cu | 27 void CalUniformCandidateSampler(const int64_t true_size, const int64_t num_sampled, const S prob_va… in CalUniformCandidateSampler()
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_log_uniform_candidate_sampler.py | 26 def __init__(self, num_true=1, num_sampled=5, unique=True, range_max=5, seed=0): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | random_ops.py | 625 …def __init__(self, num_true, num_sampled, unique, range_max, seed=0, remove_accidental_hits=False): argument
699 def __init__(self, num_true=1, num_sampled=5, unique=True, range_max=5, seed=0): argument
|
| /third_party/skia/third_party/externals/abseil-cpp/absl/container/internal/ |
| D | hashtablez_sampler_test.cc | 219 int64_t num_sampled = 0; in TEST() local
|
| /third_party/abseil-cpp/absl/container/internal/ |
| D | hashtablez_sampler_test.cc | 213 int64_t num_sampled = 0; in TEST() local
|
| /third_party/mindspore/mindspore/nn/loss/ |
| D | loss.py | 808 def __init__(self, num_sampled, num_classes, num_true=1, argument
|