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.
 
 
 
 
 
 
yujianfeng 34407391e6 Add an output to apply_proximal_adagrad op register 5 years ago
..
mkldnn !2369 add cpu reduce op and cpu softmax_cross_entropy_with_logits op 5 years ago
addn_cpu_kernel.cc applymomentum 5 years ago
addn_cpu_kernel.h multi_input 5 years ago
allgather_cpu_kernel.cc support host reduce 5 years ago
allgather_cpu_kernel.h support host reduce 5 years ago
apply_momentum_cpu_kernel.cc change cpu kernel dir 5 years ago
apply_momentum_cpu_kernel.h applymomentum 5 years ago
argmax_cpu_kernel.cc change cpu kernel dir 5 years ago
argmax_cpu_kernel.h fix cpu argmax register 5 years ago
bias_add_cpu_kernel.cc change cpu kernel dir 5 years ago
bias_add_cpu_kernel.h cpu kernel register type 5 years ago
bias_add_grad_cpu_kernel.cc change cpu kernel dir 5 years ago
bias_add_grad_cpu_kernel.h cpu kernel register type 5 years ago
concat_cpu_kernel.cc review 5 years ago
concat_cpu_kernel.h multi_input 5 years ago
cpu_kernel.cc optimize reduce sparse gradient 5 years ago
cpu_kernel.h Add SparseAdam and SparseLazyAdam cpu kernel 5 years ago
cpu_kernel_factory.cc multi_input 5 years ago
cpu_kernel_factory.h multi_input 5 years ago
embedding_look_up_comm_grad_cpu_kernel.cc add cpu op embedding_look_up 5 years ago
embedding_look_up_comm_grad_cpu_kernel.h add cpu op embedding_look_up 5 years ago
embedding_look_up_cpu_kernel.cc fix review code 5 years ago
embedding_look_up_cpu_kernel.h add cpu op embedding_look_up 5 years ago
equal_count_cpu_kernel.cc change cpu kernel dir 5 years ago
equal_count_cpu_kernel.h cpu kernel register type 5 years ago
gather_cpu_kernel.cc sync cpu output 5 years ago
gather_cpu_kernel.h review 5 years ago
one_hot_cpu_kernel.cc change cpu kernel dir 5 years ago
one_hot_cpu_kernel.h cpu kernel register type 5 years ago
reduce_cpu_kernel.cc add reduce and softmax_cross_entropy_with_logits 5 years ago
reduce_cpu_kernel.h add reduce and softmax_cross_entropy_with_logits 5 years ago
reduce_scatter_cpu_kernel.cc support host reduce 5 years ago
reduce_scatter_cpu_kernel.h support host reduce 5 years ago
reshape_cpu_kernel.cc change cpu kernel dir 5 years ago
reshape_cpu_kernel.h cpu kernel support mutil dtype 5 years ago
slice_cpu_kernel.cc applymomentum 5 years ago
slice_cpu_kernel.h applymomentum 5 years ago
slice_grad_cpu_kernel.cc applymomentum 5 years ago
slice_grad_cpu_kernel.h applymomentum 5 years ago
sparse_apply_adam_cpu_kernel.cc Add multiple process for computation of sparse optimizers 5 years ago
sparse_apply_adam_cpu_kernel.h Add multiple process for computation of sparse optimizers 5 years ago
sparse_apply_ftrl_cpu_kernel.cc Add multiple process for computation of sparse optimizers 5 years ago
sparse_apply_ftrl_cpu_kernel.h optimize reduce sparse gradient 5 years ago
sparse_apply_lazy_adam_cpu_kernel.cc Add multiple process for computation of sparse optimizers 5 years ago
sparse_apply_lazy_adam_cpu_kernel.h Add implementation of SparseApplyProximalAdagrad cpu kernel 5 years ago
sparse_apply_proximal_adagrad_cpu_kernel.cc Add multiple process for computation of sparse optimizers 5 years ago
sparse_apply_proximal_adagrad_cpu_kernel.h Add an output to apply_proximal_adagrad op register 5 years ago
sub_cpu_kernel.cc fix review code 5 years ago
sub_cpu_kernel.h fix sub input not match 5 years ago
transpose_cpu_kernel.cc add cpu transpose 5 years ago
transpose_cpu_kernel.h add cpu transpose 5 years ago