| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/nvtit/ |
| D | md_iteration_setup_random_state_gpu_impl.cu | 21 void MD_Iteration_Setup_Random_State(int float4_numbers, curandStatePhilox4_32_10_t *rand_state, in… in MD_Iteration_Setup_Random_State() argument
24 … rand_state, seed); in MD_Iteration_Setup_Random_State()
27 void MD_Iteration_Setup_Random_State(int float4_numbers, curandStatePhilox4_32_10_t *rand_state, in…
|
| D | md_iteration_leap_frog_liujian_gpu_impl.cu | 54 … curandStatePhilox4_32_10_t *rand_state, float *rand_frc, float *output, in MD_Iteration_Leap_Frog_With_LiuJian() argument
56 …mal<<<ceilf(static_cast<float>(float4_numbers) / 32.), 32, 0, stream>>>(float4_numbers, rand_state, in MD_Iteration_Leap_Frog_With_LiuJian()
|
| D | md_iteration_setup_random_state_gpu_impl.cuh | 21 void MD_Iteration_Setup_Random_State(int float4_numbers, curandStatePhilox4_32_10_t *rand_state, in…
|
| D | md_iteration_leap_frog_liujian_with_max_vel_impl.cu | 65 … float *acc, curandStatePhilox4_32_10_t *rand_state, in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel() argument
68 …mal<<<ceilf(static_cast<float>(float4_numbers) / 32.), 32, 0, stream>>>(float4_numbers, rand_state, in MD_Iteration_Leap_Frog_With_LiuJian_With_Max_Vel()
|
| D | md_iteration_leap_frog_liujian_gpu_impl.cuh | 25 … curandStatePhilox4_32_10_t *rand_state, float *rand_frc, float *output,
|
| D | md_iteration_leap_frog_liujian_with_max_vel_impl.cuh | 25 … float *acc, curandStatePhilox4_32_10_t *rand_state,
|
| /third_party/ffmpeg/libavcodec/ |
| D | elbg.c | 57 AVLFG *rand_state; member
114 r = av_lfg_get(elbg->rand_state) % (unsigned int)elbg->utility_inc[elbg->numCB-1] + 1; in get_high_utility_cell()
116 r = av_lfg_get(elbg->rand_state); in get_high_utility_cell()
117 r = (av_lfg_get(elbg->rand_state) + (r<<32)) % elbg->utility_inc[elbg->numCB-1] + 1; in get_high_utility_cell()
339 AVLFG *rand_state) in avpriv_init_elbg() argument
355 numCB, 2 * max_steps, closest_cb, rand_state); in avpriv_init_elbg()
361 numCB, 2 * max_steps, closest_cb, rand_state); in avpriv_init_elbg()
373 AVLFG *rand_state) in avpriv_do_elbg() argument
402 elbg->rand_state = rand_state; in avpriv_do_elbg()
|
| D | elbg.h | 43 AVLFG *rand_state);
55 AVLFG *rand_state);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/nvtit/ |
| D | md_iteration_setup_random_state.h | 55 curandStatePhilox4_32_10_t *rand_state = reinterpret_cast<curandStatePhilox4_32_10_t *>(output); in Launch() local
56 …MD_Iteration_Setup_Random_State(float4_numbers, rand_state, seed, reinterpret_cast<cudaStream_t>(s… in Launch()
|
| D | md_iteration_leap_frog_liujian_gpu_with_max_vel_kernel.h | 63 auto rand_state = GetDeviceAddress<float>(inputs, 6); in Launch() local
70 reinterpret_cast<curandStatePhilox4_32_10_t *>(rand_state), rand_frc, output, max_vel, in Launch()
|
| D | md_iteration_leap_frog_liujian_gpu_kernel.h | 62 auto rand_state = GetDeviceAddress<float>(inputs, 6); in Launch() local
69 … reinterpret_cast<curandStatePhilox4_32_10_t *>(rand_state), rand_frc, output, in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/ |
| D | common_sponge.cuh | 285 __global__ static void Rand_Normal(const int float4_numbers, curandStatePhilox4_32_10_t *rand_state, in Rand_Normal() argument
289 rand_float4[i] = curand_normal4(&rand_state[i]); in Rand_Normal()
293 …tic void Setup_Rand_Normal_Kernel(const int float4_numbers, curandStatePhilox4_32_10_t *rand_state, in Setup_Rand_Normal_Kernel() argument
299 curand_init(seed, id, 0, &rand_state[id]); in Setup_Rand_Normal_Kernel()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | sponge_update_ops.py | 629 … def infer_shape(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
633 validator.check_int(len(rand_state), 1, Rel.EQ, "rand_state_dim", self.name)
651 …validator.check_int(rand_state[0], math.ceil(self.atom_numbers * 3 / 4.0) * 16, Rel.EQ, "rand_stat…
654 … def infer_dtype(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
662 validator.check_tensor_dtype_valid('rand_state', rand_state, [mstype.float32], self.name)
|
| D | sponge_ops.py | 2795 … def infer_shape(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
2801 validator.check_int(len(rand_state), 1, Rel.EQ, "rand_state_dim", self.name)
2817 …validator.check_int(rand_state[0], math.ceil(self.atom_numbers * 3 / 4.0) * 16, Rel.EQ, "rand_stat…
2820 … def infer_dtype(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
2828 validator.check_tensor_dtype_valid('rand_state', rand_state, [mstype.float32], self.name)
|