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.
 
 
 
 
 
 
zhaozhenlong 30b93ecbf8 use reshape as flatten grad 5 years ago
..
__init__.py use reshape as flatten grad 5 years ago
abs.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 6 years ago
abs_grad.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 6 years ago
adam_apply_one.py add ops info for VM 5 years ago
adam_apply_one_with_decay.py add new mode for operator info register 6 years ago
add.py Add all other tbe op info register 6 years ago
add_n.py Add all other tbe op info register 6 years ago
apply_adam.py Add all other tbe op info register 6 years ago
apply_ftrl.py code 516 for test 5 years ago
apply_momentum.py code 516 for test 5 years ago
apply_rms_prop.py add applyrmsprop op for vm 5 years ago
arg_max.py fixed arg_max 5 years ago
arg_max_with_value.py Add all other tbe op info register 6 years ago
arg_min_with_value.py Fix arg_min_with_value bug 6 years ago
assign.py adapt assign assignAdd relu6 5 years ago
assign_add.py adapt assign assignAdd relu6 5 years ago
assign_sub.py fix assign sub return value 0 5 years ago
atomic_addr_clean.py Add all other tbe op info register 6 years ago
avg_pool.py dock AcoshGrad for GE and AvgPool AvgPoolGrad for Vm 5 years ago
avg_pool_grad.py dock AcoshGrad for GE and AvgPool AvgPoolGrad for Vm 5 years ago
batch_matmul.py Add all other tbe op info register 6 years ago
batch_to_space.py support float32 5 years ago
batchnorm.py add apply_momentum 5 years ago
batchnorm_grad.py fixed batch_norm_grad 5 years ago
bias_add.py Add all other tbe op info register 6 years ago
bias_add_grad.py handle-get-summary-node 5 years ago
binary_cross_entropy.py add vms for binarycrossentropy and binarycrossentropygrad 5 years ago
binary_cross_entropy_grad.py add vms for binarycrossentropy and binarycrossentropygrad 5 years ago
bn_infer.py add tbe op info of BNInfer and BNInferGrad for VM 5 years ago
bn_infer_grad.py add tbe op info of BNInfer and BNInferGrad for VM 5 years ago
bn_training_reduce.py Support specifying reshape type for batchnorm fused op 5 years ago
bn_training_reduce_grad.py Support specifying reshape type for batchnorm fused op 5 years ago
bn_training_update.py Support specifying reshape type for batchnorm fused op 5 years ago
bn_training_update_grad.py Support specifying reshape type for batchnorm fused op 5 years ago
bn_training_update_v2.py support vm for bn_training_update_v2 5 years ago
bounding_box_decode.py add op BoundingBoxDecode 5 years ago
bounding_box_encode.py add op BoundingBoxDecode 5 years ago
cast.py auto mix precision 6 years ago
check_valid.py add op BoundingBoxDecode 5 years ago
clip_by_norm_no_div_sum.py Add all other tbe op info register 6 years ago
clip_by_value.py Add all other tbe op info register 6 years ago
concat.py Add all other tbe op info register 6 years ago
confusion_mul_grad.py Add ReluV2/ReluGradV2/ConfusionMulGrad for VM 6 years ago
confusion_softmax_grad.py Add all other tbe op info register 6 years ago
confusion_transpose_d.py Add all other tbe op info register 6 years ago
conv2d.py Add all other tbe op info register 6 years ago
conv2d_backprop_filter.py fix attribute mapping when docking open source operators 6 years ago
conv2d_backprop_input.py add attr for Conv2dBackpropInput 5 years ago
cos.py add vm support sin and cumsum 5 years ago
cum_sum.py add vm support sin and cumsum 5 years ago
cumprod.py add applyrmsprop op for vm 5 years ago
depth_to_space.py support vm for space_to_depth 5 years ago
depthwise_conv2d.py dock DepthwiseConv2dBackprop DepthwiseConv2dBackpropFilter DepthwiseConv2dBackpropInput 6 years ago
depthwise_conv2d_backprop_filter.py fix attribute mapping when docking open source operators 6 years ago
depthwise_conv2d_backprop_input.py fix attribute mapping when docking open source operators 6 years ago
div.py Add all other tbe op info register 6 years ago
dropout_do_mask.py Add all other tbe op info register 6 years ago
elu.py add ops for VM 5 years ago
elu_grad.py add ops for VM 5 years ago
equal.py Add all other tbe op info register 6 years ago
erf.py Add Erf\FillD operator for VM 6 years ago
erfc.py vm for erfc 5 years ago
exp.py Add all other tbe op info register 6 years ago
expand_dims.py Add all other tbe op info register 6 years ago
fill.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 5 years ago
flatten.py support vm for flatten 5 years ago
flatten_grad.py use reshape as flatten grad 5 years ago
floor.py support vm for floor 5 years ago
floor_div.py Add all other tbe op info register 6 years ago
floor_mod.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 5 years ago
fused_mul_add.py Add all other tbe op info register 6 years ago
fused_mul_add_n.py Add all other tbe op info register 6 years ago
fused_mul_apply_momentum.py adapte FusedMulApplyMomentum 5 years ago
gather_nd.py add ops info for VM 5 years ago
gather_v2.py Add all other tbe op info register 6 years ago
gelu.py Add all other tbe op info register 6 years ago
gelu_grad.py Add all other tbe op info register 6 years ago
greater.py Add all other tbe op info register 6 years ago
greater_equal.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 5 years ago
iou.py add op BoundingBoxDecode 5 years ago
l2_loss.py Add L2Loss op for VM 6 years ago
l2_normalize.py 为L2Normalize/L2NormalizeGrad增加VM支持 5 years ago
l2_normalize_grad.py 为L2Normalize/L2NormalizeGrad增加VM支持 5 years ago
lamb_next_mv.py Add all other tbe op info register 6 years ago
lamb_next_mv_with_decay.py adapt LambNextMVWithDecay 5 years ago
lamb_update_with_lr.py Add all other tbe op info register 6 years ago
lamb_update_with_lr_v2.py Add all other tbe op info register 6 years ago
lars_update.py add op BoundingBoxDecode 5 years ago
layer_norm.py add epsilon parameter for layernorm 5 years ago
layer_norm_beta_gamma_backprop.py Add all other tbe op info register 6 years ago
layer_norm_grad.py Add all other tbe op info register 6 years ago
layer_norm_x_backprop.py Add all other tbe op info register 6 years ago
less.py Add all other tbe op info register 6 years ago
less_equal.py Add all other tbe op info register 6 years ago
log.py Add all other tbe op info register 6 years ago
log1p.py support vm for log1p 5 years ago
logical_and.py Add all other tbe op info register 6 years ago
logical_not.py Add all other tbe op info register 6 years ago
logical_or.py Add all other tbe op info register 6 years ago
logsoftmax.py Add all other tbe op info register 6 years ago
logsoftmax_grad.py Add all other tbe op info register 6 years ago
matmul.py Add all other tbe op info register 6 years ago
max_pool.py Add all other tbe op info register 6 years ago
max_pool_grad.py Add all other tbe op info register 6 years ago
max_pool_grad_with_argmax.py Add all other tbe op info register 6 years ago
max_pool_with_argmax.py Add all other tbe op info register 6 years ago
maximum.py Add all other tbe op info register 6 years ago
maximum_grad.py Add all other tbe op info register 6 years ago
minimum.py Add all other tbe op info register 6 years ago
minimum_grad.py Add all other tbe op info register 6 years ago
mul.py Add all other tbe op info register 6 years ago
neg.py Add all other tbe op info register 6 years ago
nms_with_mask.py add op BoundingBoxDecode 5 years ago
not_equal.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 5 years ago
npu_alloc_float_status.py Add all other tbe op info register 6 years ago
npu_clear_float_status.py Add all other tbe op info register 6 years ago
npu_get_float_status.py Add all other tbe op info register 6 years ago
one_hot.py Add all other tbe op info register 6 years ago
ones_like.py docking OnesLike for open source process 5 years ago
pack.py support vm for pack 5 years ago
pad_d.py Add all other tbe op info register 6 years ago
pow.py Add all other tbe op info register 6 years ago
prelu.py vm for prelu and prelugrad 5 years ago
prelu_grad.py vm for prelu and prelugrad 5 years ago
real_div.py Add all other tbe op info register 6 years ago
reciprocal.py Add all other tbe op info register 6 years ago
reduce_max.py Add all other tbe op info register 6 years ago
reduce_mean.py support buffer fusion 5 years ago
reduce_mean_d.py Add all other tbe op info register 6 years ago
reduce_min.py Add all other tbe op info register 6 years ago
reduce_prod.py add applyrmsprop op for vm 5 years ago
reduce_sum.py Add all other tbe op info register 6 years ago
relu.py add matmul eltwise buffer fusion pass 5 years ago
relu6.py adapt assign assignAdd relu6 5 years ago
relu6_grad.py get input info of fusionop by visitkernel 5 years ago
relu_grad.py Add all other tbe op info register 6 years ago
relu_grad_v2.py Add ReluV2/ReluGradV2/ConfusionMulGrad for VM 6 years ago
relu_v2.py Add ReluV2/ReluGradV2/ConfusionMulGrad for VM 6 years ago
reshape.py Add all other tbe op info register 6 years ago
resize_bilinear.py 接入算子resizebilinear and resizebilineargrad 5 years ago
resize_bilinear_grad.py 接入算子resizebilinear and resizebilineargrad 5 years ago
resize_nearest_neighbor.py Add all other tbe op info register 6 years ago
resize_nearest_neighbor_d.py Add all other tbe op info register 6 years ago
resize_nearest_neighbor_grad_d.py Add all other tbe op info register 6 years ago
roi_align.py add register info for ROIAlign and ROIAlignGrad 5 years ago
roi_align_grad.py add register info for ROIAlign and ROIAlignGrad 5 years ago
round.py Add all other tbe op info register 6 years ago
rsqrt.py Add all other tbe op info register 6 years ago
scatter_nd.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 5 years ago
scatter_nd_d.py Add all other tbe op info register 6 years ago
scatter_nd_update.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 5 years ago
scatter_update.py support tensor get value by tensor index 5 years ago
select.py Add all other tbe op info register 6 years ago
sgd.py add op BoundingBoxDecode 5 years ago
sigmoid.py Add all other tbe op info register 6 years ago
sigmoid_cross_entropy_with_logits.py Add all other tbe op info register 6 years ago
sigmoid_cross_entropy_with_logits_grad.py Add all other tbe op info register 6 years ago
sigmoid_grad.py Add all other tbe op info register 6 years ago
sign.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 6 years ago
sin.py add vm support sin and cumsum 5 years ago
slice.py Add all other tbe op info register 6 years ago
smooth_l1_loss.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 6 years ago
smooth_l1_loss_grad.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 6 years ago
softmax.py Add all other tbe op info register 6 years ago
softmax_cross_entropy_with_logits.py Add all other tbe op info register 6 years ago
softplus.py adapt Softplus\SoftplusGrad\ApplyFtrlD for VM 5 years ago
softplus_grad.py adapt Softplus\SoftplusGrad\ApplyFtrlD for VM 5 years ago
space_to_batch.py support float32 5 years ago
space_to_depth.py support vm for space_to_depth 5 years ago
split_d.py Add all other tbe op info register 6 years ago
sqrt.py Add all other tbe op info register 6 years ago
square.py Add all other tbe op info register 6 years ago
square_sum_all.py add squaresumall op 5 years ago
square_sum_v1.py Add all other tbe op info register 6 years ago
square_sum_v2.py Add all other tbe op info register 6 years ago
squeeze.py Add all other tbe op info register 6 years ago
strided_slice_d.py add ops info for VM 5 years ago
strided_slice_grad_d.py add ops info for VM 5 years ago
sub.py Add all other tbe op info register 6 years ago
tanh.py fix bert percision problem 5 years ago
tanh_grad.py Add all other tbe op info register 6 years ago
tensor_add.py Add all other tbe op info register 6 years ago
tile.py Add all other tbe op info register 6 years ago
top_k.py adapt vm topk 5 years ago
trans_data.py Add all other tbe op info register 6 years ago
transpose_d.py Add all other tbe op info register 6 years ago
unpack.py support vm for pack 5 years ago
unsorted_segment_sum.py Add all other tbe op info register 6 years ago
zeros_like.py Add all other tbe op info register 6 years ago