mindspore2022/mindspore/ccsrc/backend/kernel_compiler/cpu
i-robot 56f330c0f1 !19688 optimize CPU arithmetic ops
Merge pull request !19688 from 范吉斌/optimize_ops
2021-07-09 11:44:11 +00:00
..
fl Fix fl namespace issue. 2021-07-08 21:52:21 +08:00
mkldnn fix conv3d when pad_mode is pad 2021-07-06 17:06:23 +08:00
nnacl !19700 add avx512 compile option 2021-07-09 02:16:40 +00:00
ps sync bug fix 2021-07-05 11:52:16 +08:00
quantum mindquantum doc format 2021-07-06 10:12:02 +08:00
OWNERS add owners of cpu op 2021-06-11 15:10:59 +08:00
adam_cpu_kernel.cc fix opimizer bug 2021-06-23 17:55:58 +08:00
adam_cpu_kernel.h add fp16 support for CPU Adam 2021-05-28 16:52:08 +08:00
adam_delta_cpu_kernel.cc clean pclint for cpu operator 2021-05-27 17:38:02 +08:00
adam_delta_cpu_kernel.h format some code 2021-05-24 20:00:11 +08:00
adam_weight_decay_cpu_kernel.cc zsd AdamWeightDecay CPU op avx512 2021-07-01 17:20:54 +08:00
adam_weight_decay_cpu_kernel.h zsd AdamWeightDecay CPU op avx512 2021-07-01 17:20:54 +08:00
allgather_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
allgather_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
apply_adagrad_cpu_kernel.cc fix opimizer bug 2021-06-16 15:30:03 +08:00
apply_adagrad_cpu_kernel.h clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
apply_momentum_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
apply_momentum_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
argmax_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
argmax_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
argmin_with_value_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
argmin_with_value_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
arithmetic_cpu_kernel.cc optimize cpu arithmetic operators 2021-07-08 16:30:22 +08:00
arithmetic_cpu_kernel.h optimize cpu arithmetic operators 2021-07-08 16:30:22 +08:00
arithmetic_logic_cpu_kernel.cc optimize cpu arithmetic operators 2021-07-08 16:30:22 +08:00
arithmetic_logic_cpu_kernel.h optimize cpu arithmetic operators 2021-07-08 16:30:22 +08:00
arithmetic_self_cpu_kernel.cc fix cpu lstm and precision issues 2021-06-03 09:17:01 +08:00
arithmetic_self_cpu_kernel.h Add identity op for cpu and gpu 2021-05-24 13:56:09 +00:00
assign_cpu_kernel.cc cpu assign bug fix 2021-06-28 09:21:55 +08:00
assign_cpu_kernel.h Add assign op for cpu 2020-12-05 10:52:07 +08:00
bias_add_cpu_kernel.cc revert reimplementation of biasAdd using nnacl 2021-06-15 16:59:23 +08:00
bias_add_cpu_kernel.h revert reimplementation of biasAdd using nnacl 2021-06-15 16:59:23 +08:00
bias_add_grad_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
bias_add_grad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
binary_cross_entropy_cpu_kernel.cc !16629 Binary_cross_entropy op supports optional weight input 2021-05-25 09:26:00 +08:00
binary_cross_entropy_cpu_kernel.h binary_cross_entropy op supports optional weight input 2021-05-20 10:47:12 +08:00
binary_cross_entropy_grad_kernel.cc binary_cross_entropy op supports optional weight input 2021-05-20 10:47:12 +08:00
binary_cross_entropy_grad_kernel.h binary_cross_entropy op supports optional weight input 2021-05-20 10:47:12 +08:00
broadcast_to_cpu_kernel.cc convert the implementation of BroadCastTo, ReluGrad, ReLU6Grad CPU operators to nnacl 2021-04-22 15:36:26 +08:00
broadcast_to_cpu_kernel.h convert the implementation of BroadCastTo, ReluGrad, ReLU6Grad CPU operators to nnacl 2021-04-22 15:36:26 +08:00
cast_cpu_kernel.cc code security check 2021-06-16 11:25:13 +08:00
cast_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
concat_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
concat_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
cpu_kernel.cc Add MaskedSelect CPU Operation 2021-06-01 17:17:58 +08:00
cpu_kernel.h Add MaskedSelect CPU Operation 2021-06-01 17:17:58 +08:00
cpu_kernel_factory.cc fix code review problems 2021-06-15 10:54:22 +08:00
cpu_kernel_factory.h !17111 clean pclint for cpu operator 2021-05-29 18:25:51 +08:00
ctcloss_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
ctcloss_cpu_kernel.h clean code again master 2021-06-01 18:08:49 +08:00
cumsum_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
cumsum_cpu_kernel.h Add CumSum Kerner for CPU 2020-12-25 16:30:11 +08:00
debug_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
debug_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
depthtospace_cpu_kernel.cc add ops :depthtospace,spacetodepth 2021-05-14 08:08:52 +00:00
depthtospace_cpu_kernel.h add ops :depthtospace,spacetodepth 2021-05-14 08:08:52 +00:00
dropout_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
dropout_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
dropout_grad_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
dropout_grad_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
dynamic_assign_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
dynamic_assign_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
eltwise_grad_cpu_kernel.cc code security check 2021-06-16 11:25:13 +08:00
eltwise_grad_cpu_kernel.h codex 2021-04-29 15:15:09 +08:00
elu_grad_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
elu_grad_cpu_kernel.h 告警清理 2021-04-29 11:27:51 +08:00
embedding_look_up_comm_grad_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
embedding_look_up_comm_grad_cpu_kernel.h support int64 shape 2020-11-05 21:15:49 +08:00
embedding_look_up_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
embedding_look_up_cpu_kernel.h !13305 fix pynative memory leak 2021-03-15 20:46:21 +08:00
equal_count_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
equal_count_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
expm1_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
expm1_cpu_kernel.h expm1 operator 2020-12-15 20:28:07 +08:00
gather_cpu_kernel.cc sync bugfix of Gather op from enterprize 2021-06-28 19:48:16 +08:00
gather_cpu_kernel.h [CPU] Optimize slice and gather ops's performance 2021-05-28 16:16:28 +08:00
gather_d_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
gather_d_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
gather_d_grad_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
gather_d_grad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
gathernd_cpu_kernel.cc !17515 cpu ops pclint clean 2021-06-02 14:18:59 +08:00
gathernd_cpu_kernel.h complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul 2021-06-01 17:40:45 +08:00
hsigmoid_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
hsigmoid_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
hsigmoid_grad_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
hsigmoid_grad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
hswish_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
hswish_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
hswish_grad_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
hswish_grad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
iou_cpu_kernel.cc add CPU IOU 2021-04-17 17:34:25 +08:00
iou_cpu_kernel.h add CPU IOU 2021-04-17 17:34:25 +08:00
isfinite_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
isfinite_cpu_kernel.h !17111 clean pclint for cpu operator 2021-05-29 18:25:51 +08:00
isnan_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
isnan_cpu_kernel.h fix 2021-05-26 17:18:47 +08:00
l2_normalize_cpu_kernel.cc fix l2normalize ops parallel problem 2021-04-15 13:37:26 +08:00
l2_normalize_cpu_kernel.h fix l2normalize ops parallel problem 2021-04-15 13:37:26 +08:00
l2normalize_grad_cpu_kernel.cc Add L2NormalizeGradCPUKernel 2021-04-12 11:41:40 +08:00
l2normalize_grad_cpu_kernel.h Add L2NormalizeGradCPUKernel 2021-04-12 11:41:40 +08:00
layer_norm_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
layer_norm_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
layer_norm_grad_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
layer_norm_grad_cpu_kernel.h !17111 clean pclint for cpu operator 2021-05-29 18:25:51 +08:00
map_cache_idx_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
map_cache_idx_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
map_uniform_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
map_uniform_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
masked_select_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
masked_select_cpu_kernel.h Add MaskedSelect CPU Operation 2021-06-01 17:17:58 +08:00
masked_select_grad_cpu_kernel.cc [bugfix] The output of MaskEdSelectGrad is wrong due to output addr hasn't been memset 2021-07-08 21:13:56 +08:00
masked_select_grad_cpu_kernel.h Add MaskedSelect CPU Operation 2021-06-01 17:17:58 +08:00
maximum_cpu_kernel.cc clean pclint for cpu operator 2021-05-27 17:38:02 +08:00
maximum_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
maximum_grad_cpu_kernel.cc fix cpu lstm and precision issues 2021-06-03 09:17:01 +08:00
maximum_grad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
minimum_cpu_kernel.cc clear pclint CodeCheck warnings 2021-05-27 16:17:18 +08:00
minimum_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
minimum_grad_cpu_kernel.cc fix slice and minimum op bug 2021-06-11 11:54:10 +08:00
minimum_grad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
mirror_pad_cpu_kernel.cc !18832 change infer type to device type for cpu 2021-06-26 07:59:09 +00:00
mirror_pad_cpu_kernel.h clean code again master 2021-06-01 18:08:49 +08:00
mirror_pad_grad_cpu_kernel.cc !18832 change infer type to device type for cpu 2021-06-26 07:59:09 +00:00
mirror_pad_grad_cpu_kernel.h code_clean 2021-06-10 14:51:46 +08:00
one_hot_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
one_hot_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
pack_cpu_kernel.cc codex 2021-04-29 15:15:09 +08:00
pack_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
pad_and_shift_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
pad_and_shift_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
pad_cpu_kernel.cc !18832 change infer type to device type for cpu 2021-06-26 07:59:09 +00:00
pad_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
print_cpu_kernel.cc [CPU] Add TensorScatterUpdate and Print Ops 2021-05-11 19:53:10 +08:00
print_cpu_kernel.h [CPU] Add TensorScatterUpdate and Print Ops 2021-05-11 19:53:10 +08:00
random_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
random_cpu_kernel.h CPU operation dev:StandardNormal 2020-12-04 17:43:03 +08:00
range_cpu_kernel.cc pclint clean 2021-06-03 21:01:31 +08:00
range_cpu_kernel.h pclint clean 2021-06-02 09:52:32 +08:00
reduce_cpu_kernel.cc Add uniqueName for data dump 2021-07-08 09:37:34 +08:00
reduce_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
reduce_scatter_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
reduce_scatter_cpu_kernel.h Unified code style 2020-07-20 10:51:55 +08:00
reshape_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +08:00
reshape_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
resize_bilinear_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
resize_bilinear_cpu_kernel.h add_resize_ops 2020-12-19 01:06:27 +08:00
resize_bilinear_grad_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
resize_bilinear_grad_cpu_kernel.h add_resize_ops 2020-12-19 01:06:27 +08:00
resize_nearest_neighbor_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
resize_nearest_neighbor_cpu_kernel.h add_int64_float64 2021-02-26 19:36:24 +08:00
resize_nearest_neighbor_grad_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
resize_nearest_neighbor_grad_cpu_kernel.h add_int64_float64 2021-02-26 19:36:24 +08:00
rmsprop_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
rmsprop_cpu_kernel.h optimize rmsprop op by simd instruction 2021-05-11 20:21:24 +08:00
scatter_arithmetic_cpu_kernel.cc fix bug: remove multi-thread, because the input will operate multiple times. 2021-04-22 20:19:41 +08:00
scatter_arithmetic_cpu_kernel.h add cpu operations: ScatterAdd ScatterSub ScatterMul ScatterDiv ScatterMax ScatterMin ScatterUpdate 2021-04-16 16:46:36 +08:00
scatter_nd_update_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
scatter_nd_update_cpu_kernel.h [CPU] Add TensorScatterUpdate and Print Ops 2021-05-11 19:53:10 +08:00
searchsorted_cpu_kernel.cc adding SearchSorted operator to CPU 2021-06-03 10:30:55 +08:00
searchsorted_cpu_kernel.h adding SearchSorted operator to CPU 2021-06-03 10:30:55 +08:00
select_cpu_kernel.cc backup select, less, pow, sub CPU ops's performance 2021-05-31 11:33:20 +08:00
select_cpu_kernel.h 告警清理 2021-04-26 16:51:50 +08:00
sgd_cpu_kernel.cc added SGD operation in CPU 2021-04-14 15:19:10 +08:00
sgd_cpu_kernel.h added SGD operation in CPU 2021-04-14 15:19:10 +08:00
sigmoid_cross_entropy_with_logits_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
sigmoid_cross_entropy_with_logits_cpu_kernel.h add_cpu_op 2020-10-30 13:07:35 +08:00
sigmoid_cross_entropy_with_logits_grad_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
sigmoid_cross_entropy_with_logits_grad_cpu_kernel.h add_cpu_op 2020-10-30 13:07:35 +08:00
slice_cpu_kernel.cc code security check 2021-06-16 11:25:13 +08:00
slice_cpu_kernel.h code security check 2021-06-16 11:25:13 +08:00
slice_grad_cpu_kernel.cc fix slice and minimum op bug 2021-06-11 11:54:10 +08:00
slice_grad_cpu_kernel.h fix slice op bug 2021-03-24 14:40:56 +08:00
smooth_l1_loss_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
smooth_l1_loss_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
smooth_l1_loss_grad_cpu_kernel.cc clean code for CPU and ModelZoo 2021-05-21 10:31:27 +08:00
smooth_l1_loss_grad_cpu_kernel.h Refactor some cpu ops 2021-03-16 20:44:55 +08:00
spacetodepth_cpu_kernel.cc add ops :depthtospace,spacetodepth 2021-05-14 08:08:52 +00:00
spacetodepth_cpu_kernel.h add ops :depthtospace,spacetodepth 2021-05-14 08:08:52 +00:00
sparse_apply_adam_cpu_kernel.cc Fix exception log and check secure func ret 2021-06-11 08:55:22 +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 Fix exception log and check secure func ret 2021-06-11 08:55:22 +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 Fix exception log and check secure func ret 2021-06-11 08:55:22 +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 Fix pclint plus warning 2021-05-26 15:21:58 +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 use thread pool for some cpu kernel 2020-12-17 17:11:34 +08:00
sparse_tensor_dense_matmul_cpu_kernel.cc fix bug of CPU actor runtime 2021-06-19 17:49:31 +08:00
sparse_tensor_dense_matmul_cpu_kernel.h complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul 2021-06-01 17:40:45 +08:00
sparse_to_dense_cpu_kernal.cc fix bug of CPU actor runtime 2021-06-19 17:49:31 +08:00
sparse_to_dense_cpu_kernal.h complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul 2021-06-01 17:40:45 +08:00
split_cpu_kernel.cc fix split memory leak 2021-06-28 18:25:00 +08:00
split_cpu_kernel.h fix split memory leak 2021-06-28 18:25:00 +08:00
strided_slice_grad_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
strided_slice_grad_cpu_kernel.h Cloud ops: add StridedSliceGrad ops 2021-04-30 14:26:14 +08:00
stridedslice_cpu_kernel.cc code security check 2021-06-16 11:25:13 +08:00
stridedslice_cpu_kernel.h convert the implementation of Slice and StridedSlice CPU operators to nnacl 2021-04-27 14:09:02 +08:00
sub_and_filter_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
sub_and_filter_cpu_kernel.h fix pynative memory leak 2021-03-15 14:20:50 +08:00
tensor_copy_slices_cpu_kernel.cc Add TensorCopySlice Op 2021-06-29 16:52:35 +08:00
tensor_copy_slices_cpu_kernel.h Add TensorCopySlice Op 2021-06-29 16:52:35 +08:00
tensoradd_cpu_kernel.cc codex 2021-04-29 15:15:09 +08:00
tensoradd_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
tile_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
tile_cpu_kernel.h !16988 upload unify cpu ops info library code 2021-05-27 16:58:24 +08:00
topk_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
topk_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
transpose_cpu_kernel.cc optimize cpu arithmetic operators 2021-07-08 16:30:22 +08:00
transpose_cpu_kernel.h upload unify cpu ops info library code 2021-05-26 16:29:25 +08:00
unique_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
unique_cpu_kernel.h cpu unique kernel support fp16 2021-04-20 11:40:13 +08:00
unique_with_pad_cpu_kernel.cc cpu unique kernel support fp16 2021-04-20 11:40:13 +08:00
unique_with_pad_cpu_kernel.h cpu unique kernel support fp16 2021-04-20 11:40:13 +08:00
unpack_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
unpack_cpu_kernel.h converter the implementation of DropGrad and unpack and transpose CPU operators to nnacl 2021-05-13 21:56:23 +08:00
unsorted_segment_sum_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
unsorted_segment_sum_cpu_kernel.h convert the implementation of tile unsortedSegmentSum CPU ops to nnacl 2021-05-20 19:11:12 +08:00
update_cache_cpu_kernel.cc change infer type to device type for cpu 2021-06-25 09:20:25 +08:00
update_cache_cpu_kernel.h clean pclint 2021-05-27 10:23:46 +08:00