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 8f2ec32d95 gpu support bert finetune 5 years ago
..
addn_gpu_kernel.cc initial version 6 years ago
addn_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
assign_add_gpu_kernel.cc initial version 6 years ago
assign_add_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
bias_add_gpu_kernel.cc initial version 6 years ago
bias_add_gpu_kernel.h MatMul & BiasAdd support null input 5 years ago
broadcast_gpu_kernel.cc gpu support bert finetune 5 years ago
broadcast_gpu_kernel.h gpu support bert finetune 5 years ago
broadcast_grad_gpu_kernel.cc gpu support MinimumGrad & MaximumGrad kernel 5 years ago
broadcast_grad_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
equalcount_gpu_kernel.cc initial version 6 years ago
equalcount_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
float_status_gpu_kernel.cc gpu add float_status kernel 5 years ago
float_status_gpu_kernel.h Change uintptr_t to void ptr 5 years ago
matmul_gpu_kernel.cc gpu support batchmatmul kernel 5 years ago
matmul_gpu_kernel.h MatMul & BiasAdd support null input 5 years ago
unary_op_gpu_kernel.cc gpu queue support unary 5 years ago
unary_op_gpu_kernel.h Change uintptr_t to void ptr 5 years ago