| .. |
|
test_abs_op.py
|
Fix bug: AbsGrad support int as input.
|
2021-11-12 10:46:45 +08:00 |
|
test_acos_grad_op.py
|
Add Asin, ACos, AsinGrad, ACosGrad for CPU
|
2021-01-30 11:11:59 +08:00 |
|
test_acos_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_acosh_grad_op.py
|
fix G.AsinhGrad, G.AcoshGrad on cpu
|
2021-05-18 09:54:23 +08:00 |
|
test_acosh_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_adam_op.py
|
add Adam CPU op
|
2020-11-04 09:29:17 +08:00 |
|
test_adam_weight_decay_op.py
|
fix test adamweightdecay st
|
2021-10-28 09:08:39 +08:00 |
|
test_addn_op.py
|
Support float64 for CPU OP, including Sqrt, Abs, AbsGrad, AddN.
|
2021-11-03 17:05:23 +08:00 |
|
test_apply_adagrad_dad_op.py
|
ApplyAdagradDAD_operator_cpu_support
|
2022-06-10 11:12:10 +08:00 |
|
test_apply_adagrad_op.py
|
Fix: ApplyAdagrad output.
|
2021-03-02 10:41:16 +08:00 |
|
test_apply_adagrad_v2_op.py
|
cpu kernel and st test done, vmap not started
|
2022-05-18 20:38:48 +00:00 |
|
test_apply_gradient_descent_op.py
|
Add cpu version of ApplyGradientDescent op
|
2022-05-18 17:16:04 -04:00 |
|
test_apply_power_sign.py
|
fix apply power sign bug and add st
|
2022-05-05 19:16:17 +08:00 |
|
test_apply_proximal_adagrad_op.py
|
support apply_proximal_adagrad vmap
|
2022-06-07 10:56:56 +08:00 |
|
test_argmax_op.py
|
fix cpu/gpu argmax op
|
2021-03-04 16:42:41 +08:00 |
|
test_argmaxwithvalue_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_argminwithvalue_op.py
|
add CPU operators: Div/Mod/Minimum/Floor/ArgMinWithValue
|
2020-12-26 10:05:12 +08:00 |
|
test_arithmetic_op.py
|
Fix CI probabilistic filter failed in gate building<test_arithmetic_op.py>
|
2022-01-25 20:45:26 +08:00 |
|
test_arithmetic_self_op.py
|
Add dynamic_shape tests.
|
2022-06-01 17:58:33 +08:00 |
|
test_asin_grad_op.py
|
Add Asin, ACos, AsinGrad, ACosGrad for CPU
|
2021-01-30 11:11:59 +08:00 |
|
test_asin_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_asinh_grad_op.py
|
fix G.AsinhGrad, G.AcoshGrad on cpu
|
2021-05-18 09:54:23 +08:00 |
|
test_asinh_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_assert_op.py
|
fix the error of testcase attr
|
2022-05-24 15:18:55 +08:00 |
|
test_assign_add_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_assign_op.py
|
Add assign op for cpu
|
2020-12-05 10:52:07 +08:00 |
|
test_assign_sub_op.py
|
fix the error of testcase attr
|
2022-05-24 15:18:55 +08:00 |
|
test_atan2_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_atan_grad_op.py
|
add sin, cos, tan, atan for cpu
|
2021-02-20 14:19:17 +08:00 |
|
test_atan_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_atanh_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_avgpool_grad_op.py
|
add CPU ops: Sign, NotEqual, Split, AvgPool, AvgPoolGradCpu
|
2020-12-19 16:25:38 +08:00 |
|
test_avgpool_op.py
|
develop maxpool3d, avgpool3d and conv3d on CPU(including forward and backward)
|
2022-03-07 20:44:38 +08:00 |
|
test_batch_matmul.py
|
call nnacl implement for matmul and batchmatmul on ARM
|
2021-06-24 09:29:45 +08:00 |
|
test_batchdot_op.py
|
Bug fix for ISSUE #I3CN9Q
|
2021-03-24 17:12:05 +08:00 |
|
test_batchnorm_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_batchtospacend_op.py
|
batch_to_space_nd
|
2022-06-06 14:21:57 +08:00 |
|
test_bce_with_logits_loss.py
|
add bce_with_logits_loss cpu implement
|
2022-04-14 22:39:07 +08:00 |
|
test_bessel.py
|
修复Bessel i0/k0/k0e精度问题，修复Bessel算子st
|
2022-06-08 23:36:38 +08:00 |
|
test_bias_add.py
|
switch implementation of BiasAdd op to nnacl
|
2021-05-28 14:07:10 +08:00 |
|
test_bias_add_grad.py
|
fix check of input dims for BiasAdd.
|
2021-03-16 10:58:31 +08:00 |
|
test_bidense.py
|
add 1d inputs support for nn.BiDense
|
2022-05-30 10:11:38 +08:00 |
|
test_binary_cross_entropy_op.py
|
binary_cross_entropy op supports optional weight input
|
2021-05-20 10:47:12 +08:00 |
|
test_bitwise_op.py
|
[MS][ops]ajust bitwise ops to new interface
|
2022-04-28 10:51:07 +08:00 |
|
test_boundingbox_decode_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_boundingbox_encode_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_broadcast_to_op.py
|
takedown windows cpu testcases and add windows ST back to gate level0
|
2022-04-26 17:16:57 +08:00 |
|
test_cache_ops.py
|
add security isolate for save_graphs
|
2021-09-14 10:24:12 +08:00 |
|
test_cast_op.py
|
Add full support for cpu cast op
|
2020-12-25 17:10:50 +08:00 |
|
test_cdist_grad_op.py
|
add cdist grad
|
2022-05-07 15:38:06 +08:00 |
|
test_cdist_op.py
|
add cdist
|
2022-04-20 10:47:12 +08:00 |
|
test_celu_op.py
|
support functional tensor vmap for celu
|
2022-06-10 09:12:39 +08:00 |
|
test_check_valid_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_concat_op.py
|
Support vmap for op concat
|
2022-05-23 19:06:00 +08:00 |
|
test_conv2d_backprop_filter_op.py
|
develop maxpool3d, avgpool3d and conv3d on CPU(including forward and backward)
|
2022-03-07 20:44:38 +08:00 |
|
test_conv2d_backprop_input_op.py
|
develop maxpool3d, avgpool3d and conv3d on CPU(including forward and backward)
|
2022-03-07 20:44:38 +08:00 |
|
test_conv_op.py
|
develop maxpool3d, avgpool3d and conv3d on CPU(including forward and backward)
|
2022-03-07 20:44:38 +08:00 |
|
test_cos_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_cosh_op.py
|
add sinh, cosh, asinh, acosh, atanh, atan2, asinhgrad, acoshgrad for cpu
|
2021-02-25 10:07:21 +08:00 |
|
test_cpu_type.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_crop_and_resize_op.py
|
codedex fix
|
2021-09-03 14:14:12 +08:00 |
|
test_ctcloss_op.py
|
mod_conv2d_pool_add_ctcloss
|
2021-01-26 10:30:10 +08:00 |
|
test_ctclossv2_op.py
|
Add CTCLossV2 CPU kernel
|
2022-05-17 16:24:04 +08:00 |
|
test_cum_minmax_op.py
|
Add CPU/GPU implementation of Cummax operator.
|
2022-04-25 12:32:29 +08:00 |
|
test_cumsum_op.py
|
Add CumSum Kerner for CPU
|
2020-12-25 16:30:11 +08:00 |
|
test_dataformatdimmap_op.py
|
add dataformatdimmap
|
2022-06-02 11:28:18 +08:00 |
|
test_deformable_conv.py
|
Add the implement of cpu kernel for deformable_offsets
|
2022-06-08 11:53:52 +08:00 |
|
test_deformable_offsets_grad.py
|
add bprop of deformable_offsets_grad
|
2022-06-01 09:57:01 +08:00 |
|
test_depthtospace_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_dot_op.py
|
Optimize the compile performance in Parser, FG, Manager and Renormalize:
|
2021-03-04 16:01:54 +08:00 |
|
test_dropout_grad_op.py
|
throw exception when tensor has zero in shape
|
2021-03-16 10:39:48 +08:00 |
|
test_dropout_nd_op.py
|
add mish lrn lrn_grad vmap_rule and some test case
|
2022-06-08 16:43:02 +08:00 |
|
test_dropout_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_dynamic_op.py
|
add cpu dynamic memory
|
2021-10-18 14:18:06 +08:00 |
|
test_elu_grad_op.py
|
Add EluGrad for CPU.
|
2021-01-26 15:54:26 +08:00 |
|
test_elu_op.py
|
add elu for cpu
|
2021-01-19 16:58:07 +08:00 |
|
test_embedding_look_up_op.py
|
adapt input to attr between cpu and aicpu embeddinglookup
|
2020-08-11 17:21:49 +08:00 |
|
test_equal_op.py
|
add cpu op equal
|
2020-11-06 12:15:03 +08:00 |
|
test_equalcount_op.py
|
change tensor dtype and shape from function to attr
|
2020-06-12 19:03:23 +08:00 |
|
test_erf_op.py
|
[MS][OP] add erf and erfc op
|
2022-04-25 18:50:41 +08:00 |
|
test_erfc_op.py
|
[MS][OP] add erf and erfc op
|
2022-04-25 18:50:41 +08:00 |
|
test_exp_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_expm1_op.py
|
expm1 operator
|
2020-12-15 20:28:07 +08:00 |
|
test_fast_gelu_grad_op.py
|
fast_gelu_test
|
2022-05-17 17:35:50 +08:00 |
|
test_fast_gelu_op.py
|
[MS][ops]adjust ger cpu operator
|
2022-05-13 15:43:47 +08:00 |
|
test_fused_ada_factor_op.py
|
fused global norm for cpu fused adafactor
|
2022-01-11 14:56:48 +08:00 |
|
test_gather_d_grad_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_gather_d_op.py
|
add cpu operators: Reciprocal/LessEqual/MinimumGrad/GatherD/GatherDGrad
|
2020-12-30 08:10:18 +08:00 |
|
test_gather_op.py
|
gather: add vmap rule for gather operations
|
2022-05-30 16:59:00 +08:00 |
|
test_gathernd_op.py
|
add gather nd interface
|
2022-05-18 18:26:02 +08:00 |
|
test_gelu_grad_op.py
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
test_gelu_op.py
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
test_ger_op.py
|
[MS][ops]adjust ger cpu operator
|
2022-05-13 15:43:47 +08:00 |
|
test_greater_equal_op.py
|
sort op support axis value not -1
|
2022-04-11 09:13:19 +08:00 |
|
test_greater_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_hshrink_grad_op.py
|
[OP] add vmap for meshgrid, hardshrink
|
2022-05-26 14:04:25 +08:00 |
|
test_hshrink_op.py
|
[OP] add function and tensor interface for hard shrink op
|
2022-05-13 09:37:42 +08:00 |
|
test_hsigmoid_op.py
|
add hsigmoid hswis cpu op
|
2021-01-14 14:30:07 +08:00 |
|
test_hswish_op.py
|
add hsigmoid hswis cpu op
|
2021-01-14 14:30:07 +08:00 |
|
test_in_top_k_op.py
|
add intopk op for cpu
|
2022-03-22 11:48:06 +08:00 |
|
test_index_add_op.py
|
fix axis check range
|
2022-06-06 19:38:46 +08:00 |
|
test_inplace_op.py
|
add inplace update
|
2022-06-01 13:50:17 +08:00 |
|
test_inv_grad_op.py
|
Add dynamic_shape tests.
|
2022-06-01 17:58:33 +08:00 |
|
test_iou_op.py
|
add CPU IOU
|
2021-04-17 17:34:25 +08:00 |
|
test_is_close_op.py
|
Add cpu implementation of IsClose operator.
|
2022-04-08 12:44:48 +08:00 |
|
test_isfinite_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_isnan_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_kl_div_op.py
|
add CPU support for KLDivLoss op
|
2022-05-20 15:34:31 +08:00 |
|
test_l1_regularizer_op.py
|
add security isolate for save_graphs
|
2021-09-14 10:24:12 +08:00 |
|
test_l2loss_op.py
|
add CPU l2loss op
|
2021-08-03 20:19:15 +08:00 |
|
test_l2normalize_grad_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_l2normalize_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_layer_norm_grad_op.py
|
add some CPU operators
|
2021-01-15 17:07:39 +08:00 |
|
test_layer_norm_op.py
|
add some CPU operators
|
2021-01-15 17:07:39 +08:00 |
|
test_lerp_op.py
|
add lp_norm vmap and dynamic shape test case.
|
2022-05-28 16:33:05 +08:00 |
|
test_less_equal_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_less_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_lin_space.py
|
Add Linspace functional API && enable batch support && float64 support
|
2022-05-31 16:08:03 +08:00 |
|
test_log_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_logical_op.py
|
Add cpu op LogicalAnd, LogicalOr, LogicalNot
|
2021-01-19 14:14:26 +08:00 |
|
test_logsoftmax_op.py
|
Add half LogSoftmax
|
2022-05-07 14:48:36 +08:00 |
|
test_lp_norm_op.py
|
add lp_norm vmap and dynamic shape test case.
|
2022-05-28 16:33:05 +08:00 |
|
test_lrn_grad_op.py
|
add mish lrn lrn_grad vmap_rule and some test case
|
2022-06-08 16:43:02 +08:00 |
|
test_lrn_op.py
|
add mish lrn lrn_grad vmap_rule and some test case
|
2022-06-08 16:43:02 +08:00 |
|
test_lstm_op.py
|
unify GRU and LSTM operaters and change LSTMCell to right version
|
2021-12-08 16:51:51 +08:00 |
|
test_masked_select_op.py
|
add security isolate for save_graphs
|
2021-09-14 10:24:12 +08:00 |
|
test_maskedfill_op.py
|
add masked fill
|
2022-04-24 14:35:58 +08:00 |
|
test_matmul.py
|
add double type support for matmul cpu op
|
2021-10-21 10:43:21 +08:00 |
|
test_matrix_band_part.py
|
Extend the MatrixBandPart kernel to support batch.
|
2022-06-08 20:45:19 +08:00 |
|
test_matrix_diag_v3_op.py
|
Add GPU implementation of MatrixDiagV3 operator.
|
2022-05-13 14:57:03 +08:00 |
|
test_matrix_solve.py
|
Add MatrixSolve Op
|
2022-06-07 10:44:39 +08:00 |
|
test_maximum_grad.py
|
takedown test_broadcast_grade_cpu
|
2022-01-25 09:46:00 +08:00 |
|
test_maximum_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_maxpool_op.py
|
implement Conv2dBackward and log of pool
|
2022-03-15 21:26:40 +08:00 |
|
test_meshgrid_op.py
|
[OP] add vmap for meshgrid, hardshrink
|
2022-05-26 14:04:25 +08:00 |
|
test_minimum_grad_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_minimum_op.py
|
fix code review alarms
|
2021-04-13 14:56:47 +08:00 |
|
test_mirror_pad.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_mish_grad_op.py
|
Fix the mish_grad test case.
|
2022-06-07 17:18:17 +08:00 |
|
test_mish_op.py
|
add mish lrn lrn_grad vmap_rule and some test case
|
2022-06-08 16:43:02 +08:00 |
|
test_momentum_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_mul_op.py
|
support int32 dtype of Mul cpu op
|
2020-12-15 19:38:28 +08:00 |
|
test_multinomial_op.py
|
add multinomial cpu kernel
|
2021-08-17 14:54:38 +08:00 |
|
test_neg_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_nllloss_op.py
|
develop nllloss and nlllossgrad for CPU and lite
|
2022-03-22 10:35:07 +08:00 |
|
test_nms_with_mask_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_nn_reflectionpad.py
|
fixed reflectionpad bug in st cases of acend and cpu
|
2022-06-07 15:34:33 +08:00 |
|
test_norm_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_notequal_op.py
|
add CPU ops: Sign, NotEqual, Split, AvgPool, AvgPoolGradCpu
|
2020-12-19 16:25:38 +08:00 |
|
test_one_hot_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_onednn_dfx.py
|
onednn dfx testcase skip win and mac
|
2022-02-10 11:07:43 +08:00 |
|
test_oneslike_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_pack_op.py
|
modify pack to stack
|
2021-02-04 18:54:31 +08:00 |
|
test_pad.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_pad_op.py
|
add functional, tensor interface and vmap for ops.Pad
|
2022-06-09 14:26:36 +08:00 |
|
test_padding.py
|
Add dynamic_shape tests.
|
2022-06-01 17:58:33 +08:00 |
|
test_pdist_op.py
|
add pdist_cpu_kernel
|
2022-05-27 09:01:07 +08:00 |
|
test_pow_op.py
|
Add test cases for pow
|
2022-05-18 11:41:11 +08:00 |
|
test_priority_replay_buffer_test.py
|
add destory operator for PriorityReplayBuffer
|
2022-04-18 18:59:09 +08:00 |
|
test_qr_op.py
|
Add the QR decomposition for cpu kernel.
|
2021-10-26 16:31:51 +08:00 |
|
test_random_choice_with_mask_op.py
|
Modifies test case for RandomChoiceWIthMask on mac
|
2021-12-09 11:27:24 +08:00 |
|
test_random_gamma_op.py
|
add python implement and st for random gamma
|
2022-06-01 11:20:36 +08:00 |
|
test_range_op.py
|
add CPU ops: Greater/GreaterEqual/Range/GatherNd for center net
|
2020-12-29 20:07:54 +08:00 |
|
test_rank_op.py
|
add rank cpu kernel
|
2021-09-08 15:02:59 +08:00 |
|
test_realdiv_op.py
|
Fix: cpu real_div support fp16.
|
2021-11-05 10:21:16 +08:00 |
|
test_reduce_op.py
|
fix bug when reduce axis have duplicate elements
|
2022-06-09 21:15:59 +08:00 |
|
test_relu6_grad_op.py
|
convert the implementation of BroadCastTo, ReluGrad, ReLU6Grad CPU operators to nnacl
|
2021-04-22 15:36:26 +08:00 |
|
test_relu6_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_relu_grad_op.py
|
convert the implementation of BroadCastTo, ReluGrad, ReLU6Grad CPU operators to nnacl
|
2021-04-22 15:36:26 +08:00 |
|
test_relu_grad_v2_op.py
|
add reluv2 grad ops
|
2022-04-24 15:45:57 +08:00 |
|
test_relu_op.py
|
Fixing some tiny faults about Pylint in my code(ops)
|
2020-05-25 21:27:45 +08:00 |
|
test_relu_v2_op.py
|
add reluv2 grad ops
|
2022-04-24 15:45:57 +08:00 |
|
test_renorm_op.py
|
add renorm cpu implementation; vmap for renorm
|
2022-06-06 14:23:43 +08:00 |
|
test_resize_bilinear_grad_op.py
|
add_resize_ops
|
2020-12-19 01:06:27 +08:00 |
|
test_resize_bilinear_op.py
|
add half pixel center to resize bilinear
|
2022-03-05 15:43:20 +08:00 |
|
test_resize_linear_1d_op.py
|
add resize linear 1d cpu
|
2022-06-07 15:00:29 +08:00 |
|
test_resize_nearest_neighbor_grad_op.py
|
add_resize_ops
|
2020-12-19 01:06:27 +08:00 |
|
test_resize_nearest_neighbor_op.py
|
add_int64_float64
|
2021-02-26 19:36:24 +08:00 |
|
test_rl_buffer_net.py
|
update buffer sample gpu
|
2021-08-16 14:05:51 +08:00 |
|
test_rl_buffer_op.py
|
update buffer sample gpu
|
2021-08-16 14:05:51 +08:00 |
|
test_rmsprop.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_roi_align_grad_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_roi_align_op.py
|
adding 9 object detection operators
|
2021-07-21 10:49:17 +08:00 |
|
test_rolling_op.py
|
add rolling cpu kernel
|
2021-08-25 11:57:18 +08:00 |
|
test_scatter_add_with_axis_op.py
|
optimize scatter elements performance
|
2022-05-30 15:13:26 +08:00 |
|
test_scatter_arithmetic_op.py
|
add cpu operations: ScatterAdd ScatterSub ScatterMul ScatterDiv ScatterMax ScatterMin ScatterUpdate
|
2021-04-16 16:46:36 +08:00 |
|
test_scatter_element_op.py
|
add scatter elements and scatter add with axis cpu kernel
|
2022-04-27 17:08:56 +08:00 |
|
test_scatter_nd_arithmetic_op.py
|
Add functional doc and interface for ScatterNdMul and ScatterNdMax operator.
|
2022-06-06 20:39:02 +08:00 |
|
test_scatter_nd_op.py
|
Op ScatterNd, rewrite infer function with c++, supports vmap, supports dynamic shape on Gpu and Ascend.
|
2022-06-06 11:44:33 +08:00 |
|
test_scatter_nd_update_op.py
|
takedown scatter_nd_update and sparse_segment_mean testcases
|
2022-06-10 14:32:26 +08:00 |
|
test_searchsorted_op.py
|
adding SearchSorted operator to CPU
|
2021-06-03 10:30:55 +08:00 |
|
test_select_op.py
|
functional select support x or y is a scalar
|
2022-01-05 10:12:44 +08:00 |
|
test_selu_op.py
|
add mish lrn lrn_grad vmap_rule and some test case
|
2022-06-08 16:43:02 +08:00 |
|
test_sgd_op.py
|
added SGD operation in CPU
|
2021-04-14 15:19:10 +08:00 |
|
test_shift_op.py
|
reduce testcase
|
2021-12-21 17:37:21 +08:00 |
|
test_sigmoid_cross_entropy_with_logits_grad_op.py
|
add_cpu_op
|
2020-10-30 13:07:35 +08:00 |
|
test_sigmoid_cross_entropy_with_logits_op.py
|
add_cpu_op
|
2020-10-30 13:07:35 +08:00 |
|
test_sigmoid_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_sign_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_sin_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_sinh_op.py
|
add sinh, cosh, asinh, acosh, atanh, atan2, asinhgrad, acoshgrad for cpu
|
2021-02-25 10:07:21 +08:00 |
|
test_slice_grad_op.py
|
SliceGrad supports 8D input in CPU and GPU
|
2022-05-16 10:06:22 +08:00 |
|
test_slice_op.py
|
fix slice op bug
|
2021-03-24 14:40:56 +08:00 |
|
test_smoothl1loss_op.py
|
add security isolate for save_graphs
|
2021-09-14 10:24:12 +08:00 |
|
test_soft_shrink_op.py
|
softshrink softshrinkgrad
|
2022-05-20 17:09:07 +08:00 |
|
test_softmax_cross_entropy_with_logits_op.py
|
add reduce and softmax_cross_entropy_with_logits
|
2020-06-19 21:37:43 +08:00 |
|
test_softmax_op.py
|
fix softmax
|
2020-09-27 09:53:27 +08:00 |
|
test_softmax_with_cross_entropy_op.py
|
change tensor dtype and shape from function to attr
|
2020-06-12 19:03:23 +08:00 |
|
test_softplus_grad_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_softplus_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_sort.py
|
Add CPU Sort OP
|
2021-07-30 16:21:02 +08:00 |
|
test_spacetobatchnd_op.py
|
move vmap testcases to st/ops directory
|
2022-05-18 16:07:30 +08:00 |
|
test_spacetodepth_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_sparse_apply_adam_op.py
|
modify_bug
|
2020-09-09 18:20:32 +08:00 |
|
test_sparse_apply_ftrl_op.py
|
modify_bug
|
2020-09-09 18:20:32 +08:00 |
|
test_sparse_apply_proximal_adagrad_op.py
|
modify_bug
|
2020-09-09 18:20:32 +08:00 |
|
test_sparse_segment_mean_op.py
|
Add CPU implementation of SparseSegmentMean operator.
|
2022-05-19 09:39:38 +08:00 |
|
test_sparse_tensor_dense_matmul_op.py
|
complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul
|
2021-06-01 17:40:45 +08:00 |
|
test_sparse_to_dense_op.py
|
complete forward and backward of SparseToDense and check of SparseTensorDenseMatmul
|
2021-06-01 17:40:45 +08:00 |
|
test_split_op.py
|
Modify split Kernel for CPU
|
2020-12-31 11:01:46 +08:00 |
|
test_sqrt_op.py
|
Support float64 for CPU OP, including Sqrt, Abs, AbsGrad, AddN.
|
2021-11-03 17:05:23 +08:00 |
|
test_square_op.py
|
[feat][assistant][I48O6D] Add Square operator
|
2022-01-14 15:31:28 +08:00 |
|
test_square_sum_all.py
|
Add dynamic shupport to SquareSumAll op and fix rounding issue in cpu version
|
2022-05-12 11:29:41 -04:00 |
|
test_squared_difference_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
|
test_squeeze_op.py
|
add type support for concat, reshape, transpose, tile, squeeze, expandims cpu op
|
2021-01-06 13:11:46 +08:00 |
|
test_standard_normal.py
|
fix standard_normal test occasional failure
|
2020-12-10 11:23:08 +08:00 |
|
test_stridedslice_grad_op.py
|
fix cpu StridedSliceGrad
|
2020-06-02 19:20:15 +08:00 |
|
test_stridedslice_op.py
|
optimize performance
|
2020-06-04 16:20:51 +08:00 |
|
test_svd_op.py
|
add support on CPU platform for op: svd
|
2022-05-28 15:36:43 +08:00 |
|
test_tan_op.py
|
add sin, cos, tan, atan for cpu
|
2021-02-20 14:19:17 +08:00 |
|
test_tanh_op.py
|
add some cpu operator
|
2020-10-29 14:17:03 +08:00 |
|
test_tensor_array.py
|
update tensorarry stack for static usage
|
2022-02-14 09:14:35 +08:00 |
|
test_tensor_scatter_arithmetic.py
|
tensor scatter div vmap
|
2022-05-28 17:29:53 +08:00 |
|
test_tensoradd.py
|
Change TensorAdd to Add, from r1.1 to master
|
2021-02-01 17:53:52 +08:00 |
|
test_tile_op.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
test_time_distributed_op.py
|
takedown testcases to ensure gate stability
|
2022-03-09 10:18:51 +08:00 |
|
test_topk_op.py
|
add CPU TopK
|
2020-12-28 17:34:52 +08:00 |
|
test_transpose_op.py
|
set default context mode to GRAPH_MODE
|
2021-06-09 14:40:49 +08:00 |
|
test_trunc_op.py
|
Trunc op commit
|
2022-06-03 08:45:54 -04:00 |
|
test_truncate_div_op.py
|
TruncateDiv and TruncateMod CPU ops
|
2022-06-01 22:59:38 -04:00 |
|
test_truncate_mod_op.py
|
TruncateDiv and TruncateMod CPU ops
|
2022-06-01 22:59:38 -04:00 |
|
test_unary_op.py
|
add ceil kernel
|
2022-05-11 16:15:28 +08:00 |
|
test_uniform_int.py
|
add uniform_int and uniform_real on cpu
|
2021-08-16 20:11:18 +08:00 |
|
test_uniform_real.py
|
add uniform_int and uniform_real on cpu
|
2021-08-16 20:11:18 +08:00 |
|
test_unique_op.py
|
cpu unique kernel support fp16
|
2021-04-20 11:40:13 +08:00 |
|
test_unpack_op.py
|
fixed decorators in cpu tests
|
2021-07-21 20:26:21 +08:00 |
|
test_unsorted_segment_arithmetic.py
|
unsorted segment prod gpu inference
|
2022-05-19 18:59:19 +08:00 |
|
test_unsorted_segment_sum.py
|
add CPU unsortedsegmentsum
|
2020-12-17 17:10:12 +08:00 |
|
test_zeroslike_op.py
|
Support float64 for CPU OP, including arithmetic, reshape, scatter_nd_update, select. And Fix floor.cc.
|
2021-10-30 12:41:26 +08:00 |
i-robot
402431e754