yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mindspore2022/mindspore/ops/_op_impl/tbe
History
buxue 70c80c0595 dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
..
__init__.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
abs.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 2020-04-18 15:21:44 +08:00
abs_grad.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 2020-04-18 15:21:44 +08:00
adam_apply_one_with_decay.py add new mode for operator info register 2020-04-03 15:54:12 +08:00
add.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
add_n.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
apply_adam.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
apply_momentum.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
arg_max_with_value.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
arg_min_with_value.py Fix arg_min_with_value bug 2020-04-15 14:39:05 +08:00
assign.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
assign_add.py pynative-support-topk-and-print 2020-04-21 11:58:48 +08:00
assign_sub.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
atomic_addr_clean.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
batch_matmul.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
batchnorm.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
batchnorm_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
bias_add.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
bias_add_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
bn_training_reduce.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
bn_training_reduce_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
bn_training_update.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
bn_training_update_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
cast.py auto mix precision 2020-04-17 20:46:42 +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 Add all other tbe op info register 2020-04-14 17:37:56 +08:00
concat.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
confusion_mul_grad.py Add ReluV2/ReluGradV2/ConfusionMulGrad for VM 2020-04-22 21:09:37 +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 Add all other tbe op info register 2020-04-14 17:37:56 +08:00
conv2d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
conv2d_backprop_filter.py fix attribute mapping when docking open source operators 2020-04-20 20:52:05 +08:00
conv2d_backprop_input.py fix attribute mapping when docking open source operators 2020-04-20 20:52:05 +08:00
depthwise_conv2d.py dock DepthwiseConv2dBackprop DepthwiseConv2dBackpropFilter DepthwiseConv2dBackpropInput 2020-04-18 10:27:32 +08:00
depthwise_conv2d_backprop_filter.py fix attribute mapping when docking open source operators 2020-04-20 20:52:05 +08:00
depthwise_conv2d_backprop_input.py fix attribute mapping when docking open source operators 2020-04-20 20:52:05 +08:00
div.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
dropout_do_mask.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
equal.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
erf.py Add Erf\FillD operator for VM 2020-04-22 15:32:42 +08:00
exp.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
expand_dims.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
fill.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
floor_div.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
floor_mod.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
fused_mul_add.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
fused_mul_add_n.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
fused_mul_apply_momentum.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
gather_v2.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
gelu.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
gelu_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
greater.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
greater_equal.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
l2_loss.py Add L2Loss op for VM 2020-04-22 09:31:14 +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_v1.py Add all other tbe op info register 2020-04-14 17:37:56 +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
layer_norm.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
layer_norm_beta_gamma_backprop.py Add all other tbe op info register 2020-04-14 17:37:56 +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 Add all other tbe op info register 2020-04-14 17:37:56 +08:00
less.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
less_equal.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
log.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
logical_and.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
logical_not.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
logical_or.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
logsoftmax.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
logsoftmax_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
matmul.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
max_pool.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
max_pool_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
max_pool_grad_with_argmax.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
max_pool_with_argmax.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
maximum.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
maximum_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
minimum.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
minimum_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
mul.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
neg.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
not_equal.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +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 all other tbe op info register 2020-04-14 17:37:56 +08:00
pad_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
pow.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
real_div.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
reciprocal.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
reduce_max.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
reduce_mean.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
reduce_mean_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
reduce_min.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
reduce_sum.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
relu.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
relu6.py dock relu6 for open source process and fix pow bprop 2020-04-13 15:25:04 +08:00
relu6_grad.py dock relu6 for open source process and fix pow bprop 2020-04-13 15:25:04 +08:00
relu_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
relu_grad_v2.py Add ReluV2/ReluGradV2/ConfusionMulGrad for VM 2020-04-22 21:09:37 +08:00
relu_v2.py Add ReluV2/ReluGradV2/ConfusionMulGrad for VM 2020-04-22 21:09:37 +08:00
reshape.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
resize_nearest_neighbor.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
resize_nearest_neighbor_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
resize_nearest_neighbor_grad_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
round.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
rsqrt.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
scatter_nd.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
scatter_nd_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
scatter_nd_update.py dock FloorMod GreaterEqual NotEqual ScatterNdUpdate 2020-04-27 19:48:16 +08:00
select.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sigmoid.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sigmoid_cross_entropy_with_logits.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sigmoid_cross_entropy_with_logits_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sigmoid_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sign.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 2020-04-18 15:21:44 +08:00
slice.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
smooth_l1_loss.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 2020-04-18 15:21:44 +08:00
smooth_l1_loss_grad.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 2020-04-18 15:21:44 +08:00
softmax.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
softmax_cross_entropy_with_logits.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
split_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sqrt.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
square.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
square_sum_v1.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
square_sum_v2.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
squeeze.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
strideslice_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
strideslicegrad_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
sub.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
tanh.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
tanh_grad.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
tensor_add.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
tile.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
top_k.py Add Abs\AbsGrad\Sign\SmoothL1Loss\SmoothL1LossGrad and modify TopKV2->TopK for VM 2020-04-18 15:21:44 +08:00
trans_data.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
transpose_d.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
unsorted_segment_sum.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00
zeros_like.py Add all other tbe op info register 2020-04-14 17:37:56 +08:00