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.
 
 
 
 
 
 
wilfChen 59c4cf256c gpu support broadcast kernels 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 5 years ago
concatv2_gpu_kernel.h gpu concat kernel support 4 inputs 5 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 5 years ago
select_gpu_kernel.h gpu add kernel select 5 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 5 years ago
unsorted_segment_sum_gpu_kernel.cc gpu support UnsortedSegmentSum kernel 5 years ago
unsorted_segment_sum_gpu_kernel.h gpu support broadcast kernels 5 years ago