| .. |
|
mkldnn
|
fix lstm weight initializer issue
|
2020-06-11 11:16:11 +08:00 |
|
addn_cpu_kernel.cc
|
addn
|
2020-05-26 20:16:41 +08:00 |
|
addn_cpu_kernel.h
|
multi_input
|
2020-05-27 22:04:22 +08:00 |
|
allgather_cpu_kernel.cc
|
support host reduce scatter and allgather
|
2020-06-04 17:20:47 +08:00 |
|
allgather_cpu_kernel.h
|
support host reduce scatter and allgather
|
2020-06-04 17:20:47 +08:00 |
|
apply_momentum_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
apply_momentum_cpu_kernel.h
|
cpu kernel register type
|
2020-05-09 17:31:45 +08:00 |
|
argmax_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
argmax_cpu_kernel.h
|
cpu kernel register type
|
2020-05-09 17:31:45 +08:00 |
|
bias_add_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
bias_add_cpu_kernel.h
|
cpu kernel register type
|
2020-05-09 17:31:45 +08:00 |
|
bias_add_grad_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
bias_add_grad_cpu_kernel.h
|
cpu kernel register type
|
2020-05-09 17:31:45 +08:00 |
|
concat_cpu_kernel.cc
|
review
|
2020-05-25 15:59:08 +08:00 |
|
concat_cpu_kernel.h
|
multi_input
|
2020-05-27 22:04:22 +08:00 |
|
cpu_kernel.cc
|
optimize performance
|
2020-06-04 16:20:51 +08:00 |
|
cpu_kernel.h
|
Add SparseAdam and SparseLazyAdam cpu kernel
|
2020-06-09 22:24:38 +08:00 |
|
cpu_kernel_factory.cc
|
multi_input
|
2020-05-27 22:04:22 +08:00 |
|
cpu_kernel_factory.h
|
multi_input
|
2020-05-27 22:04:22 +08:00 |
|
embedding_look_up_comm_grad_cpu_kernel.cc
|
add cpu op embedding_look_up
|
2020-06-05 14:28:18 +08:00 |
|
embedding_look_up_comm_grad_cpu_kernel.h
|
add cpu op embedding_look_up
|
2020-06-05 14:28:18 +08:00 |
|
embedding_look_up_cpu_kernel.cc
|
fix embedding look up op run fail
|
2020-06-06 10:55:31 +08:00 |
|
embedding_look_up_cpu_kernel.h
|
add cpu op embedding_look_up
|
2020-06-05 14:28:18 +08:00 |
|
equal_count_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
equal_count_cpu_kernel.h
|
cpu kernel register type
|
2020-05-09 17:31:45 +08:00 |
|
gather_cpu_kernel.cc
|
review
|
2020-05-25 15:59:08 +08:00 |
|
gather_cpu_kernel.h
|
review
|
2020-05-25 15:59:08 +08:00 |
|
one_hot_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
one_hot_cpu_kernel.h
|
cpu kernel register type
|
2020-05-09 17:31:45 +08:00 |
|
reduce_scatter_cpu_kernel.cc
|
support host reduce scatter and allgather
|
2020-06-04 17:20:47 +08:00 |
|
reduce_scatter_cpu_kernel.h
|
support host reduce scatter and allgather
|
2020-06-04 17:20:47 +08:00 |
|
reshape_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
reshape_cpu_kernel.h
|
cpu kernel support mutil dtype
|
2020-05-20 19:11:13 +08:00 |
|
slice_cpu_kernel.cc
|
optimize performance
|
2020-06-04 16:20:51 +08:00 |
|
slice_cpu_kernel.h
|
optimize performance
|
2020-06-04 16:20:51 +08:00 |
|
slice_grad_cpu_kernel.cc
|
optimize performance
|
2020-06-04 16:20:51 +08:00 |
|
slice_grad_cpu_kernel.h
|
optimize performance
|
2020-06-04 16:20:51 +08:00 |
|
sparse_apply_adam_cpu_kernel.cc
|
Add SparseAdam and SparseLazyAdam cpu kernel
|
2020-06-09 22:24:38 +08:00 |
|
sparse_apply_adam_cpu_kernel.h
|
Add SparseAdam and SparseLazyAdam cpu kernel
|
2020-06-09 22:24:38 +08:00 |
|
sparse_apply_ftrl_cpu_kernel.cc
|
Add SparseAdam and SparseLazyAdam cpu kernel
|
2020-06-09 22:24:38 +08:00 |
|
sparse_apply_ftrl_cpu_kernel.h
|
Add SparseApplyFtrl cpu kernel
|
2020-06-04 22:06:39 +08:00 |
|
sparse_apply_lazy_adam_cpu_kernel.cc
|
Add SparseAdam and SparseLazyAdam cpu kernel
|
2020-06-09 22:24:38 +08:00 |
|
sparse_apply_lazy_adam_cpu_kernel.h
|
Add SparseAdam and SparseLazyAdam cpu kernel
|
2020-06-09 22:24:38 +08:00 |
|
subscalar_cpu_kernel.cc
|
add cpu op embedding_look_up
|
2020-06-05 14:28:18 +08:00 |
|
subscalar_cpu_kernel.h
|
add cpu op embedding_look_up
|
2020-06-05 14:28:18 +08:00 |
|
transpose_cpu_kernel.cc
|
add cpu transpose
|
2020-05-27 20:51:47 +08:00 |
|
transpose_cpu_kernel.h
|
add cpu transpose
|
2020-05-27 20:51:47 +08:00 |