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.
 
 
 
 
 
 
i-robot e14942017d
!31673 support tranpose with neg perm
4 years ago
..
addn_cpu_kernel.cc clean codecheck warning for ignoring return value 4 years ago
addn_cpu_kernel.h refactor cpu factory 4 years ago
batch_norm_cpu_kernel.cc support tranpose with neg perm 4 years ago
batch_norm_cpu_kernel.h cpu kernel registerd 4 years ago
batch_norm_grad_cpu_kernel.cc cpu kernel registerd 4 years ago
batch_norm_grad_cpu_kernel.h cpu kernel registerd 4 years ago
conv_cpu_kernel.cc cpu kernel registerd 4 years ago
conv_cpu_kernel.h cpu kernel registerd 4 years ago
conv_grad_filter_cpu_kernel.cc refactor cpu factory 4 years ago
conv_grad_filter_cpu_kernel.h refactor cpu factory 4 years ago
conv_grad_input_cpu_kernel.cc refactor cpu factory 4 years ago
conv_grad_input_cpu_kernel.h refactor cpu factory 4 years ago
eltwise_cpu_kernel.cc cpu kernel registerd 4 years ago
eltwise_cpu_kernel.h cpu kernel registerd 4 years ago
log_softmax_cpu_kernel.cc cpu kernel registerd 4 years ago
log_softmax_cpu_kernel.h cpu kernel registerd 4 years ago
log_softmax_grad_cpu_kernel.cc cpu kernel registerd 4 years ago
log_softmax_grad_cpu_kernel.h cpu kernel registerd 4 years ago
lstm_cpu_kernel.cc support tranpose with neg perm 4 years ago
lstm_cpu_kernel.h develop nllloss and nlllossgrad for CPU and lite 4 years ago
lstm_grad_cpu_kernel.cc cpu kernel registerd 4 years ago
lstm_grad_cpu_kernel.h cpu kernel registerd 4 years ago
matmul_cpu_kernel_func.cc refactor cpu factory 4 years ago
matmul_cpu_kernel_func.h refactor cpu factory 4 years ago
mkl_cpu_kernel.cc develop nllloss and nlllossgrad for CPU and lite 4 years ago
mkl_cpu_kernel.h support tranpose with neg perm 4 years ago
pooling_cpu_kernel.cc develop nllloss and nlllossgrad for CPU and lite 4 years ago
pooling_cpu_kernel.h develop nllloss and nlllossgrad for CPU and lite 4 years ago
pooling_grad_cpu_kernel.cc develop nllloss and nlllossgrad for CPU and lite 4 years ago
pooling_grad_cpu_kernel.h develop nllloss and nlllossgrad for CPU and lite 4 years ago
softmax_cpu_kernel.cc cpu kernel registerd 4 years ago
softmax_cpu_kernel.h cpu kernel registerd 4 years ago
softmax_cross_entropy_with_logits_cpu_kernel.cc cpu kernel registerd 4 years ago
softmax_cross_entropy_with_logits_cpu_kernel.h cpu kernel registerd 4 years ago
sparse_softmax_cross_entropy_with_logits_cpu_kernel.cc cpu kernel registerd 4 years ago
sparse_softmax_cross_entropy_with_logits_cpu_kernel.h cpu kernel registerd 4 years ago