You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
i-robot 20c4e389ac !22469 code check 4 years ago
..
fl Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
mkldnn !22236 clean code in cpu operater 4 years ago
nnacl !22469 code check 4 years ago
ps [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
pyfunc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
quantum mindquantum doc format 4 years ago
rl update buffer sample gpu 4 years ago
OWNERS add owners of cpu op 4 years ago
adam_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
adam_cpu_kernel.h clean code in cpu operater 4 years ago
adam_delta_cpu_kernel.cc clean pclint for cpu operator 4 years ago
adam_delta_cpu_kernel.h format some code 4 years ago
adam_weight_decay_cpu_kernel.cc Adam op performance optimization 4 years ago
adam_weight_decay_cpu_kernel.h Adam op performance optimization 4 years ago
allgather_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
allgather_cpu_kernel.h Unified code style 5 years ago
apply_adagrad_cpu_kernel.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
apply_adagrad_cpu_kernel.h clean code for CPU and ModelZoo 4 years ago
apply_momentum_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
apply_momentum_cpu_kernel.h Unified code style 5 years ago
argmax_cpu_kernel.cc fix vector push_back realloc memory issue 4 years ago
argmax_cpu_kernel.h upload unify cpu ops info library code 4 years ago
argmax_with_value_cpu_kernel.cc adding 9 object detection operators 4 years ago
argmax_with_value_cpu_kernel.h adding 9 object detection operators 4 years ago
argmin_with_value_cpu_kernel.cc fix vector push_back realloc memory issue 4 years ago
argmin_with_value_cpu_kernel.h upload unify cpu ops info library code 4 years ago
arithmetic_cpu_kernel.cc clean code in cpu operater 4 years ago
arithmetic_cpu_kernel.h clean code in cpu operater 4 years ago
arithmetic_logic_cpu_kernel.cc clean code in cpu operater 4 years ago
arithmetic_logic_cpu_kernel.h clean code in cpu operater 4 years ago
arithmetic_self_cpu_kernel.cc clean code in cpu operater 4 years ago
arithmetic_self_cpu_kernel.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
assign_cpu_kernel.cc cpu assign bug fix 4 years ago
assign_cpu_kernel.h Add assign op for cpu 5 years ago
bias_add_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
bias_add_cpu_kernel.h revert reimplementation of biasAdd using nnacl 4 years ago
bias_add_grad_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
bias_add_grad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
binary_cross_entropy_cpu_kernel.cc fix codecheck on master 4 years ago
binary_cross_entropy_cpu_kernel.h change from CRLF to LF 4 years ago
binary_cross_entropy_grad_kernel.cc change from CRLF to LF 4 years ago
binary_cross_entropy_grad_kernel.h change from CRLF to LF 4 years ago
boundingbox_decode_cpu_kernel.cc adding 9 object detection operators 4 years ago
boundingbox_decode_cpu_kernel.h adding 9 object detection operators 4 years ago
boundingbox_encode_cpu_kernel.cc adding 9 object detection operators 4 years ago
boundingbox_encode_cpu_kernel.h adding 9 object detection operators 4 years ago
broadcast_to_cpu_kernel.cc clean code in cpu operater 4 years ago
broadcast_to_cpu_kernel.h convert the implementation of BroadCastTo, ReluGrad, ReLU6Grad CPU operators to nnacl 4 years ago
cast_cpu_kernel.cc code security check 4 years ago
cast_cpu_kernel.h upload unify cpu ops info library code 4 years ago
check_valid_cpu_kernel.cc adding 9 object detection operators 4 years ago
check_valid_cpu_kernel.h adding 9 object detection operators 4 years ago
concat_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
concat_cpu_kernel.h upload unify cpu ops info library code 4 years ago
cpu_kernel.cc !22465 Adam op performance optimization 4 years ago
cpu_kernel.h add rolling cpu kernel 4 years ago
cpu_kernel_factory.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
cpu_kernel_factory.h !17111 clean pclint for cpu operator 4 years ago
crop_and_resize_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
crop_and_resize_cpu_kernel.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
ctcloss_cpu_kernel.cc change from CRLF to LF 4 years ago
ctcloss_cpu_kernel.h change from CRLF to LF 4 years ago
cumsum_cpu_kernel.cc change infer type to device type for cpu 4 years ago
cumsum_cpu_kernel.h Add CumSum Kerner for CPU 5 years ago
debug_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
debug_cpu_kernel.h Unified code style 5 years ago
depthtospace_cpu_kernel.cc clean code in cpu operater 4 years ago
depthtospace_cpu_kernel.h clean code in cpu operater 4 years ago
dropout_cpu_kernel.cc change infer type to device type for cpu 4 years ago
dropout_cpu_kernel.h upload unify cpu ops info library code 4 years ago
dropout_grad_kernel.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
dropout_grad_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
dynamic_assign_cpu_kernel.cc change infer type to device type for cpu 4 years ago
dynamic_assign_cpu_kernel.h fix pynative memory leak 5 years ago
eltwise_grad_cpu_kernel.cc clean code in cpu operater 4 years ago
eltwise_grad_cpu_kernel.h clean code in cpu operater 4 years ago
elu_grad_cpu_kernel.cc change infer type to device type for cpu 4 years ago
elu_grad_cpu_kernel.h 告警清理 4 years ago
embedding_look_up_comm_grad_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
embedding_look_up_comm_grad_cpu_kernel.h support int64 shape 5 years ago
embedding_look_up_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
embedding_look_up_cpu_kernel.h !13305 fix pynative memory leak 5 years ago
equal_count_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
equal_count_cpu_kernel.h upload unify cpu ops info library code 4 years ago
expm1_cpu_kernel.cc change infer type to device type for cpu 4 years ago
expm1_cpu_kernel.h expm1 operator 5 years ago
fused_cast_adam_weight_decay_cpu_kernel.cc Adam op performance optimization 4 years ago
fused_cast_adam_weight_decay_cpu_kernel.h Adam op performance optimization 4 years ago
gather_cpu_kernel.cc update gather CPU OP to support dynamic shape. 4 years ago
gather_cpu_kernel.h update gather CPU OP to support dynamic shape. 4 years ago
gather_d_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
gather_d_cpu_kernel.h upload unify cpu ops info library code 4 years ago
gather_d_grad_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
gather_d_grad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
gathernd_cpu_kernel.cc !17515 cpu ops pclint clean 4 years ago
gathernd_cpu_kernel.h complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul 4 years ago
hsigmoid_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
hsigmoid_cpu_kernel.h upload unify cpu ops info library code 4 years ago
hsigmoid_grad_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
hsigmoid_grad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
hswish_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
hswish_cpu_kernel.h upload unify cpu ops info library code 4 years ago
hswish_grad_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
hswish_grad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
iou_cpu_kernel.cc fix bug in gpu/array 4 years ago
iou_cpu_kernel.h clean code in cpu operater 4 years ago
isfinite_cpu_kernel.cc change infer type to device type for cpu 4 years ago
isfinite_cpu_kernel.h !17111 clean pclint for cpu operator 4 years ago
isnan_cpu_kernel.cc change infer type to device type for cpu 4 years ago
isnan_cpu_kernel.h fix 4 years ago
l2_normalize_cpu_kernel.cc clean code in cpu operater 4 years ago
l2_normalize_cpu_kernel.h clean code in cpu operater 4 years ago
l2loss_cpu_kernel.cc add CPU l2loss op 4 years ago
l2loss_cpu_kernel.h add CPU l2loss op 4 years ago
l2normalize_grad_cpu_kernel.cc clean code in cpu operater 4 years ago
l2normalize_grad_cpu_kernel.h fix vector push_back realloc memory issue 4 years ago
layer_norm_cpu_kernel.cc clean code in cpu operater 4 years ago
layer_norm_cpu_kernel.h upload unify cpu ops info library code 4 years ago
layer_norm_grad_cpu_kernel.cc clean code in cpu operater 4 years ago
layer_norm_grad_cpu_kernel.h !17111 clean pclint for cpu operator 4 years ago
map_cache_idx_cpu_kernel.cc acc reducesum 4 years ago
map_cache_idx_cpu_kernel.h fix pynative memory leak 5 years ago
map_uniform_cpu_kernel.cc change infer type to device type for cpu 4 years ago
map_uniform_cpu_kernel.h fix pynative memory leak 5 years ago
masked_select_cpu_kernel.cc clean code in cpu operater 4 years ago
masked_select_cpu_kernel.h clean code in cpu operater 4 years ago
masked_select_grad_cpu_kernel.cc clean code in cpu operater 4 years ago
masked_select_grad_cpu_kernel.h clean code in cpu operater 4 years ago
maximum_cpu_kernel.cc clean pclint for cpu operator 4 years ago
maximum_cpu_kernel.h upload unify cpu ops info library code 4 years ago
maximum_grad_cpu_kernel.cc fix cpu lstm and precision issues 4 years ago
maximum_grad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
minimum_cpu_kernel.cc clear pclint CodeCheck warnings 4 years ago
minimum_cpu_kernel.h upload unify cpu ops info library code 4 years ago
minimum_grad_cpu_kernel.cc Fix MinimumGrad for CPU, consistent with GPU/Ascend 4 years ago
minimum_grad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
mirror_pad_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
mirror_pad_cpu_kernel.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
mirror_pad_grad_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
mirror_pad_grad_cpu_kernel.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
multinomial_cpu_kernel.cc fix multinomial cpu kernel 4 years ago
multinomial_cpu_kernel.h fix multinomial cpu kernel 4 years ago
nms_with_mask_cpu_kernel.cc adding 9 object detection operators 4 years ago
nms_with_mask_cpu_kernel.h adding 9 object detection operators 4 years ago
one_hot_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
one_hot_cpu_kernel.h upload unify cpu ops info library code 4 years ago
pack_cpu_kernel.cc codex 4 years ago
pack_cpu_kernel.h upload unify cpu ops info library code 4 years ago
pad_and_shift_cpu_kernel.cc fix vector push_back realloc memory issue 4 years ago
pad_and_shift_cpu_kernel.h fix pynative memory leak 5 years ago
pad_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
pad_cpu_kernel.h upload unify cpu ops info library code 4 years ago
print_cpu_kernel.cc clean code in cpu operater 4 years ago
print_cpu_kernel.h [CPU] Add TensorScatterUpdate and Print Ops 4 years ago
random_choice_with_mask_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
random_choice_with_mask_cpu_kernel.h adding 9 object detection operators 4 years ago
random_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
random_cpu_kernel.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
range_cpu_kernel.cc pclint clean 4 years ago
range_cpu_kernel.h pclint clean 4 years ago
reduce_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
reduce_cpu_kernel.h acc reducesum 4 years ago
reduce_scatter_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
reduce_scatter_cpu_kernel.h Unified code style 5 years ago
reshape_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
reshape_cpu_kernel.h fix pynative memory leak 5 years ago
resize_bilinear_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
resize_bilinear_cpu_kernel.h clean code in cpu operater 4 years ago
resize_bilinear_grad_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
resize_bilinear_grad_cpu_kernel.h Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
resize_nearest_neighbor_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
resize_nearest_neighbor_cpu_kernel.h add_int64_float64 5 years ago
resize_nearest_neighbor_grad_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
resize_nearest_neighbor_grad_cpu_kernel.h add_int64_float64 5 years ago
rmsprop_cpu_kernel.cc clean code in cpu operater 4 years ago
rmsprop_cpu_kernel.h optimize rmsprop op by simd instruction 4 years ago
roi_align_cpu_kernel.cc adding 9 object detection operators 4 years ago
roi_align_cpu_kernel.h adding 9 object detection operators 4 years ago
roi_align_grad_cpu_kernel.cc adding 9 object detection operators 4 years ago
roi_align_grad_cpu_kernel.h adding 9 object detection operators 4 years ago
rolling_cpu_kernel.cc add rolling cpu kernel 4 years ago
rolling_cpu_kernel.h add rolling cpu kernel 4 years ago
scatter_arithmetic_cpu_kernel.cc clean code in cpu operater 4 years ago
scatter_arithmetic_cpu_kernel.h clean code in cpu operater 4 years ago
scatter_nd_cpu_kernel.cc fixes bug in scatternd 4 years ago
scatter_nd_cpu_kernel.h adding 9 object detection operators 4 years ago
scatter_nd_update_cpu_kernel.cc clean code warning for debug 4 years ago
scatter_nd_update_cpu_kernel.h [CPU] Add TensorScatterUpdate and Print Ops 4 years ago
searchsorted_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
searchsorted_cpu_kernel.h [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
select_cpu_kernel.cc backup select, less, pow, sub CPU ops's performance 4 years ago
select_cpu_kernel.h 告警清理 4 years ago
sgd_cpu_kernel.cc clean code in cpu operater 4 years ago
sgd_cpu_kernel.h clean code in cpu operater 4 years ago
shift_cpu_kernel.cc update shift API and st 4 years ago
shift_cpu_kernel.h update shift API and st 4 years ago
sigmoid_cross_entropy_with_logits_cpu_kernel.cc change infer type to device type for cpu 4 years ago
sigmoid_cross_entropy_with_logits_cpu_kernel.h add_cpu_op 5 years ago
sigmoid_cross_entropy_with_logits_grad_cpu_kernel.cc change infer type to device type for cpu 4 years ago
sigmoid_cross_entropy_with_logits_grad_cpu_kernel.h add_cpu_op 5 years ago
slice_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
slice_cpu_kernel.h acc mae cpu ops 4 years ago
slice_grad_cpu_kernel.cc fix slice and minimum op bug 4 years ago
slice_grad_cpu_kernel.h fix slice op bug 5 years ago
smooth_l1_loss_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
smooth_l1_loss_cpu_kernel.h Refactor some cpu ops 5 years ago
smooth_l1_loss_grad_cpu_kernel.cc clean code for CPU and ModelZoo 4 years ago
smooth_l1_loss_grad_cpu_kernel.h Refactor some cpu ops 5 years ago
sort_cpu_kernel.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
sort_cpu_kernel.h Add CPU Sort OP 4 years ago
spacetodepth_cpu_kernel.cc clean code in cpu operater 4 years ago
spacetodepth_cpu_kernel.h clean code in cpu operater 4 years ago
sparse_apply_adam_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
sparse_apply_adam_cpu_kernel.h Support int64 for cpu sparse optimizers 5 years ago
sparse_apply_ftrl_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
sparse_apply_ftrl_cpu_kernel.h Support int64 for cpu sparse optimizers 5 years ago
sparse_apply_lazy_adam_cpu_kernel.cc Fix exception log and check secure func ret 4 years ago
sparse_apply_lazy_adam_cpu_kernel.h Support int64 for cpu sparse optimizers 5 years ago
sparse_apply_proximal_adagrad_cpu_kernel.cc Fix pclint plus warning 4 years ago
sparse_apply_proximal_adagrad_cpu_kernel.h Support int64 for cpu sparse optimizers 5 years ago
sparse_optimizer_cpu_kernel.h use thread pool for some cpu kernel 5 years ago
sparse_tensor_dense_matmul_cpu_kernel.cc clean code in cpu operater 4 years ago
sparse_tensor_dense_matmul_cpu_kernel.h clean code in cpu operater 4 years ago
sparse_to_dense_cpu_kernal.cc fix pclint warnning 4 years ago
sparse_to_dense_cpu_kernal.h complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul 4 years ago
split_cpu_kernel.cc clean code in cpu operater 4 years ago
split_cpu_kernel.h fix split memory leak 4 years ago
strided_slice_grad_cpu_kernel.cc clean code in cpu operater 4 years ago
strided_slice_grad_cpu_kernel.h clean code in cpu operater 4 years ago
stridedslice_cpu_kernel.cc fix cpu ops bug 4 years ago
stridedslice_cpu_kernel.h convert the implementation of Slice and StridedSlice CPU operators to nnacl 4 years ago
sub_and_filter_cpu_kernel.cc fix vector push_back realloc memory issue 4 years ago
sub_and_filter_cpu_kernel.h fix pynative memory leak 5 years ago
tensor_copy_slices_cpu_kernel.cc !22322 clean code 4 years ago
tensor_copy_slices_cpu_kernel.h Add TensorCopySlice Op 4 years ago
tensoradd_cpu_kernel.cc Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
tensoradd_cpu_kernel.h upload unify cpu ops info library code 4 years ago
tile_cpu_kernel.cc clean code in cpu operater 4 years ago
tile_cpu_kernel.h !16988 upload unify cpu ops info library code 4 years ago
topk_cpu_kernel.cc clean code in cpu operater 4 years ago
topk_cpu_kernel.h change from CRLF to LF 4 years ago
transpose_cpu_kernel.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
transpose_cpu_kernel.h change from CRLF to LF 4 years ago
unique_cpu_kernel.cc fix vector push_back realloc memory issue 4 years ago
unique_cpu_kernel.h cpu unique kernel support fp16 4 years ago
unique_with_pad_cpu_kernel.cc cpu unique kernel support fp16 4 years ago
unique_with_pad_cpu_kernel.h cpu unique kernel support fp16 4 years ago
unpack_cpu_kernel.cc [feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset 4 years ago
unpack_cpu_kernel.h converter the implementation of DropGrad and unpack and transpose CPU operators to nnacl 4 years ago
unsorted_segment_sum_cpu_kernel.cc clean code in cpu operater 4 years ago
unsorted_segment_sum_cpu_kernel.h convert the implementation of tile unsortedSegmentSum CPU ops to nnacl 4 years ago
update_cache_cpu_kernel.cc change infer type to device type for cpu 4 years ago
update_cache_cpu_kernel.h clean pclint 4 years ago