mindspore2022/mindspore/ccsrc/kernel/cpu
mindspore-ci-bot a022d42132 !1992 fix lstm weight initializer
Merge pull request !1992 from baihuawei/cpulstm
2020-06-11 15:55:57 +08:00
..
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