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.
 
 
 
 
 
 
fanjibin b6acbd1055 unified cpu thread pool 4 years ago
..
addn_cpu_kernel.cc unified cpu thread pool 4 years ago
addn_cpu_kernel.h Support float64 for CPU OP, including Sqrt, Abs, AbsGrad, AddN. 4 years ago
assignadd_cpu_kernel.cc code check for cpu ops 4 years ago
assignadd_cpu_kernel.h add AddN,AssignAdd,Conv2d,BN,Relu6,TensorAdd CPU operator 5 years ago
batch_norm_cpu_kernel.cc code check for cpu ops 4 years ago
batch_norm_cpu_kernel.h IR operators of GPU and CPU are unified as batchnorm 5 years ago
batch_norm_grad_cpu_kernel.cc !24671 Enable CPU BatchNorm grad in eval mode 4 years ago
batch_norm_grad_cpu_kernel.h clean code in cpu operater 4 years ago
conv2d_grad_filter_cpu_kernel.cc optimize the comment and log description 4 years ago
conv2d_grad_filter_cpu_kernel.h code check for cpu ops 4 years ago
conv2d_grad_input_cpu_kernel.cc optimize the comment and log description 4 years ago
conv2d_grad_input_cpu_kernel.h code check for cpu ops 4 years ago
conv_cpu_kernel.cc code check for cpu ops 4 years ago
conv_cpu_kernel.h code check for cpu ops 4 years ago
eltwise_cpu_kernel.cc code check fix 4 years ago
eltwise_cpu_kernel.h code check fix 4 years ago
log_softmax_cpu_kernel.cc code check fix 4 years ago
log_softmax_cpu_kernel.h code check fix 4 years ago
log_softmax_grad_cpu_kernel.cc code check fix 4 years ago
log_softmax_grad_cpu_kernel.h code check fix 4 years ago
lstm_cpu_kernel.cc imporve performance of CPU LSTMGrad 4 years ago
lstm_cpu_kernel.h imporve performance of CPU LSTMGrad 4 years ago
lstm_grad_cpu_kernel.cc code check fix 4 years ago
lstm_grad_cpu_kernel.h code check fix 4 years ago
matmul_cpu_kernel.cc remove unused code 4 years ago
matmul_cpu_kernel.h remove unused code 4 years ago
mkl_cpu_kernel.cc code check fix 4 years ago
mkl_cpu_kernel.h code check fix 4 years ago
mkl_kernel_engine.cc code check fix 4 years ago
mkl_kernel_engine.h code check fix 4 years ago
pooling_avg_grad_cpu_kernel.cc sync cpu op review on master 4 years ago
pooling_avg_grad_cpu_kernel.h code check for cpu ops 4 years ago
pooling_cpu_kernel.cc sync cpu op review on master 4 years ago
pooling_cpu_kernel.h sync cpu op review on master 4 years ago
pooling_max_grad_cpu_kernel.cc sync cpu op review on master 4 years ago
pooling_max_grad_cpu_kernel.h sync cpu op review on master 4 years ago
softmax_cpu_kernel.cc sync cpu op review on master 4 years ago
softmax_cpu_kernel.h sync cpu op review on master 4 years ago
softmax_cross_entropy_with_logits_cpu_kernel.cc refactor include files 4 years ago
softmax_cross_entropy_with_logits_cpu_kernel.h sync cpu op review on master 4 years ago
sparse_softmax_cross_entropy_with_logits_cpu_kernel.cc sync cpu op review on master 4 years ago
sparse_softmax_cross_entropy_with_logits_cpu_kernel.h sync cpu op review on master 4 years ago