Searched defs:rand_state (Results 1 – 10 of 10) sorted by relevance
| /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()
|
| 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()
|
| 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()
|
| /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
|
| D | md_iteration_leap_frog_liujian_gpu_kernel.h | 62 auto rand_state = GetDeviceAddress<float>(inputs, 6); in Launch() local
|
| D | md_iteration_leap_frog_liujian_gpu_with_max_vel_kernel.h | 63 auto rand_state = GetDeviceAddress<float>(inputs, 6); in Launch() local
|
| /third_party/ffmpeg/libavcodec/ |
| D | elbg.c | 57 AVLFG *rand_state; member
339 AVLFG *rand_state) in avpriv_init_elbg()
373 AVLFG *rand_state) in avpriv_do_elbg()
|
| /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()
293 …tic void Setup_Rand_Normal_Kernel(const int float4_numbers, curandStatePhilox4_32_10_t *rand_state, 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
654 … def infer_dtype(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
|
| D | sponge_ops.py | 2795 … def infer_shape(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
2820 … def infer_dtype(self, inverse_mass, sqrt_mass_inverse, vel, crd, frc, acc, rand_state, rand_frc): argument
|