Home
last modified time | relevance | path

Searched refs:MS_REG_GPU_KERNEL_THREE (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/random/
Drandom_categorical_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
28 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
35 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
42 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
49 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
56 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
63 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
70 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
77 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
84 MS_REG_GPU_KERNEL_THREE(RandomCategorical,
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dsync_batch_norm_grad_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
32 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
43 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
54 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
65 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
76 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
87 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
98 MS_REG_GPU_KERNEL_THREE(SyncBatchNormGrad,
Dsync_batch_norm_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
34 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
47 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
60 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
73 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
86 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
99 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
112 MS_REG_GPU_KERNEL_THREE(SyncBatchNorm,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dadagrad_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(ApplyAdagrad,
30 MS_REG_GPU_KERNEL_THREE(ApplyAdagrad,
39 MS_REG_GPU_KERNEL_THREE(ApplyAdagrad,
48 MS_REG_GPU_KERNEL_THREE(ApplyAdagrad,
57 MS_REG_GPU_KERNEL_THREE(ApplyAdagrad,
66 MS_REG_GPU_KERNEL_THREE(ApplyAdagrad,
Dmomentum_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(ApplyMomentum,
30 MS_REG_GPU_KERNEL_THREE(ApplyMomentum,
39 MS_REG_GPU_KERNEL_THREE(ApplyMomentum,
48 MS_REG_GPU_KERNEL_THREE(ApplyMomentum,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dbroadcast_complex_gpu_kernel.cc22 …MS_REG_GPU_KERNEL_THREE(OPNAME, …
25 …MS_REG_GPU_KERNEL_THREE(OPNAME, …
28 …MS_REG_GPU_KERNEL_THREE(OPNAME, …
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/bond/
Dbond_energy_cuda_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(BondEnergy,
Dbond_force_cuda_gpu_kernel.cc21 MS_REG_GPU_KERNEL_THREE(BondForce,
Dbond_force_with_atom_energy_and_virial_kernel.cc21 MS_REG_GPU_KERNEL_THREE(BondForceWithAtomEnergyAndVirial,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/simple_constrain/
Dconstrain_force_cycle_kernel.cc21 MS_REG_GPU_KERNEL_THREE(ConstrainForceCycle,
Dconstrain_force_cycle_with_virial_kernel.cc25 MS_REG_GPU_KERNEL_THREE(ConstrainForceCycleWithVirial,
Dconstrain_force_virial_kernel.cc25 MS_REG_GPU_KERNEL_THREE(ConstrainForceVirial,
Dconstrain_force_kernel.cc24 MS_REG_GPU_KERNEL_THREE(ConstrainForce,
Dconstrain_kernel.cc24 MS_REG_GPU_KERNEL_THREE(Constrain,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/lj/
Dlj_direct_cf_force_with_lj_virial_direct_cf_energy_kernel.cc24 MS_REG_GPU_KERNEL_THREE(LJForceWithVirialEnergy,
Dlj_force_with_virial_energy_update_kernel.cc24 MS_REG_GPU_KERNEL_THREE(LJForceWithVirialEnergyUpdate,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel_factory.h112 #define MS_REG_GPU_KERNEL_THREE(OPNAME, ATTR, OPCLASS, T, S, G) … macro