You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
mindspore-ci-bot 0edc6d254a !370 Gpu Support UnsortedSegmentSum kernel 5 years ago
..
argmax_gpu_kernel.cc initial version 6 years ago
argmax_gpu_kernel.h initial version 6 years ago
array_reduce_gpu_kernel.cc initial version 6 years ago
array_reduce_gpu_kernel.h GPU kernels adapt with special dimensions. 5 years ago
concatv2_gpu_kernel.cc gpu concat kernel support 4 inputs 6 years ago
concatv2_gpu_kernel.h gpu concat kernel support 4 inputs 6 years ago
gather_gpu_kernel.cc initial version 6 years ago
gather_gpu_kernel.h initial version 6 years ago
one_hot_gpu_kernel.cc initial version 6 years ago
one_hot_gpu_kernel.h initial version 6 years ago
select_gpu_kernel.cc gpu add kernel select 6 years ago
select_gpu_kernel.h gpu add kernel select 6 years ago
slice_gpu_kernel.cc initial version 6 years ago
slice_gpu_kernel.h !324 Gpu Slice kernel performance improve 5 years ago
slice_grad_gpu_kernel.cc initial version 6 years ago
slice_grad_gpu_kernel.h initial version 6 years ago
transpose_gpu_kernel.cc initial version 6 years ago
transpose_gpu_kernel.h gpu dynamic memory pool supports multi-allReduce 6 years ago
unsorted_segment_sum_gpu_kernel.cc gpu support UnsortedSegmentSum kernel 5 years ago
unsorted_segment_sum_gpu_kernel.h gpu support UnsortedSegmentSum kernel 5 years ago