Searched refs:UniformInt (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | random_op_impl.cu | 69 bool UniformInt(int seed, int seed2, curandState *globalState, T *input1, size_t input_size_1, in UniformInt() function
107 template bool UniformInt<float>(int seed, int seed2, curandState *globalState, float *input1, size_…
110 template bool UniformInt<int>(int seed, int seed2, curandState *globalState, int *input1, size_t in…
|
| D | random_op_impl.cuh | 28 bool UniformInt(int seed, int seed2, curandState *globalState,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | random_op_gpu_kernel.cc | 23 MS_REG_GPU_KERNEL_ONE(UniformInt,
|
| D | random_op_gpu_kernel.h | 84 …bool ret = UniformInt(seed_, seed2_, devStates, input_addr_1, inputs[1]->size / sizeof(T), input_a… in Launch()
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_uniform_int.py | 28 self.uniformint = P.UniformInt(seed=seed)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | random_cpu_kernel.h | 50 MS_REG_CPU_KERNEL(UniformInt,
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_uniform_int.py | 29 self.uniformint = P.UniformInt(seed=seed)
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_uniform_int.py | 34 self.uniformint = P.UniformInt(seed, seed2)
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | random_ops.py | 194 random_uniform = P.UniformInt(seed1, seed2)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 64 from .random_ops import (RandomChoiceWithMask, StandardNormal, Gamma, Poisson, UniformInt, UniformR…
|
| D | random_ops.py | 269 class UniformInt(PrimitiveWithInfer): class
|
| /third_party/skia/third_party/externals/abseil-cpp/absl/random/ |
| D | distributions_test.cc | 303 TEST_F(RandomDistributionsTest, UniformInt) { in TEST_F() argument
|
| /third_party/abseil-cpp/absl/random/ |
| D | distributions_test.cc | 293 TEST_F(RandomDistributionsTest, UniformInt) { in TEST_F() argument
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_random.py | 311 self.uniform_int = P.UniformInt(seed, seed2)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 48 {"op_name": "UniformInt", "inputs": [{"index": 0, "name": "shape", "param_type": "required"}, {"ind…
|