mindspore2022/mindspore/ccsrc/backend/kernel_compiler/cpu
huanghui a4b485eb51 thread pool 2020-10-12 09:12:16 +08:00
..
mkldnn !6921 fix cpu sparse_softmax_encropy_with_logits 2020-09-28 09:51:02 +08:00
ps Set weight size limit for PS mode. 2020-09-24 10:23:26 +08:00
allgather_cpu_kernel.cc dlopen cpu mpi adapter 2020-08-09 14:29:23 +08:00
allgather_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
apply_momentum_cpu_kernel.cc decoupling core and context 2020-08-05 08:15:14 +08:00
apply_momentum_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
argmax_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
argmax_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
arithmetic_cpu_kernel.cc cpu Sub support broadcast shape 2020-09-28 15:18:51 +08:00
arithmetic_cpu_kernel.h cpu Sub support broadcast shape 2020-09-28 15:18:51 +08:00
arithmetic_self_cpu_kernel.cc Add CPU kernels: TensorAdd, Sub, Mul, Div 2020-09-10 20:45:03 +08:00
arithmetic_self_cpu_kernel.h Add CPU kernels: TensorAdd, Sub, Mul, Div 2020-09-10 20:45:03 +08:00
assign_cpu_kernel.cc add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
assign_cpu_kernel.h add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
bias_add_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
bias_add_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
bias_add_grad_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
bias_add_grad_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
cache_swap_hashmap_cpu_kernel.cc add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
cache_swap_hashmap_cpu_kernel.h add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
concat_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
concat_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
cpu_kernel.h Add CPU kernels: TensorAdd, Sub, Mul, Div 2020-09-10 20:45:03 +08:00
cpu_kernel_factory.cc optimize code compile performance 2020-08-18 14:33:24 +08:00
cpu_kernel_factory.h decoupling core and context 2020-08-05 08:15:14 +08:00
debug_cpu_kernel.cc enable debugger by default and set correct log message severity 2020-09-10 21:42:00 -04:00
debug_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
embedding_look_up_comm_grad_cpu_kernel.cc dlopen cpu mpi adapter 2020-08-09 14:29:23 +08:00
embedding_look_up_comm_grad_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
embedding_look_up_cpu_kernel.cc Fix the bug of memcpy_s for embeddinglookup cpu kernel 2020-09-20 12:20:22 +08:00
embedding_look_up_cpu_kernel.h add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
equal_count_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
equal_count_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
gather_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
gather_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
map_cache_idx_cpu_kernel.cc add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
map_cache_idx_cpu_kernel.h add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
one_hot_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
one_hot_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
reduce_cpu_kernel.cc clear warnings 2020-09-22 19:43:08 +08:00
reduce_cpu_kernel.h clear warnings 2020-09-22 19:43:08 +08:00
reduce_scatter_cpu_kernel.cc dlopen cpu mpi adapter 2020-08-09 14:29:23 +08:00
reduce_scatter_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
reshape_cpu_kernel.cc add ScatterNdUpdate cpu kernel 2020-08-17 14:14:26 +08:00
reshape_cpu_kernel.h add ScatterNdUpdate cpu kernel 2020-08-17 14:14:26 +08:00
scatter_nd_update_cpu_kernel.cc thread pool 2020-10-12 09:12:16 +08:00
scatter_nd_update_cpu_kernel.h optimize ScatterNdUpdate cpu kernel 2020-10-09 16:01:35 +08:00
search_cache_idx_cpu_kernel.cc add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
search_cache_idx_cpu_kernel.h add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
slice_cpu_kernel.cc fix cpu slice 2020-10-09 17:26:12 +08:00
slice_cpu_kernel.h generalize CPU Slice op 2020-09-26 22:24:26 +08:00
slice_grad_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
slice_grad_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
sparse_apply_adam_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_adam_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_ftrl_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_ftrl_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_lazy_adam_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_lazy_adam_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_proximal_adagrad_cpu_kernel.cc Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_apply_proximal_adagrad_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
sparse_optimizer_cpu_kernel.h Support int64 for cpu sparse optimizers 2020-08-27 10:57:53 +08:00
transpose_cpu_kernel.cc mindspore path adjust 2020-07-14 18:07:28 +08:00
transpose_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
unique_cpu_kernel.cc add Unique cpu kernel 2020-08-11 21:36:55 +08:00
unique_cpu_kernel.h clear codex and reviewbot warning 2020-08-28 17:27:43 +08:00
unique_with_pad_cpu_kernel.cc add unique_with_pad cpu kernel 2020-08-13 21:55:00 +08:00
unique_with_pad_cpu_kernel.h clear codex and reviewbot warning 2020-08-28 17:27:43 +08:00
update_cache_cpu_kernel.cc add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00
update_cache_cpu_kernel.h add cache ops for cpu and aicpu 2020-09-28 17:45:25 +08:00