mindspore2022/mindspore/ccsrc/kernel/gpu/arrays
limingqi107 2891f0d20d gpu dynamic memory pool supports multi-allReduce 2020-04-22 15:43:45 +08:00
..
argmax_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
argmax_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
array_reduce_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
array_reduce_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
concatv2_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
concatv2_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
gather_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
gather_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
one_hot_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
one_hot_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
select_gpu_kernel.cc gpu add kernel select 2020-04-14 20:03:33 +08:00
select_gpu_kernel.h gpu add kernel select 2020-04-14 20:03:33 +08:00
slice_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
slice_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
slice_grad_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
slice_grad_gpu_kernel.h initial version 2020-03-27 22:54:54 +08:00
transpose_gpu_kernel.cc initial version 2020-03-27 22:54:54 +08:00
transpose_gpu_kernel.h gpu dynamic memory pool supports multi-allReduce 2020-04-22 15:43:45 +08:00