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
..
addn_gpu_kernel.cc initial version 6 years ago
addn_gpu_kernel.h initial version 6 years ago
assign_add_gpu_kernel.cc initial version 6 years ago
assign_add_gpu_kernel.h initial version 6 years ago
bias_add_gpu_kernel.cc initial version 6 years ago
bias_add_gpu_kernel.h remove make_unique.h 6 years ago
broadcast_gpu_kernel.cc gpu support broadcast kernels 5 years ago
broadcast_gpu_kernel.h gpu support broadcast kernels 5 years ago
broadcast_grad_gpu_kernel.cc gpu support MinimumGrad & MaximumGrad kernel 5 years ago
broadcast_grad_gpu_kernel.h gpu support MinimumGrad & MaximumGrad kernel 5 years ago
equalcount_gpu_kernel.cc initial version 6 years ago
equalcount_gpu_kernel.h initial version 6 years ago
float_status_gpu_kernel.cc gpu add float_status kernel 5 years ago
float_status_gpu_kernel.h gpu add float_status kernel 5 years ago
matmul_gpu_kernel.cc gpu support batchmatmul kernel 5 years ago
matmul_gpu_kernel.h matmul support fp16 5 years ago
tensoradd_gpu_kernel.cc fix cudnn type error 5 years ago
tensoradd_gpu_kernel.h GPU kernels adapt with special dimensions. 5 years ago
unary_op_gpu_kernel.cc gpu queue support unary 5 years ago
unary_op_gpu_kernel.h GPU kernels adapt with special dimensions. 5 years ago