| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/tests/st/ops/cpu/ |
| 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
|
| D | test_candidate_sampler_op.py | 29 def __init__(self, num_true, num_sampled, unique, range_max): argument
37 def uniform_candidate_sampler(x, num_true, num_sampled, unique, range_max): argument
43 def uniform_candidate_sampler_functional(x, num_true, num_sample, unique, range_max): argument
48 def uniform_candidate_sampler_int64(x, num_true, num_sampled, unique, range_max): argument
55 def __init__(self, num_true, num_sampled, unique, range_max, seed, remove_accidental_hits): argument
68 def uniform_candidate_sampler_hit(x, num_true, num_sampled, unique, range_max, seed, remove_acciden… argument
294 def uniform_candidate_sampler_vmap(x, num_true, num_sampled, unique, range_max, in_axes, out_axes=0… argument
334 def uniform_candidate_sampler_vmap2_int64(x, num_true, num_sampled, unique, range_max, in_axes, out… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | log_uniform_candidate_sampler.cc | 42 void LogUniformCandidateSampler::Init(int64_t num_true, int64_t num_sampled, bool unique, int64_t r… in Init()
65 auto num_true = GetValue<int64_t>(primitive->GetAttr(kNumTrue)); in InferShape() local
|
| D | uniform_candidate_sampler.cc | 86 int64_t num_true = GetValue<int64_t>(primitive->GetAttr("num_true")); in UCSInferShape() local
138 void UniformCandidateSampler::Init(int64_t num_true, int64_t num_sampled, bool unique, int64_t rang… in Init()
148 void UniformCandidateSampler::set_num_true(int64_t num_true) { in set_num_true()
|
| D | compute_accidental_hits.cc | 62 const auto num_true = GetValue<int64_t>(num_true_ptr); in ComputeAccidentalHitsInferShape() local
116 const auto num_true = GetValue<int64_t>(num_true_ptr); in ComputeAccidentalHitsInfer() local
|
| D | log_uniform_candidate_sampler.h | 53 …inline void set_num_true(int64_t num_true) { (void)this->AddAttr(kNumTrue, api::MakeValue(num_true… in set_num_true()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_uniform_candidate_sampler.py | 31 …def __init__(self, embedding_weight, num_true, num_sampled, unique, range_max, seed, remove_accide… argument
56 … def __init__(self, mul_weight, num_true, num_sampled, unique, range_max, seed, remove_accidential, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_compute_accidental_hits.py | 26 def __init__(self, num_true=1): argument
|
| D | test_uniform_candidate_sampler.py | 27 def __init__(self, true_classes, num_true, num_sampled, unique, range_max, seed): argument
|
| 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-src/source/mindspore/python/mindspore/ops/operations/ |
| D | random_ops.py | 1056 …def __init__(self, num_true, num_sampled, unique, range_max, seed=0, remove_accidental_hits=False): argument
1128 def __init__(self, num_true=1, num_sampled=5, unique=True, range_max=5, seed=0): argument
|
| D | nn_ops.py | 3522 def __init__(self, num_true=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | random_func.py | 410 num_true, argument
579 def log_uniform_candidate_sampler(true_classes, num_true=1, num_sampled=5, unique=True, range_max=5… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | random_ops_proto.cc | 228 int64_t num_true; in CUST_IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/loss/ |
| D | loss.py | 1044 def __init__(self, num_sampled, num_classes, num_true=1, argument
1130 num_true=1, argument
|