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.
 
 
 
 
 
 
VectorSL 46afb18e25 update argmaxwithvalue 5 years ago
..
argmax_gpu_kernel.cc initial version 6 years ago
argmax_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
argmaxwithvalue_gpu_kernel.cc gpu add argmaxwithvalue 5 years ago
argmaxwithvalue_gpu_kernel.h update argmaxwithvalue 5 years ago
array_reduce_gpu_kernel.cc initial version 6 years ago
array_reduce_gpu_kernel.h gpu refactor cuda datatype 5 years ago
concatv2_gpu_kernel.cc gpu concat kernel support 4 inputs 5 years ago
concatv2_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
gather_gpu_kernel.cc initial version 6 years ago
gather_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
one_hot_gpu_kernel.cc initial version 6 years ago
one_hot_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
select_gpu_kernel.cc gpu add kernel select 5 years ago
select_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
slice_gpu_kernel.cc initial version 6 years ago
slice_gpu_kernel.h gpu slice support null output 5 years ago
slice_grad_gpu_kernel.cc initial version 6 years ago
slice_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
transpose_gpu_kernel.cc initial version 6 years ago
transpose_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
unsorted_segment_sum_gpu_kernel.cc LayerNormGrad fix & codex 5 years ago
unsorted_segment_sum_gpu_kernel.h Change uintptr_t to void ptr 5 years ago