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 ee7a64018c gpu update conv kernel for auto-mixed-precision 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
binary_op_gpu_kernel.cc update some ops 5 years ago
binary_op_gpu_kernel.h gpu update conv kernel for auto-mixed-precision 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 6 years ago
float_status_gpu_kernel.h gpu add float_status kernel 6 years ago
matmul_gpu_kernel.cc initial version 6 years ago
matmul_gpu_kernel.h initial version 6 years ago
tensoradd_gpu_kernel.cc fix cudnn type error 5 years ago
tensoradd_gpu_kernel.h fix cudnn type error 5 years ago
unary_op_gpu_kernel.cc update some ops 5 years ago
unary_op_gpu_kernel.h gpu dynamic memory pool supports multi-allReduce 5 years ago