| .. |
|
__init__.py
|
add some dynamic ops exp, log, neg, not_equal, reciprocal and support
|
2021-06-28 18:29:47 +08:00 |
|
abs.py
|
tbe select broadcast reduce dynamic
|
2020-06-12 17:41:06 +08:00 |
|
abs_grad.py
|
tbe select broadcast reduce dynamic
|
2020-06-12 17:41:06 +08:00 |
|
accumulate_n_v2.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
acos.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
acos_grad.py
|
为算子进行VM流程对接
|
2020-05-29 16:45:14 +08:00 |
|
acosh.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
acosh_grad.py
|
为算子进行VM流程对接
|
2020-05-29 16:45:14 +08:00 |
|
act_ulq_clamp_max_grad.py
|
Add quant ops.
|
2020-11-04 19:27:49 +08:00 |
|
act_ulq_clamp_min_grad.py
|
Add quant ops.
|
2020-11-04 19:27:49 +08:00 |
|
acts_ulq.py
|
Add quant ops.
|
2020-11-04 19:27:49 +08:00 |
|
acts_ulq_input_grad.py
|
Add quant ops.
|
2020-11-04 19:27:49 +08:00 |
|
adam_apply_one.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
adam_apply_one_assign.py
|
fix bugs and add register info
|
2020-09-04 14:44:15 +08:00 |
|
adam_apply_one_with_decay.py
|
add new mode for operator info register
|
2020-04-03 15:54:12 +08:00 |
|
adam_apply_one_with_decay_assign.py
|
fix bugs and add register info
|
2020-09-04 14:44:15 +08:00 |
|
add_n.py
|
tbe select broadcast reduce dynamic
|
2020-06-12 17:41:06 +08:00 |
|
add_n_ds.py
|
add op dynamic shape
|
2020-12-03 10:08:06 +08:00 |
|
apply_ada_max.py
|
add tbe some ops about optimizer for vm
|
2020-06-11 12:35:00 +08:00 |
|
apply_adadelta.py
|
add tbe some ops about optimizer for vm
|
2020-06-11 12:35:00 +08:00 |
|
apply_adagrad.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
apply_adagrad_v2.py
|
Add ops ReverseSequence for GE.
|
2020-08-13 10:18:32 +08:00 |
|
apply_adam.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
apply_add_sign.py
|
Add optimizer operators for VM.
|
2020-06-29 11:35:45 +08:00 |
|
apply_centered_rms_prop.py
|
fix apply_centered_rms_prop_op_info.
|
2020-12-30 09:27:11 +08:00 |
|
apply_ftrl.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
apply_gradient_descent.py
|
Add optimizer operators for VM.
|
2020-06-29 11:35:45 +08:00 |
|
apply_momentum.py
|
adapt some 3d ops.
|
2021-03-02 14:53:21 +08:00 |
|
apply_power_sign.py
|
Add optimizer operators for VM.
|
2020-06-29 11:35:45 +08:00 |
|
apply_proximal_adagrad.py
|
Adapt ApplyProximalAdagrad and SparseApplyProximalAdagrad
|
2020-06-19 11:49:27 +08:00 |
|
apply_proximal_gradient_descent.py
|
Add optimizer operators for VM.
|
2020-06-29 11:35:45 +08:00 |
|
apply_rms_prop.py
|
add applyrmsprop op for vm
|
2020-05-21 23:02:32 -04:00 |
|
approximate_equal.py
|
Fixing some tiny mistake of InplaceAdd, InplaceSub and InplaceUpdate vm ops and apply new character of dynamic format
|
2020-07-05 11:45:43 +08:00 |
|
arg_max.py
|
fixed arg_max
|
2020-05-19 15:30:30 +08:00 |
|
arg_max_with_value.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
arg_min.py
|
为算子进行VM流程对接
|
2020-05-29 16:45:14 +08:00 |
|
arg_min_with_value.py
|
Fix arg_min_with_value bug
|
2020-04-15 14:39:05 +08:00 |
|
asin.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
asin_grad.py
|
add vm ops: asin asinh asingrad asinhgrad
|
2020-05-28 14:58:23 +08:00 |
|
asinh.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
asinh_grad.py
|
add vm ops: asin asinh asingrad asinhgrad
|
2020-05-28 14:58:23 +08:00 |
|
assign.py
|
adapt assign assignAdd relu6
|
2020-05-06 21:20:32 +08:00 |
|
assign_add.py
|
upgrade 0204
|
2021-02-06 10:51:43 +08:00 |
|
assign_sub.py
|
fix assign sub return value 0
|
2020-05-09 18:00:16 +08:00 |
|
atan.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
atan2.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
atan_grad.py
|
add register for Atan AtanGrad Atanh
|
2020-05-30 15:08:28 +08:00 |
|
atanh.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
atomic_addr_clean.py
|
fix atomic clean
|
2020-06-22 14:27:25 +08:00 |
|
avg_pool.py
|
add op's attr default value
|
2021-03-10 14:43:45 +08:00 |
|
avg_pool_3d.py
|
support avgpool3d
|
2021-05-22 14:44:11 +08:00 |
|
avg_pool_3d_grad.py
|
support avgpool3d
|
2021-05-22 14:44:11 +08:00 |
|
avg_pool_grad.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
avg_pool_grad_vm.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
basic_lstm_cell.py
|
support new format frac_zn_lstm
|
2020-09-08 14:38:27 +08:00 |
|
basic_lstm_cell_c_state_grad.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
basic_lstm_cell_c_state_grad_v2.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
basic_lstm_cell_input_grad.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
basic_lstm_cell_weight_grad.py
|
support new format frac_zn_lstm
|
2020-09-08 14:38:27 +08:00 |
|
batch_matmul.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
batch_matmul_ds.py
|
transdata/cast pass and relu et support dynamicshape
|
2021-06-16 15:45:48 +08:00 |
|
batch_to_space.py
|
support float32
|
2020-05-07 17:13:16 +08:00 |
|
batch_to_space_nd.py
|
fixed SpaceToBatchND, Conv2DBackpropFilter
|
2020-06-30 14:48:11 +08:00 |
|
batchnorm.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
batchnorm_grad.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
bce_with_logits_loss.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
bessel_i0e.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
bessel_i1e.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
bias_add.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
bias_add_grad.py
|
fix op_info of bias_add_grad.py
|
2021-03-08 11:38:07 +08:00 |
|
binary_cross_entropy.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
binary_cross_entropy_grad.py
|
add vms for binarycrossentropy and binarycrossentropygrad
|
2020-05-20 20:28:56 +08:00 |
|
bitwise_and.py
|
Add ops ReverseSequence for GE.
|
2020-08-13 10:18:32 +08:00 |
|
bitwise_or.py
|
Add ops ReverseSequence for GE.
|
2020-08-13 10:18:32 +08:00 |
|
bitwise_xor.py
|
Add ops ReverseSequence for GE.
|
2020-08-13 10:18:32 +08:00 |
|
bn_infer.py
|
add tbe op info of BNInfer and BNInferGrad for VM
|
2020-05-14 19:35:50 +08:00 |
|
bn_infer_grad.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
bn_inference.py
|
Add register information of BNInference.
|
2020-07-22 17:03:39 +08:00 |
|
bn_training_reduce.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
bn_training_reduce_grad.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
bn_training_update.py
|
Support specifying reshape type for batchnorm fused op
|
2020-05-09 17:44:36 +08:00 |
|
bn_training_update_grad.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
bn_training_update_v2.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
bn_training_update_v3.py
|
add bnll, bninference and bntrainingupdatev3 for vm
|
2020-06-01 07:23:12 -04:00 |
|
bounding_box_decode.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
bounding_box_encode.py
|
add op BoundingBoxDecode
|
2020-05-16 02:40:16 -04:00 |
|
broadcast_to.py
|
fix 4 issue
|
2020-08-08 11:46:41 +08:00 |
|
cast.py
|
add checksupport func
|
2021-01-18 16:59:52 +08:00 |
|
cast_ds.py
|
add checksupport func
|
2021-01-18 16:59:52 +08:00 |
|
ceil.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
centralization.py
|
[Less BN]New add FN, GC optimizer.
|
2021-04-14 11:29:18 +08:00 |
|
check_valid.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
clip_by_norm_no_div_sum.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
clip_by_value.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
concat.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
concat_ds.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
confusion_matrix.py
|
add vm support for ConfusionMatrix
|
2020-06-04 14:12:13 +08:00 |
|
confusion_mul_grad.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
confusion_softmax_grad.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
confusion_transpose_d.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
conv2d.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
conv2d_backprop_filter.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
conv2d_backprop_input.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
conv3d.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
conv3d_backprop_filter.py
|
unify Conv's attr pad_list
|
2021-01-21 15:31:28 +08:00 |
|
conv3d_backprop_input.py
|
unify Conv's attr pad_list
|
2021-01-21 15:31:28 +08:00 |
|
conv3d_transpose.py
|
add op's attr default value
|
2021-03-10 14:43:45 +08:00 |
|
cos.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
cosh.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
cum_sum.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
cumprod.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
data_format_dim_map.py
|
add vm for histogramfixedwidth and dataformatdimmap
|
2020-06-12 09:34:58 +08:00 |
|
depth_to_space.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
depthwise_conv2d.py
|
auto tune stage one: construct json
|
2021-03-03 10:59:01 +08:00 |
|
depthwise_conv2d_backprop_filter.py
|
unify Conv's attr pad_list
|
2021-01-21 15:31:28 +08:00 |
|
depthwise_conv2d_backprop_input.py
|
unify Conv's attr pad_list
|
2021-01-21 15:31:28 +08:00 |
|
div.py
|
fix neg output of Div.
|
2021-04-20 15:44:21 +08:00 |
|
div_ds.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
div_no_nan.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
dropout_do_mask.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
dropout_do_mask_ds.py
|
dropout support dynamic shape
|
2021-05-26 11:08:02 +08:00 |
|
dynamic_gru_v2.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
dynamic_rnn.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
elu.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
elu_grad.py
|
add ops for VM
|
2020-05-07 10:44:51 +08:00 |
|
equal.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
equal_ds.py
|
row split dynamic shape infer function
|
2020-11-09 15:38:30 +08:00 |
|
erf.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
erfc.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
exp.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
exp_ds.py
|
add some dynamic ops exp, log, neg, not_equal, reciprocal and support
|
2021-06-28 18:29:47 +08:00 |
|
expand_dims.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
expm1.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
extract_image_patches.py
|
Add ExtractImagePatches for new backend.
|
2020-09-12 10:29:36 +08:00 |
|
fake_quant_with_min_max_vars.py
|
add fake-quant operators.
|
2020-10-16 15:50:18 +08:00 |
|
fake_quant_with_min_max_vars_gradient.py
|
add fake-quant operators.
|
2020-10-16 15:50:18 +08:00 |
|
fake_quant_with_min_max_vars_per_channel.py
|
add fake-quant operators.
|
2020-10-16 15:50:18 +08:00 |
|
fake_quant_with_min_max_vars_per_channel_gradient.py
|
add fake-quant operators.
|
2020-10-16 15:50:18 +08:00 |
|
fast_gelu.py
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
fast_gelu_grad.py
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
fill.py
|
add checksupport func
|
2021-01-18 16:59:52 +08:00 |
|
fill_ds.py
|
add op dynamic shape
|
2020-12-03 10:08:06 +08:00 |
|
flatten.py
|
add optional axis attr for flatten and update example of PSNR.
|
2021-07-01 17:33:32 +08:00 |
|
flatten_grad.py
|
use reshape as flatten grad
|
2020-05-25 16:57:26 +08:00 |
|
floor.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
floor_div.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
floor_mod.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
fused_mul_add.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
fused_mul_add_n.py
|
fix bugs
|
2020-11-07 16:24:29 +08:00 |
|
fused_mul_add_n_l2loss.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
fused_mul_apply_momentum.py
|
adapte FusedMulApplyMomentum
|
2020-05-07 20:22:55 +08:00 |
|
fused_mul_apply_momentum_extern.py
|
support vm for range
|
2020-06-10 15:51:35 +08:00 |
|
gather_nd.py
|
add ops info for VM
|
2020-05-05 18:19:34 +08:00 |
|
gather_v2.py
|
Change GatherV2 to Gather r1.1 to master
|
2021-01-30 14:13:47 +08:00 |
|
gather_v2_ds.py
|
Change GatherV2 to Gather r1.1 to master
|
2021-01-30 14:13:47 +08:00 |
|
gelu.py
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
gelu_grad.py
|
modify Gelu、FastGelu to GeLU and FastGeLU
|
2021-02-05 17:19:52 +08:00 |
|
greater.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
greater_equal.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
gru_v2_hidden_grad.py
|
Adapt DynamicGRUV2 forward for Ascend new backend.
|
2020-11-13 14:54:40 +08:00 |
|
histogram_fixed_width.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
ifmr.py
|
Add IFMR op for new backend.
|
2020-09-22 10:12:12 +08:00 |
|
in_top_k.py
|
update the registration info of InTopK operator.
|
2021-02-08 14:36:53 +08:00 |
|
inplace_add.py
|
add InplaceAddD and InplaceSubD vm ops
|
2020-06-16 09:29:11 +08:00 |
|
inplace_sub.py
|
add InplaceAddD and InplaceSubD vm ops
|
2020-06-16 09:29:11 +08:00 |
|
inplace_update.py
|
add validation of ApplyMomentum and update annotation of Broadcast, DistributedGradReducer, GlobalBatchNorm, etc. operators.
|
2021-01-21 17:22:15 +08:00 |
|
inv.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
inv_grad.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
invert.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
iou.py
|
add op BoundingBoxDecode
|
2020-05-16 02:40:16 -04:00 |
|
l2_loss.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
l2_normalize.py
|
为L2Normalize/L2NormalizeGrad增加VM支持
|
2020-05-12 16:46:02 +08:00 |
|
l2_normalize_grad.py
|
fix spell error
|
2020-06-13 13:23:28 +08:00 |
|
lamb_apply_optimizer_assign.py
|
add tbe fusion operators LambApplyOptimizerAssign and LambApplyWeightAssign for lamb optimizer
|
2020-12-31 08:26:11 +08:00 |
|
lamb_apply_weight_assign.py
|
add tbe fusion operators LambApplyOptimizerAssign and LambApplyWeightAssign for lamb optimizer
|
2020-12-31 08:26:11 +08:00 |
|
lamb_next_mv.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
lamb_next_mv_with_decay.py
|
adapt LambNextMVWithDecay
|
2020-05-08 11:36:08 +08:00 |
|
lamb_next_right.py
|
support vm for range
|
2020-06-10 15:51:35 +08:00 |
|
lamb_update_with_lr.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
lamb_update_with_lr_v2.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
lars_update.py
|
add op BoundingBoxDecode
|
2020-05-16 02:40:16 -04:00 |
|
layer_norm.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
layer_norm_beta_gamma_backprop.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
layer_norm_beta_gamma_backprop_ds.py
|
layernorm support dynamic shape
|
2021-05-25 14:44:32 +08:00 |
|
layer_norm_ds.py
|
layernorm support dynamic shape
|
2021-05-25 14:44:32 +08:00 |
|
layer_norm_grad.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
layer_norm_x_backprop.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
layer_norm_x_backprop_ds.py
|
layernorm support dynamic shape
|
2021-05-25 14:44:32 +08:00 |
|
less.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
less_equal.py
|
add full name in tune result file
|
2021-03-13 10:07:16 +08:00 |
|
lin_space.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
log.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
log1p.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
log_ds.py
|
add some dynamic ops exp, log, neg, not_equal, reciprocal and support
|
2021-06-28 18:29:47 +08:00 |
|
logical_and.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
logical_not.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
logical_or.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
logsoftmax.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
logsoftmax_grad.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
lrn.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
lrn_grad.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
lstm_input_grad.py
|
dynamic_rnn for new backend opensource.
|
2020-09-14 19:18:15 +08:00 |
|
matmul.py
|
add op's attr default value
|
2021-03-10 14:43:45 +08:00 |
|
matmul_ds.py
|
transdata/cast pass and relu et support dynamicshape
|
2021-06-16 15:45:48 +08:00 |
|
matrix_diag.py
|
vm for MatrixDiag,MatrixDiagPart.MatrixSetDiag
|
2020-06-22 20:24:51 +08:00 |
|
matrix_diag_part.py
|
vm for MatrixDiag,MatrixDiagPart.MatrixSetDiag
|
2020-06-22 20:24:51 +08:00 |
|
matrix_set_diag.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
max_pool.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
max_pool3d.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
max_pool3d_grad.py
|
Adapt MaxPool3DGrad
|
2021-04-14 16:04:56 +08:00 |
|
max_pool3d_grad_grad.py
|
Add MaxPool3D,MaxPool3DGrad,MaxPool3DGradGrad ops for Ascend.
|
2021-03-10 15:33:07 +08:00 |
|
max_pool_grad.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
max_pool_grad_grad.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
max_pool_grad_grad_with_argmax.py
|
update Pool's attr kernel_size, pad_mode
|
2021-01-18 12:10:20 +08:00 |
|
max_pool_grad_with_argmax.py
|
Add need_check_supported for maxpoolwithargmax.
|
2021-03-20 09:38:41 +08:00 |
|
max_pool_with_argmax.py
|
Add need_check_supported for maxpoolwithargmax.
|
2021-03-20 09:38:41 +08:00 |
|
maximum.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
maximum_grad.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
minimum.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
minimum_ds.py
|
row split dynamic shape infer function
|
2020-11-09 15:38:30 +08:00 |
|
minimum_grad.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
mish.py
|
add_mish_mulnonan_selu_operations
|
2021-01-22 10:31:31 +08:00 |
|
mod.py
|
Fixing some tiny mistake of InplaceAdd, InplaceSub and InplaceUpdate vm ops and apply new character of dynamic format
|
2020-07-05 11:45:43 +08:00 |
|
mul.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
mul_ds.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
mul_no_nan.py
|
add_mish_mulnonan_selu_operations
|
2021-01-22 10:31:31 +08:00 |
|
neg.py
|
fix bugs
|
2020-10-10 17:48:26 +08:00 |
|
neg_ds.py
|
add some dynamic ops exp, log, neg, not_equal, reciprocal and support
|
2021-06-28 18:29:47 +08:00 |
|
nll_loss.py
|
add default value of optional attribute for nllloss.
|
2021-05-28 14:56:50 +08:00 |
|
nll_loss_grad.py
|
add default value of optional attribute for nllloss.
|
2021-05-28 14:56:50 +08:00 |
|
nms_with_mask.py
|
add op BoundingBoxDecode
|
2020-05-16 02:40:16 -04:00 |
|
not_equal.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
not_equal_ds.py
|
add some dynamic ops exp, log, neg, not_equal, reciprocal and support
|
2021-06-28 18:29:47 +08:00 |
|
npu_alloc_float_status.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
npu_clear_float_status.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
npu_get_float_status.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
one_hot.py
|
add op's attr default value
|
2021-03-10 14:43:45 +08:00 |
|
one_hot_ds.py
|
onehot and softmax support dynamic shape
|
2021-05-25 12:33:40 +08:00 |
|
ones_like.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
ones_like_ds.py
|
ctcloss support dynamic shape
|
2021-05-18 20:31:34 +08:00 |
|
pack.py
|
change need_check_supported of pack register info from True to False.
|
2021-06-28 19:33:25 +08:00 |
|
pad_d.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
pad_d_ds.py
|
gather_v2 pad optimizer pass
|
2020-11-26 11:48:53 +08:00 |
|
parallel_concat.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
population_count.py
|
vm for PopulationCount
|
2020-07-03 17:47:11 +08:00 |
|
pow.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
prelu.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
prelu_grad.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
range.py
|
support vm for range
|
2020-06-10 15:51:35 +08:00 |
|
real_div.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
real_div_ds.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
reciprocal.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
reciprocal_ds.py
|
add some dynamic ops exp, log, neg, not_equal, reciprocal and support
|
2021-06-28 18:29:47 +08:00 |
|
reciprocal_grad.py
|
add 4 grad ops
|
2020-08-03 14:31:30 +08:00 |
|
reduce_all.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
reduce_any.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
reduce_max.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
reduce_mean.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
reduce_min.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
reduce_prod.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
reduce_sum.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
relu.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
relu6.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
relu6_grad.py
|
get input info of fusionop by visitkernel
|
2020-04-30 14:18:41 +08:00 |
|
relu_ds.py
|
support forward unique
|
2020-11-18 15:22:14 +08:00 |
|
relu_grad.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
relu_grad_ds.py
|
transdata/cast pass and relu et support dynamicshape
|
2021-06-16 15:45:48 +08:00 |
|
relu_grad_v2.py
|
tbe select broadcast reduce dynamic
|
2020-06-12 17:41:06 +08:00 |
|
relu_grad_v2_ds.py
|
transdata/cast pass and relu et support dynamicshape
|
2021-06-16 15:45:48 +08:00 |
|
relu_v2.py
|
Add ReluV2/ReluGradV2/ConfusionMulGrad for VM
|
2020-04-22 21:09:37 +08:00 |
|
relu_v2_ds.py
|
transdata/cast pass and relu et support dynamicshape
|
2021-06-16 15:45:48 +08:00 |
|
reshape.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
resize_bilinear.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
resize_bilinear_grad.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
resize_nearest_neighbor.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
resize_nearest_neighbor_grad.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
reverse_v2_d.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
rint.py
|
add 5 vm op
|
2020-07-21 11:34:10 +08:00 |
|
roi_align.py
|
Add attr in ROIAlign.
|
2020-07-10 16:17:51 +08:00 |
|
roi_align_grad.py
|
add register info for ROIAlign and ROIAlignGrad
|
2020-05-15 14:58:33 +08:00 |
|
round.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
rsqrt.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
rsqrt_grad.py
|
add 4 grad ops
|
2020-08-03 14:31:30 +08:00 |
|
scatter_add.py
|
complement scatter add support dtype
|
2020-06-28 16:51:53 +08:00 |
|
scatter_add_ds.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
scatter_div.py
|
ScatterMull and ScatterDiv vm
|
2020-07-01 15:49:24 +08:00 |
|
scatter_max.py
|
add ScatterMax ScatterMin ScatterSub for vm
|
2020-06-29 17:21:08 +08:00 |
|
scatter_min.py
|
add ScatterMax ScatterMin ScatterSub for vm
|
2020-06-29 17:21:08 +08:00 |
|
scatter_mul.py
|
ScatterMull and ScatterDiv vm
|
2020-07-01 15:49:24 +08:00 |
|
scatter_nd.py
|
dock FloorMod GreaterEqual NotEqual ScatterNdUpdate
|
2020-04-27 19:48:16 +08:00 |
|
scatter_nd_add.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
scatter_nd_d.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
scatter_nd_sub.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
scatter_nd_update.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
scatter_non_aliasing_add.py
|
add 5 vm op
|
2020-07-21 11:34:10 +08:00 |
|
scatter_sub.py
|
add ScatterMax ScatterMin ScatterSub for vm
|
2020-06-29 17:21:08 +08:00 |
|
scatter_update.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
scatter_update_ds.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
select.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
selu.py
|
update_dump_map
|
2021-01-25 11:26:35 +08:00 |
|
sgd.py
|
Adapt some ops for 3d format.
|
2021-03-04 23:15:28 +08:00 |
|
sigmoid.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
sigmoid_cross_entropy_with_logits.py
|
adapt some 3d ops.
|
2021-03-02 14:53:21 +08:00 |
|
sigmoid_cross_entropy_with_logits_grad.py
|
adapt some 3d ops.
|
2021-03-02 14:53:21 +08:00 |
|
sigmoid_grad.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
sign.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
sin.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
sinh.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
slice.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
smooth_l1_loss.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
smooth_l1_loss_grad.py
|
update some op's attr name
|
2021-01-20 09:16:43 +08:00 |
|
softmax.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
softmax_cross_entropy_with_logits.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
softmax_ds.py
|
onehot and softmax support dynamic shape
|
2021-05-25 12:33:40 +08:00 |
|
softmax_grad_ext.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
softplus.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
softplus_grad.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
softsign.py
|
Fixing some tiny mistake of InplaceAdd, InplaceSub and InplaceUpdate vm ops and apply new character of dynamic format
|
2020-07-05 11:45:43 +08:00 |
|
sort.py
|
Add ExtractImagePatches for new backend.
|
2020-09-12 10:29:36 +08:00 |
|
space_to_batch.py
|
support float32
|
2020-05-07 17:13:16 +08:00 |
|
space_to_batch_nd.py
|
fixed SpaceToBatchND, Conv2DBackpropFilter
|
2020-06-30 14:48:11 +08:00 |
|
space_to_depth.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
sparse_apply_adagrad.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_apply_adagrad_v2.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_apply_ftrl_d.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_apply_ftrl_d_ds.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_apply_ftrl_v2.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_apply_proximal_adagrad.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_apply_proximal_adagrad_ds.py
|
Modify register of TBE ops.
|
2021-01-12 16:13:09 +08:00 |
|
sparse_gather_v2.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
sparse_gather_v2_ds.py
|
register gatherv2
|
2020-10-29 22:08:46 +08:00 |
|
split_d.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
splitv.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
sqrt.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
sqrt_ds.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
sqrt_grad.py
|
add 4 grad ops
|
2020-08-03 14:31:30 +08:00 |
|
sqrt_grad_ds.py
|
support SqrtGrad dynamic shape
|
2020-12-01 10:33:20 +08:00 |
|
square.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
square_ds.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
square_sum_all.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
square_sum_v1.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
square_sum_v2.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
squared_difference.py
|
Add TBE op SquaredDifference for VM.
|
2020-08-03 18:52:18 +08:00 |
|
squeeze.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
strided_read.py
|
support StridedWrite StridedRead
|
2020-06-09 11:47:18 +08:00 |
|
strided_slice_d.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
strided_slice_grad_d.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
strided_write.py
|
support StridedWrite StridedRead
|
2020-06-09 11:47:18 +08:00 |
|
sub.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
sub_ds.py
|
row split dynamic shape infer function
|
2020-11-09 15:38:30 +08:00 |
|
tan.py
|
Add TBE ops Tan/TruncateDiv/TruncateMod for VM.
|
2020-07-14 16:27:01 +08:00 |
|
tanh.py
|
Support dynamic broadcast reduce format with operator
|
2020-07-02 10:33:10 +08:00 |
|
tanh_grad.py
|
Add all other tbe op info register
|
2020-04-14 17:37:56 +08:00 |
|
tensor_add.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
tensor_add_ds.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
tensor_move.py
|
add ops TensorMove
|
2020-07-10 11:19:59 +08:00 |
|
tensor_scatter_update.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
tile.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
top_k.py
|
change the sequence of register info of topk.
|
2021-05-27 10:44:04 +08:00 |
|
trans_data.py
|
replace tbe build api
|
2021-05-21 09:53:05 +08:00 |
|
trans_data_ds.py
|
transdata/cast pass and relu et support dynamicshape
|
2021-06-16 15:45:48 +08:00 |
|
transpose_d.py
|
fix check of input dims for BiasAdd.
|
2021-03-16 10:58:31 +08:00 |
|
truncate_div.py
|
Add TBE ops Tan/TruncateDiv/TruncateMod for VM.
|
2020-07-14 16:27:01 +08:00 |
|
truncate_mod.py
|
Add TBE ops Tan/TruncateDiv/TruncateMod for VM.
|
2020-07-14 16:27:01 +08:00 |
|
unpack.py
|
add default value for optional attribute.
|
2021-05-21 17:40:16 +08:00 |
|
unsorted_segment_max.py
|
add checksupport func
|
2021-01-18 16:59:52 +08:00 |
|
unsorted_segment_min.py
|
add checksupport func
|
2021-01-18 16:59:52 +08:00 |
|
unsorted_segment_prod.py
|
add checksupport func
|
2021-01-18 16:59:52 +08:00 |
|
unsorted_segment_sum.py
|
fix is dynamic format api description
|
2021-04-22 17:26:40 +08:00 |
|
unsorted_segment_sum_ds.py
|
Ascend Dynamic Shape
|
2020-10-10 19:11:47 +08:00 |
|
wts_arq.py
|
add default value for optional attribute.
|
2021-05-24 17:29:54 +08:00 |
|
xdivy.py
|
Add TBE op SquaredDifference for VM.
|
2020-08-03 18:52:18 +08:00 |
|
xlogy.py
|
Add TBE op SquaredDifference for VM.
|
2020-08-03 18:52:18 +08:00 |
|
zeros_like.py
|
Support dynamic broadcast reduce format with operator of ZerosLike ops vm
|
2020-06-22 14:57:40 +08:00 |
|
zeros_like_ds.py
|
ctcloss support dynamic shape
|
2021-05-18 20:31:34 +08:00 |
f227d6ffb0