| .. |
|
mkldnn
|
!2369 add cpu reduce op and cpu softmax_cross_entropy_with_logits op
|
2020-06-22 09:50:44 +08:00 |
|
addn_cpu_kernel.cc
|
applymomentum
|
2020-06-17 12:12:44 +08:00 |
|
addn_cpu_kernel.h
|
multi_input
|
2020-05-27 22:04:22 +08:00 |
|
allgather_cpu_kernel.cc
|
check mpi_adapter instance
|
2020-06-24 14:41:20 +08:00 |
|
allgather_cpu_kernel.h
|
change HostAllGather and HostReduceScatter to internal interface
|
2020-06-23 16:56:04 +08:00 |
|
apply_momentum_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
apply_momentum_cpu_kernel.h
|
applymomentum
|
2020-06-17 12:12:44 +08:00 |
|
argmax_cpu_kernel.cc
|
change cpu kernel dir
|
2020-04-27 21:14:03 +08:00 |
|
argmax_cpu_kernel.h
|
fix cpu argmax register
|
2020-06-17 20:09:39 +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 reduce sparse gradient
|
2020-06-12 20:36:22 +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 |
|
debug_cpu_kernel.cc
|
Implementation for mindspore debugger
|
2020-06-22 17:15:30 -04:00 |
|
debug_cpu_kernel.h
|
Implementation for mindspore debugger
|
2020-06-22 17:15:30 -04:00 |
|
embedding_look_up_comm_grad_cpu_kernel.cc
|
check mpi_adapter instance
|
2020-06-24 14:41:20 +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
|
clear warning
|
2020-06-28 17:10:03 +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
|
sync cpu output
|
2020-06-20 15:55:02 +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_cpu_kernel.cc
|
clear warning
|
2020-06-28 17:10:03 +08:00 |
|
reduce_cpu_kernel.h
|
clear warning
|
2020-06-28 17:10:03 +08:00 |
|
reduce_scatter_cpu_kernel.cc
|
check mpi_adapter instance
|
2020-06-24 14:41:20 +08:00 |
|
reduce_scatter_cpu_kernel.h
|
change HostAllGather and HostReduceScatter to internal interface
|
2020-06-23 16:56:04 +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
|
applymomentum
|
2020-06-17 12:12:44 +08:00 |
|
slice_cpu_kernel.h
|
applymomentum
|
2020-06-17 12:12:44 +08:00 |
|
slice_grad_cpu_kernel.cc
|
applymomentum
|
2020-06-17 12:12:44 +08:00 |
|
slice_grad_cpu_kernel.h
|
applymomentum
|
2020-06-17 12:12:44 +08:00 |
|
sparse_apply_adam_cpu_kernel.cc
|
Adjust the thread num to compute optimizer in cpu
|
2020-06-30 19:25:34 +08:00 |
|
sparse_apply_adam_cpu_kernel.h
|
Add multiple process for computation of sparse optimizers
|
2020-06-22 14:19:39 +08:00 |
|
sparse_apply_ftrl_cpu_kernel.cc
|
Adjust the thread num to compute optimizer in cpu
|
2020-06-30 19:25:34 +08:00 |
|
sparse_apply_ftrl_cpu_kernel.h
|
add cpu sparse optimizer ops with no return
|
2020-06-30 12:46:15 +08:00 |
|
sparse_apply_lazy_adam_cpu_kernel.cc
|
Adjust the thread num to compute optimizer in cpu
|
2020-06-30 19:25:34 +08:00 |
|
sparse_apply_lazy_adam_cpu_kernel.h
|
Add implementation of SparseApplyProximalAdagrad cpu kernel
|
2020-06-18 11:43:27 +08:00 |
|
sparse_apply_proximal_adagrad_cpu_kernel.cc
|
Adjust the thread num to compute optimizer in cpu
|
2020-06-30 19:25:34 +08:00 |
|
sparse_apply_proximal_adagrad_cpu_kernel.h
|
add cpu sparse optimizer ops with no return
|
2020-06-30 12:46:15 +08:00 |
|
sub_cpu_kernel.cc
|
fix review code
|
2020-06-19 17:40:18 +08:00 |
|
sub_cpu_kernel.h
|
fix sub input not match
|
2020-06-19 16:56:41 +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 |
2a31e52bf1