| /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
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()
56 int64_t num_sampled = GetValue<int64_t>(primitive->GetAttr(kNumSampled)); in InferShape() local
|
| D | uniform_candidate_sampler.cc | 89 int64_t num_sampled = GetValue<int64_t>(primitive->GetAttr("num_sampled")); in UCSInferShape() local
138 void UniformCandidateSampler::Init(int64_t num_true, int64_t num_sampled, bool unique, int64_t rang… in Init()
151 void UniformCandidateSampler::set_num_sampled(int64_t num_sampled) { in set_num_sampled()
|
| D | log_uniform_candidate_sampler.h | 57 …inline void set_num_sampled(int64_t num_sampled) { (void)this->AddAttr(kNumSampled, api::MakeValue… in set_num_sampled()
|
| /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/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | uniform_candidate_sampler_impl.cu | 27 cudaError_t CalUniformCandidateSampler(const int64_t true_size, const int64_t num_sampled, const S … in CalUniformCandidateSampler()
|
| /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
|
| 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/ccsrc/plugin/device/cpu/kernel/ |
| D | log_uniform_candidate_sampler_cpu_kernel.cc | 55 static float CalcExpectedCount(float p, int num_sampled, int num_tries) { in CalcExpectedCount()
|
| /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
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | random_func.py | 411 num_sampled, argument
579 def log_uniform_candidate_sampler(true_classes, num_true=1, num_sampled=5, unique=True, range_max=5… 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/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | random_ops_proto.cc | 213 int64_t num_sampled; in CUST_IMPLEMT_INFERFUNC() local
|
| /third_party/protobuf/src/google/protobuf/ |
| D | arenaz_sampler_test.cc | 235 int64_t num_sampled = 0; in TEST() 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
|