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 1fd5efeb2b
!31324 Delete redundant AssignAdd register.
4 years ago
..
addn_cpu_kernel.cc refactor cpu factory 4 years ago
addn_cpu_kernel.h refactor cpu factory 4 years ago
batch_norm_cpu_kernel.cc cpu kernel registerd 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 cpu kernel registerd 4 years ago
lstm_cpu_kernel.h cpu kernel registerd 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 maxpool3d, avgpool3d and conv3d on CPU(including forward and backward) 4 years ago
mkl_cpu_kernel.h refactor cpu factory 4 years ago
pooling_cpu_kernel.cc implement Conv2dBackward and log of pool 4 years ago
pooling_cpu_kernel.h implement Conv2dBackward and log of pool 4 years ago
pooling_grad_cpu_kernel.cc implement Conv2dBackward and log of pool 4 years ago
pooling_grad_cpu_kernel.h implement Conv2dBackward and log of pool 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