mindspore 2.0.0a0__cp37-cp37m-win_amd64.whl → 2.0.0rc1__cp37-cp37m-win_amd64.whl

mindspore/ops/_op_impl/tbe/__init__.py

@@ -14,616 +14,32 @@
 # ============================================================================
 
 """tbe ops"""
-from .celu import _celu_tbe
-from .cummin import _cummin_tbe
-from .abs import _abs_tbe
-from .abs_ds import _abs_ds_tbe
-from .inplace_add import _inplace_add_tbe
-from .inplace_sub import _inplace_sub_tbe
-from .abs_grad import _abs_grad_tbe
-from .abs_grad_ds import _abs_grad_ds_tbe
-from .acos import _acos_tbe
-from .acos_ds import _acos_ds_tbe
-from .acos_grad import _acos_grad_tbe
-from .acos_grad_ds import _acos_grad_ds_tbe
-from .acosh import _acosh_tbe
-from .acosh_ds import _acosh_ds_tbe
-from .acosh_grad import _acosh_grad_tbe
-from .acosh_grad_ds import _acosh_grad_ds_tbe
-from .adam_apply_one_with_decay import _adam_apply_one_with_decay_tbe
-from .apply_centered_rms_prop import _apply_centered_rms_prop_tbe
-from .apply_centered_rms_prop_ds import _apply_centered_rms_prop_ds_tbe
-from .add import _add_tbe
-from .add_ds import _add_ds_tbe
-from .add_n import _add_n_tbe
-from .add_n_ds import _add_n_ds_tbe
-from .addcdiv import _addcdiv_tbe
-from .addcdiv_ds import _addcdiv_ds_tbe
-from .addcmul import _addcmul_tbe
-from .addcmul_ds import _addcmul_ds_tbe
-from .accumulate_n_v2 import _accumulate_n_v2_tbe
-from .accumulate_n_v2_ds import _accumulate_n_v2_ds_tbe
-from .apply_ftrl import _apply_ftrl_tbe
-from .apply_ftrl_ds import _apply_ftrl_ds_tbe
-from .apply_keras_momentum import _apply_keras_momentum_tbe
-from .apply_momentum import _apply_momentum_tbe
-from .apply_momentum_ds import _apply_momentum_ds_tbe
-from .apply_adam import _apply_adam_tbe
-from .apply_adam_ds import _apply_adam_ds_tbe
-from .apply_ada_max import _apply_ada_max_tbe
-from .apply_ada_max_ds import _apply_ada_max_ds_tbe
-from .apply_adadelta import _apply_adadelta_tbe
-from .apply_adadelta_ds import _apply_adadelta_ds_tbe
-from .apply_adagrad import _apply_adagrad_tbe
-from .apply_adagrad_ds import _apply_adagrad_ds_tbe
-from .apply_adagrad_v2 import _apply_adagrad_v2_tbe
-from .apply_adagrad_v2_ds import _apply_adagrad_v2_ds_tbe
-from .apply_adagrad_d_a import _apply_adagrad_d_a_tbe
-from .apply_add_sign import _apply_add_sign_tbe
-from .apply_add_sign_ds import _apply_add_sign_ds_tbe
-from .apply_power_sign import _apply_power_sign_tbe
-from .apply_power_sign_ds import _apply_power_sign_ds_tbe
-from .apply_gradient_descent import _apply_gradient_descent_tbe
-from .apply_gradient_descent_ds import _apply_gradient_descent_ds_tbe
-from .apply_proximal_gradient_descent import _apply_proximal_gradient_descent_tbe
-from .apply_proximal_gradient_descent_ds import _apply_proximal_gradient_descent_ds_tbe
-from .sparse_apply_ftrl_v2 import _sparse_apply_ftrl_v2_tbe
-from .sparse_apply_adagrad_v2 import _sparse_apply_adagrad_v2_tbe
-from .approximate_equal import _approximate_equal_tbe
-from .approximate_equal_ds import _approximate_equal_ds_tbe
-from .adam_apply_one import _adam_apply_one_tbe
-from .assign import _assign_tbe
-from .assign_ds import _assign_ds_tbe
-from .assign_add import _assign_add_tbe
-from .assign_add_ds import _assign_add_ds_tbe
-from .assign_sub import _assign_sub_tbe
-from .assign_sub_ds import _assign_sub_ds_tbe
-from .batch_matmul import _batch_matmul_tbe
-from .batch_matmul_ds import _batch_matmul_ds_tbe
-from .batch_matmul_v2 import _batch_matmul_v2_tbe
-from .batchnorm import _batch_norm_tbe
-from .batchnorm_grad import _batch_norm_grad_tbe
-from .bias_add import _bias_add_tbe
-from .bias_add_ds import _bias_add_ds_tbe
-from .bias_add_grad import _bias_add_grad_tbe
-from .cast import _cast_tbe
-from .cast_ds import _cast_ds_tbe
-from .conv2d import _conv2d_tbe
-from .conv2d_backprop_filter import _conv2d_backprop_filter_tbe
-from .conv2d_backprop_input import _conv2d_backprop_input_tbe
-from .conv2d_ds import _conv2d_ds_tbe
-from .conv2d_backprop_filter_ds import _conv2d_backprop_filter_ds_tbe
-from .conv2d_backprop_input_ds import _conv2d_backprop_input_ds_tbe
-from .conv2d_transpose import _conv2d_transpose_tbe
-from .confusion_mul_grad import _confusion_mul_grad_tbe
-from .dilation import _dilation_tbe
-from .dropout_do_mask import _dropout_do_mask_tbe
-from .dropout_do_mask_ds import _dropout_do_mask_ds_tbe
-from .dropout_do_mask_v3 import _dropout_do_mask_v3_tbe
-from .softmax_v2_with_dropout_do_mask_v3 import _softmax_v2_with_dropout_do_mask_v3_tbe
-from .gelu import _gelu_tbe
-from .gelu_ds import _gelu_ds_tbe
-from .gelu_grad import _gelu_grad_tbe
-from .gelu_grad_ds import _gelu_grad_ds_tbe
-from .ger import _ger_tbe
-from .ger_ds import _ger_ds_tbe
-from .fast_gelu import _fast_gelu_tbe
-from .fast_gelu_ds import _fast_gelu_ds_tbe
-from .fast_gelu_grad import _fast_gelu_grad_tbe
-from .fast_gelu_grad_ds import _fast_gelu_grad_ds_tbe
-from .max_pool import _max_pool_tbe
-from .max_pool_ds import _max_pool_ds_tbe
-from .max_pool_grad import _max_pool_grad_tbe
-from .max_pool_grad_with_argmax import _max_pool_grad_with_argmax_tbe
-from .max_pool_with_argmax import _max_pool_with_argmax_tbe
-from .max_pool3d import _max_pool_3d_tbe
-from .max_pool3d_grad import _max_pool_3d_grad_tbe
-from .max_pool3d_grad_grad import _max_pool_3d_grad_grad_tbe
-from .mul import _mul_tbe
-from .mul_ds import _mul_ds_tbe
-from .real_div import _real_div_tbe
-from .real_div_ds import _real_div_ds_tbe
-from .relu import _relu_tbe
-from .relu_ds import _relu_ds_tbe
-from .relu_grad import _relu_grad_tbe
-from .relu_grad_ds import _relu_grad_ds_tbe
-from .relu6 import _relu6_tbe
-from .relu6_ds import _relu6_ds_tbe
-from .relu6_grad import _relu6_grad_tbe
-from .relu6_grad_ds import _relu6_grad_ds_tbe
-from .relu_v2 import _relu_v2_tbe
-from .relu_grad_v2 import _relu_grad_v2_tbe
-from .relu_v2_ds import _relu_v2_ds_tbe
-from .relu_grad_v2_ds import _relu_grad_v2_ds_tbe
-from .softmax_cross_entropy_with_logits import _softmax_cross_entropy_with_logits_tbe
-from .softmax_cross_entropy_with_logits_ds import _softmax_cross_entropy_with_logits_ds_tbe
-from .sigmoid_cross_entropy_with_logits import _sigmoid_cross_entropy_with_logits_tbe
-from .sigmoid_cross_entropy_with_logits_ds import _sigmoid_cross_entropy_with_logits_ds_tbe
-from .sigmoid_cross_entropy_with_logits_grad import _sigmoid_cross_entropy_with_logits_grad_tbe
-from .sparse_apply_adadelta import _sparse_apply_adadelta_tbe
-from .sigmoid_cross_entropy_with_logits_grad_ds import _sigmoid_cross_entropy_with_logits_grad_ds_tbe
-from .trans_data import _trans_data_tbe
-from .trans_data_ds import _trans_data_ds_tbe
-from .trans_data_rnn import _trans_data_rnn_tbe
-from .top_k import _top_k_tbe
-from .top_k_ds import _top_k_ds_tbe
-from .matmul import _matmul_tbe
-from .matmul_ds import _matmul_ds_tbe
-from .matmul_v2 import _matmul_v2_tbe
-from .sub import _sub_tbe
-from .sub_ds import _sub_ds_tbe
-from .scatter_nd import _scatter_nd_tbe
-from .scatter_nd_ds import _scatter_nd_ds_tbe
-from .scatter_nd_d import _scatter_nd_d_tbe
-from .scatter_nd_add import _scatter_nd_add_tbe
-from .scatter_nd_sub import _scatter_nd_sub_tbe
-from .scatter_nd_sub_ds import _scatter_nd_sub_ds_tbe
-from .scatter_non_aliasing_add import _scatter_non_aliasing_add_tbe
-from .scatter_non_aliasing_add_ds import _scatter_non_aliasing_add_ds_tbe
-from .reduce_mean import _reduce_mean_tbe
-from .reduce_mean_ds import _reduce_mean_ds_tbe
-from .tile import _tile_tbe
-from .tile_ds import _tile_ds_tbe
-from .atomic_addr_clean import _atomic_addr_clean_tbe
-from .dynamic_atomic_addr_clean import _dynamic_atomic_addr_clean_tbe
-from .gather_v2 import _gather_v2_tbe
-from .gather_v2_ds import _gather_v2_ds_tbe
-from .gather_nd import _gather_nd_tbe
-from .gather_nd_ds import _gather_nd_ds_tbe
-from .bn_training_reduce import _bn_training_reduce_tbe
-from .bn_training_reduce_ds import _bn_training_reduce_ds_tbe
-from .bn_training_reduce_grad import _bn_training_reduce_grad_tbe
-from .bn_training_reduce_grad_ds import _bn_training_reduce_grad_ds_tbe
-from .bn_training_update import _bn_training_update_tbe
-from .bn_training_update_ds import _bn_training_update_ds_tbe
-from .bn_training_update_grad import _bn_training_update_grad_tbe
-from .bn_training_update_grad_ds import _bn_training_update_grad_ds_tbe
-from .bn_infer import _bn_infer_tbe
-from .bn_infer_ds import _bn_infer_ds_tbe
+from .broadcast_to import _broadcast_to_tbe  # The name is occupied
+from .broadcast_to_ds import _broadcast_to_ds_tbe  # The name is occupied
+from .batch_to_space import _batch_to_space_tbe  # attr type is listInt，not listListInt
+from .batch_to_space_nd import _batch_to_space_nd_tbe  # attr type is listInt，not listListInt
+from .batch_to_space_nd_v2 import _batch_to_space_nd_v2_tbe  # The name is occupied
+from .space_to_batch import _space_to_batch_tbe  # attr type is listInt，not listListInt
+from .space_to_batch_nd import _space_to_batch_nd_tbe  # attr type is listInt，not listListInt
+from .dynamic_gru_v2 import _dynamic_gru_v2_tbe  # input4 is None, GE will change to hidden op by pass
+from .dynamic_rnn import _dynamic_rnn_tbe  # input4 is None, GE will change to hidden op by pass
+from .kl_div_loss_grad import _kl_div_loss_grad_tbe  # Accuracy issues
+from .inplace_index_add import _inplace_index_add_tbe  # check support failed if var has only one dimension
+from .scatter_nd_update_ds import _scatter_nd_update_ds_tbe  # not support int8 in op json
+from .scatter_nd_add import _scatter_nd_add_tbe  # not support int8 in op json
+from .cast_ds import _cast_ds_tbe  # Accuracy issues
+from .avg_pool_3d_grad import _avg_pool_3d_grad_tbe  # Second device format is facz_3d, but in json, the key is ndhwc
+from .data_format_dim_map_ds import _data_format_dim_map_ds_tbe  # attr order swap
+from .depthwise_conv2d import _depthwise_conv2d_tbe  # Accuracy issues(second format is error in python)
+from .acos import _acos_tbe # Accuracy issues(task error in parallel)
+from .trans_data_ds import _trans_data_ds_tbe # performance and accuracy
+from .scatter_nd_d import _scatter_nd_d_tbe # in python no check supported
+from .assign_add_ds import _assign_add_ds_tbe # "Frac_nz in pangu not support"
+from .atomic_addr_clean import _atomic_addr_clean_tbe # need to clean addr larger than 2G, int32 is not enough
+from .assign import _assign_tbe # Different formats of assign inputs cause memory to increase
+from .npu_clear_float_status_v2 import _npu_clear_float_status_v2_tbe  # io mismatch
+from .npu_get_float_status_v2 import _npu_get_float_status_v2_tbe  # io mismatch
+from .tensor_move import _tensor_move_tbe # not support uint32 in op json
 from .bn_infer_grad import _bn_infer_grad_tbe
-from .bn_infer_grad_ds import _bn_infer_grad_ds_tbe
-from .bn_inference import _bn_inference_tbe
-from .fused_dbn_dw import _fused_dbn_dw_tbe
-from .reciprocal import _reciprocal_tbe
-from .reverse_v2_d import _reverse_v2_d_tbe
-from .rint import _rint_tbe
-from .rint_ds import _rint_ds_tbe
-from .strided_slice_d import _strided_slice_d_tbe
-from .strided_slice_ds import _strided_slice_ds_tbe
-from .strided_slice_grad_d import _strided_slice_grad_d_tbe
-from .strided_slice_grad_ds import _strided_slice_grad_ds_tbe
-from .split_d import _split_d_tbe
-from .split_d_ds import _split_d_ds_tbe
-from .exp import _exp_tbe
-from .expm1 import _expm1_tbe
-from .elu import _elu_tbe
-from .elu_ds import _elu_ds_tbe
-from .elu_grad import _elu_grad_tbe
-from .elu_grad_ds import _elu_grad_ds_tbe
-from .div import _div_tbe
-from .div_ds import _div_ds_tbe
-from .log import _log_tbe
-from .xdivy import _xdivy_tbe
-from .xdivy_ds import _xdivy_ds_tbe
-from .xlogy import _xlogy_tbe
-from .xlogy_ds import _xlogy_ds_tbe
-from .floor_div import _floor_div_tbe
-from .floor_div_ds import _floor_div_ds_tbe
-from .zeros_like import _zeros_like_tbe
-from .zeros_like_ds import _zeros_like_ds_tbe
-from .neg import _neg_tbe
-from .npu_clear_float_status import _npu_clear_float_status_tbe
-from .npu_get_float_status import _npu_get_float_status_tbe
-from .npu_alloc_float_status import _npu_alloc_float_status_tbe
-from .one_hot import _one_hot_tbe
-from .one_hot_ds import _one_hot_ds_tbe
-from .equal import _equal_tbe
-from .equal_ds import _equal_ds_tbe
-from .lerp import _lerp_tbe
-from .less import _less_tbe
-from .less_ds import _less_ds_tbe
-from .less_equal import _less_equal_tbe
-from .less_equal_ds import _less_equal_ds_tbe
-from .logical_and import _logical_and_tbe
-from .logical_and_ds import _logical_and_ds_tbe
-from .logical_not import _logical_not_tbe
-from .logical_not_ds import _logical_not_ds_tbe
-from .logical_or import _logical_or_tbe
-from .logical_or_ds import _logical_or_ds_tbe
-from .reduce_max import _reduce_max_tbe
-from .reduce_max_ds import _reduce_max_ds_tbe
-from .reduce_min import _reduce_min_tbe
-from .reduce_min_ds import _reduce_min_ds_tbe
-from .reduce_sum import _reduce_sum_tbe
-from .reduce_sum_ds import _reduce_sum_ds_tbe
-from .reduce_std import _reduce_std_tbe
-from .round import _round_tbe
-from .tanh import _tanh_tbe
-from .tanh_ds import _tanh_ds_tbe
-from .tanh_grad import _tanh_grad_tbe
-from .tanh_grad_ds import _tanh_grad_ds_tbe
-from .softmax import _softmax_tbe
-from .softmax_ds import _softmax_ds_tbe
-from .softsign import _softsign_tbe
-from .softsign_ds import _softsign_ds_tbe
-from .softplus import _softplus_tbe
-from .softplus_ds import _softplus_ds_tbe
-from .softplus_grad import _softplus_grad_tbe
-from .softplus_grad_ds import _softplus_grad_ds_tbe
-from .softmax_grad_ext import _softmax_grad_ext_tbe
-from .square import _square_tbe
-from .square_ds import _square_ds_tbe
-from .squared_difference import _squared_difference_tbe
-from .squared_difference_ds import _squared_difference_ds_tbe
-from .sqrt import _sqrt_tbe
-from .sqrt_ds import _sqrt_ds_tbe
-from .sparse_apply_ftrl_d import _sparse_apply_ftrl_d
-from .sparse_apply_ftrl_d_ds import _sparse_apply_ftrl_d_ds
-from .sparse_apply_proximal_adagrad import _sparse_apply_proximal_adagrad
-from .sparse_apply_proximal_adagrad_ds import _sparse_apply_proximal_adagrad_ds
-from .apply_proximal_adagrad import _apply_proximal_adagrad
-from .apply_proximal_adagrad_ds import _apply_proximal_adagrad_ds_tbe
-from .transpose_nod import _transpose_nod_tbe
-from .transpose_d import _transpose_d_tbe
-from .transpose import _transpose_tbe
-from .truncate_div import _truncate_div_tbe
-from .truncate_div_ds import _truncate_div_ds_tbe
-from .truncate_mod import _truncate_mod_tbe
-from .truncate_mod_ds import _truncate_mod_ds_tbe
-from .unsorted_segment_sum import _unsorted_segment_sum_tbe
-from .unsorted_segment_sum_ds import _unsorted_segment_sum_ds_tbe
-from .unsorted_segment_prod import _unsorted_segment_prod_tbe
-from .unsorted_segment_prod_ds import _unsorted_segment_prod_ds_tbe
-from .logsoftmax_grad import _logsoftmax_grad_tbe
-from .logsoftmax_grad_ds import _logsoftmax_grad_ds_tbe
-from .logsoftmax import _logsoftmax_tbe
-from .logsoftmax_ds import _logsoftmax_ds_tbe
-from .select import _select_tbe
-from .select_ds import _select_ds_tbe
-from .pow import _pow_tbe
-from .pow_ds import _pow_ds_tbe
-from .maximum import _maximum_tbe
-from .maximum_ds import _maximum_ds_tbe
-from .maximum_grad_ds import _maximum_grad_ds_tbe
-from .minimum import _minimum_tbe
-from .minimum_ds import _minimum_ds_tbe
-from .minimum_grad import _minimum_grad_tbe
-from .maximum_grad import _maximum_grad_tbe
-from .minimum_grad_ds import _minimum_grad_ds_tbe
-from .concat import _concat_tbe
-from .concat_ds import _concat_ds_tbe
-from .slice import _slice_tbe
-from .sign import _sign_tbe
-from .sign_ds import _sign_ds_tbe
-from .greater import _greater_tbe
-from .clip_by_norm_no_div_sum import _clip_by_norm_no_div_sum_tbe
-from .clip_by_norm_no_div_sum_ds import _clip_by_norm_no_div_sum_ds_tbe
-from .clip_by_value import _clip_by_value_tbe
-from .clip_by_value_ds import _clip_by_value_ds_tbe
-from .layer_norm_beta_gamma_backprop import _layer_norm_beta_gamma_backprop_tbe
-from .layer_norm_beta_gamma_backprop_ds import _layer_norm_beta_gamma_backprop_ds_tbe
-from .layer_norm_beta_gamma_backprop_v2 import _layer_norm_beta_gamma_backprop_v2_tbe
-from .layer_norm_beta_gamma_backprop_v2_ds import _layer_norm_beta_gamma_backprop_v2_ds_tbe
-from .layer_norm import _layer_norm_tbe
-from .layer_norm_ds import _layer_norm_ds_tbe
-from .layer_norm_grad import _layer_norm_grad_tbe
-from .layer_norm_x_backprop_ds import _layer_norm_x_backprop_ds_tbe
-from .layer_norm_x_backprop_v2 import _layer_norm_x_backprop_v2_tbe
-from .layer_norm_x_backprop_v2_ds import _layer_norm_x_backprop_v2_ds_tbe
-from .kl_div_loss import _kl_div_loss_tbe
-from .kl_div_loss_ds import _kl_div_loss_ds_tbe
-from .kl_div_loss_grad import _kl_div_loss_grad_tbe
-from .l2_loss import _l2_loss_tbe
-from .l2_loss_ds import _l2_loss_ds_tbe
-from .l2_normalize import _l2_normalize_tbe
-from .l2_normalize_grad import _l2_normalize_grad_tbe
-from .square_sum_v1 import _square_sum_v1_tbe
-from .square_sum_v1_ds import _square_sum_v1_ds_tbe
-from .square_sum_v2 import _square_sum_v2_tbe
-from .confusion_transpose_d import _confusion_transpose_d_tbe
-from .confusion_softmax_grad import _confusion_softmax_grad_tbe
-from .lamb_update_with_lr_v2 import _lamb_update_with_lr_v2_tbe
-from .lamb_next_mv import _lamb_next_mv_tbe
-from .lamb_next_mv_with_decay import _lamb_next_mv_with_decay_tbe
-from .lamb_update_with_lr import _lamb_update_with_lr_tbe
-from .rsqrt import _rsqrt_tbe
-from .rsqrt_ds import _rsqrt_ds_tbe
-from .sigmoid import _sigmoid_tbe
-from .sigmoid_ds import _sigmoid_ds_tbe
-from .sigmoid_grad import _sigmoid_grad_tbe
-from .sigmoid_grad_ds import _sigmoid_grad_ds_tbe
-from .resize_nearest_neighbor import _resize_nearest_neighbor_tbe
-from .resize_nearest_neighbor_ds import _resize_nearest_neighbor_ds_tbe
-from .resize_nearest_neighbor_grad import _resize_nearest_neighbor_grad_tbe
-from .resize_nearest_neighbor_grad_ds import _resize_nearest_neighbor_grad_ds_tbe
-from .pad_d import _pad_d_tbe
-from .pad_d_ds import _pad_d_ds_tbe
-from .arg_max_with_value import _arg_max_with_value_tbe
-from .arg_max_with_value_ds import _arg_max_with_value_ds_tbe
-from .arg_min_with_value import _arg_min_with_value_tbe
-from .smooth_l1_loss import _smooth_l1_loss_tbe
-from .smooth_l1_loss_ds import _smooth_l1_loss_ds_tbe
-from .smooth_l1_loss_grad import _smooth_l1_loss_grad_tbe
-from .smooth_l1_loss_grad_ds import _smooth_l1_loss_grad_ds_tbe
-from .soft_margin_loss import _soft_margin_loss_tbe
-from .soft_margin_loss_grad import _soft_margin_loss_grad_tbe
-from .fused_mul_add import _fused_mul_add_tbe
-from .fused_mul_add_n import _fused_mul_add_n_tbe
-from .fused_mul_apply_momentum import _fused_mul_apply_momentum_tbe
-from .fill import _fill_op_tbe
-from .fill_ds import _fill_ds_op_tbe
-from .erf import _erf_op_tbe
-from .erf_ds import _erf_ds_op_tbe
-from .erfc import _erfc_op_tbe
-from .erfc_ds import _erfc_ds_op_tbe
-from .depthwise_conv2d import _depthwise_conv2d_tbe
-from .depthwise_conv2d_backprop_filter import _depthwise_conv2d_backprop_filter_tbe
-from .depthwise_conv2d_backprop_input import _depthwise_conv2d_backprop_input_tbe
-from .greater_equal import _greater_equal_tbe
-from .greater_equal_ds import _greater_equal_ds_tbe
-from .not_equal import _not_equal_tbe
-from .floor_mod import _floor_mod_tbe
-from .floor_mod_ds import _floor_mod_ds_tbe
-from .scatter_nd_update import _scatter_nd_update_tbe
-from .scatter_nd_update_ds import _scatter_nd_update_ds_tbe
 from .avg_pool import _avg_pool_tbe
-from .avg_pool_ds import _avg_pool_ds_tbe
-from .avg_pool_grad import _avg_pool_grad_tbe
-from .avg_pool_grad_vm import _avg_pool_grad_vm_tbe
-from .avg_pool_3d import _avg_pool_3d_tbe
-from .avg_pool_3d_grad import _avg_pool_3d_grad_tbe
-from .ones_like import _ones_like_tbe
-from .ones_like_ds import _ones_like_ds_tbe
-from .batch_to_space import _batch_to_space_tbe
-from .space_to_batch import _space_to_batch_tbe
-from .depth_to_space_ds import _depth_to_space_ds_tbe
-from .space_to_depth import _space_to_depth_tbe
-from .extract_image_patches import _extract_image_patches_tbe
-from .sort import _sort_tbe
-from .sort_ds import _sort_ds_tbe
-from .floor import _floor_tbe
-from .floor_ds import _floor_ds_tbe
-from .ceil import _ceil_tbe
-from .ceil_ds import _ceil_ds_tbe
-from .log1p import _log1p_tbe
-from .log1p_ds import _log1p_ds_tbe
-from .resize_bilinear import _resize_bilinear_tbe
-from .resize_bilinear_v2 import _resize_bilinear_v2_tbe
-from .resize_bilinear_grad import _resize_bilinear_grad_tbe
-from .flatten import _flatten_tbe
-from .roi_align import _roi_align_tbe
-from .roi_align_ds import _roi_align_ds_tbe
-from .roi_align_grad import _roi_align_grad_tbe
-from .roi_align_grad_ds import _roi_align_grad_ds_tbe
-from .bounding_box_decode import _bounding_box_decode_tbe
-from .bounding_box_decode_ds import _bounding_box_decode_ds_tbe
-from .bounding_box_encode import _bounding_box_encode_tbe
-from .check_valid import _check_valid_tbe
-from .check_valid_ds import _check_valid_ds_tbe
-from .iou import _iou_tbe
-from .iou_ds import _iou_ds_tbe
-from .arg_max import _arg_max_tbe
-from .nms_with_mask import _nms_with_mask_tbe
-from .sgd import _sgd_tbe
-from .lars_update import _lars_update_tbe
-from .lars_update_ds import _lars_update_ds_tbe
-from .arg_min import _arg_min_tbe
-from .arg_min_v2_ds import _arg_min_v2_ds_tbe
-from .bn_training_update_v2 import _bn_training_update_v2_tbe
-from .bn_training_update_v3 import _bn_training_update_v3_tbe
-from .square_sum_all import _square_sum_all_tbe
-from .square_sum_all_ds import _square_sum_all_ds_tbe
-from .pack import _pack_tbe
-from .pack_ds import _pack_ds_tbe
-from .unpack import _unpack_tbe
-from .unpack_ds import _unpack_ds_tbe
-from .scatter_update import _scatter_update_tbe
-from .prelu import _prelu_tbe
-from .prelu_ds import _prelu_ds_tbe
-from .prelu_grad import _prelu_grad_tbe
-from .binary_cross_entropy_ds import _binary_cross_entropy_ds_tbe
-from .binary_cross_entropy import _binary_cross_entropy_tbe
-from .binary_cross_entropy_grad import _binary_cross_entropy_grad_tbe
-from .binary_cross_entropy_grad_ds import _binary_cross_entropy_grad_ds_tbe
-from .bce_with_logits_loss import _bce_with_logits_loss_op_tbe
-from .sin import _sin_tbe
-from .sin_ds import _sin_ds_tbe
-from .cos_ds import _cos_ds_tbe
-from .cos import _cos_tbe
-from .tan import _tan_tbe
-from .tan_ds import _tan_ds_tbe
-from .cum_sum import _cum_sum_tbe
-from .cum_sum_ds import _cum_sum_ds_tbe
-from .apply_rms_prop import _apply_rms_prop_tbe
-from .cumprod import _cumprop_tbe
-from .reduce_prod import _reduce_prod_tbe
-from .reduce_prod_ds import _reduce_prod_ds_tbe
-from .reciprocal_grad import _reciprocal_grad_tbe
-from .reciprocal_grad_ds import _reciprocal_grad_ds_tbe
-from .sqrt_grad import _sqrt_grad_tbe
-from .sqrt_grad_ds import _sqrt_grad_ds_tbe
-from .rsqrt_grad import _rsqrt_grad_tbe
-from .rsqrt_grad_ds import _rsqrt_grad_ds_tbe
-from .scatter_add import _scatter_add_tbe
-from .atan2 import _atan2_tbe
-from .atan2_ds import _atan2_ds_tbe
-from .bessel_i0e import _bessel_i0e_tbe
-from .bessel_i0e_ds import _bessel_i0e_ds_tbe
-from .bessel_i1e import _bessel_i1e_tbe
-from .bessel_i1e_ds import _bessel_i1e_ds_tbe
-from .batch_to_space_nd import _batch_to_space_nd_tbe
-from .batch_to_space_nd_ds import _batch_to_space_nd_ds_tbe
-from .batch_to_space_nd_v2 import _batch_to_space_nd_v2_tbe
-from .space_to_batch_nd import _space_to_batch_nd_tbe
-from .bitwise_and import _bitwise_and_tbe
-from .bitwise_and_ds import _bitwise_and_ds_tbe
-from .bitwise_or import _bitwise_or_tbe
-from .bitwise_or_ds import _bitwise_or_ds_tbe
-from .bitwise_xor import _bitwise_xor_tbe
-from .bitwise_xor_ds import _bitwise_xor_ds_tbe
-from .reduce_all import _reduce_all_tbe
-from .reduce_all_ds import _reduce_all_ds_tbe
-from .reduce_any import _reduce_any_tbe
-from .reduce_any_ds import _reduce_any_ds_tbe
-from .sparse_apply_adagrad import _sparse_apply_adagrad_tbe
-from .sparse_apply_adagrad_ds import _sparse_apply_adagrad_ds
-from .sparse_apply_adagrad_v2_ds import _sparse_apply_adagrad_v2_ds
-from .unsorted_segment_min import _unsorted_segment_min_tbe
-from .unsorted_segment_min_ds import _unsorted_segment_min_ds_tbe
-from .unsorted_segment_max import _unsorted_segment_max_tbe
-from .unsorted_segment_max_ds import _unsorted_segment_max_ds_tbe
-from .asin import _asin_tbe
-from .asin_ds import _asin_ds_tbe
-from .asin_grad import _asin_grad_tbe
-from .asin_grad_ds import _asin_grad_ds_tbe
-from .asinh import _asinh_tbe
-from .asinh_ds import _asinh_ds_tbe
-from .asinh_grad import _asinh_grad_tbe
-from .asinh_grad_ds import _asinh_grad_ds_tbe
-from .div_no_nan import _div_no_nan_tbe
-from .div_no_nan_ds import _div_no_nan_ds_tbe
-from .atan import _atan_tbe
-from .atan_ds import _atan_ds_tbe
-from .atan_grad import _atan_grad_tbe
-from .atan_grad_ds import _atan_grad_ds_tbe
-from .atanh import _atanh_tbe
-from .atanh_ds import _atanh_ds_tbe
-from .cosh import _cosh_tbe
-from .cosh_ds import _cosh_ds_tbe
-from .sinh import _sinh_tbe
-from .sinh_ds import _sinh_ds_tbe
-from .inv import _inv_tbe
-from .inv_ds import _inv_ds_tbe
-from .inv_grad import _inv_grad_tbe
-from .inv_grad_ds import _inv_grad_ds_tbe
-from .invert import _invert_tbe
-from .invert_ds import _invert_ds_tbe
-from .basic_lstm_cell import _basic_lstm_cell_tbe
-from .basic_lstm_cell_c_state_grad_v2 import _basic_lstm_cell_c_state_grad_tbe_v2
-from .basic_lstm_cell_c_state_grad import _basic_lstm_cell_c_state_grad_tbe
-from .basic_lstm_cell_weight_grad import _basic_lstm_cell_weight_grad_tbe
-from .basic_lstm_cell_input_grad import _basic_lstm_cell_input_grad_tbe
-from .dynamic_rnn import _dynamic_rnn_tbe
-from .dynamic_gru_v2 import _dynamic_gru_v2_tbe
-from .gru_v2_hidden_grad import _gru_v2_hidden_grad_tbe
-from .gru_v2_hidden_grad_cell import _gru_v2_hidden_grad_cell_tbe
-from .lstm_input_grad import _lstm_input_grad_tbe
-from .confusion_matrix import _confusion_matrix_tbe
-from .broadcast_to import _broadcast_to_tbe
-from .broadcast_to_ds import _broadcast_to_ds_tbe
-from .strided_read import _strided_read_tbe
-from .strided_write import _strided_write_tbe
-from .range import _range_tbe
-from .fused_mul_add_n_l2loss import _fused_mul_add_n_l2loss_tbe
-from .fused_mul_apply_momentum_extern import _fused_mul_apply_momentum_extern_tbe
-from .lamb_next_right import _lamb_next_right_tbe
-from .sparse_gather_v2 import _sparse_gather_v2_tbe
-from .sparse_gather_v2_ds import _sparse_gather_v2_ds_tbe
-from .sparse_apply_r_m_s_prop import _sparse_apply_r_m_s_prop_tbe
-from .sparse_apply_r_m_s_prop_ds import _sparse_apply_r_m_s_prop_ds_tbe
-from .data_format_dim_map import _data_format_dim_map_tbe
-from .data_format_dim_map_ds import _data_format_dim_map_ds_tbe
-from .histogram_fixed_width import _histogram_fixed_width_tbe
-from .tensor_scatter_update import _tensor_scatter_update_tbe
-from .inplace_update import _inplace_update_tbe
-from .inplace_update_ds import _inplace_update_v2_ds_tbe
-from .split_v import _split_v_tbe
-from .in_top_k import _in_top_k_tbe
-from .diag import _diag_tbe
-from .diag_part import _diag_part_tbe
-from .matrix_diag import _matrix_diag_tbe
-from .matrix_diag_part import _matrix_diag_part_tbe
-from .matrix_set_diag import _matrix_set_diag_tbe
-from .lrn import _lrn_tbe
-from .lrn_grad import _lrn_grad_tbe
-from .scatter_max import _scatter_max_tbe
-from .scatter_min import _scatter_min_tbe
-from .scatter_sub import _scatter_sub_tbe
-from .scatter_sub_ds import _scatter_sub_ds_tbe
-from .scatter_mul import _scatter_mul_tbe
-from .scatter_div import _scatter_div_tbe
-from .mod import _mod_tbe
-from .mod_ds import _mod_ds_tbe
-from .max_pool_grad_grad import _max_pool_grad_grad_tbe
-from .max_pool_grad_grad_with_argmax import _max_pool_grad_grad_with_argmax_tbe
-from .tensor_move import _tensor_move_tbe
-from .tensor_move_ds import _tensor_move_ds_tbe
-from .population_count import _population_count_tbe
-from .parallel_concat import _parallel_concat_tbe
-from .adam_apply_one_assign import _adam_apply_one_assign_tbe
-from .adam_apply_one_with_decay_assign import _adam_apply_one_with_decay_assign_tbe
-from .ifmr import _ifmr_tbe
-from .ifmr_ds import _ifmr_ds_tbe
-from .acts_ulq import _acts_ulq_tbe
-from .acts_ulq_input_grad import _acts_ulq_input_grad_tbe
-from .act_ulq_clamp_min_grad import _act_ulq_clamp_min_grad_tbe
-from .act_ulq_clamp_max_grad import _act_ulq_clamp_max_grad_tbe
-from .wts_arq import _wts_arq_tbe
-from .fake_quant_with_min_max_vars import _fake_quant_with_min_max_vars_tbe
-from .fake_quant_with_min_max_vars_gradient import _fake_quant_with_min_max_vars_gradient_tbe
-from .fake_quant_with_min_max_vars_per_channel import _fake_quant_with_min_max_vars_per_channel_tbe
-from .fake_quant_with_min_max_vars_per_channel_gradient import _fake_quant_with_min_max_vars_per_channel_gradient_tbe
-from .conv3d import _conv3d_tbe
-from .conv3d_backprop_input import _conv3d_backprop_input_tbe
-from .conv3d_backprop_filter import _conv3d_backprop_filter_tbe
-from .conv3d_transpose import _conv3d_transpose_tbe
-from .conv3d_transpose_ds import _conv3d_transpose_ds_tbe
-from .lamb_apply_optimizer_assign import _lamb_apply_optimizer_assign_tbe
-from .lamb_apply_weight_assign import _lamb_apply_weight_assign_tbe
-from .nll_loss import _nll_loss_tbe
-from .nll_loss_grad import _nll_loss_grad_tbe
-from .masked_fill import _masked_fill_tbe
-from .masked_fill_ds import _masked_fill_ds_tbe
-from .mish import _mish_tbe
-from .mul_no_nan import _mul_no_nan_tbe
-from .mul_no_nan_ds import _mul_no_nan_ds_tbe
-from .selu import _selu_tbe
-from .selu_ds import _selu_ds_tbe
-from .centralization import _centralization_tbe
-from .exp_ds import _exp_ds_tbe
-from .log_ds import _log_ds_tbe
-from .cdist import _cdist_tbe
-from .cdist_grad import _cdist_grad_tbe
-from .pdist import _pdist_tbe
-from .neg_ds import _neg_ds_tbe
-from .not_equal_ds import _not_ds_equal_tbe
-from .reciprocal_ds import _reciprocal_ds_tbe
-from .lp_norm import _lp_norm_tbe
-from .lp_norm_ds import _lp_norm_ds_tbe
-from .ctc_loss_v2 import _ctc_loss_v2_tbe
-from .ctc_loss_v2_grad import _ctc_loss_v2_grad_tbe
-from .roll import _roll_tbe
-from .soft_shrink import _soft_shrink_tbe
-from .erfinv import _erfinv_tbe
-from .soft_shrink_grad import _soft_shrink_grad_tbe
-from .index_add import _index_add_tbe
-from .hsigmoid_grad import _hsigmoid_grad_tbe
-from .hsigmoid import _hsigmoid_tbe
-from .hshrink import _hshrink_tbe
-from .hshrink_grad import _hshrink_grad_tbe
-from .new_im2col import _new_im2col_tbe
-from .trunc import _trunc_tbe
-from .extract_volume_patches import _extract_volume_patches_tbe
-from .multilabel_margin_loss import _multilabel_margin_loss_tbe
-from .round_ds import _round_ds_tbe
-from .is_close import _is_close_tbe
-from .apply_adam_with_amsgrad import _apply_adam_with_amsgrad_tbe
-from .apply_adam_with_amsgrad_ds import _apply_adam_with_amsgrad_ds_tbe
-from .expm1_ds import _expm1_ds_tbe
-from .deformable_offsets import _deformable_offsets_tbe
-from .deformable_offsets_grad import _deformable_offsets_grad_tbe
-from .parallel_resize_bilinear import _parallel_resize_bilinear_op_info_tbe
-from .parallel_resize_bilinear_grad import _parallel_resize_bilinear_grad_op_info_tbe
-from .p_s_r_o_i_pooling import _p_s_r_o_i_pooling_tbe
-from .p_s_r_o_i_pooling_grad import _p_s_r_o_i_pooling_grad_tbe
-from .renorm import _renorm_tbe
-from .adam_apply_one_ds import _adam_apply_one_ds_tbe
-from .adam_apply_one_with_decay_ds import _adam_apply_one_with_decay_ds_tbe
-from .adaptive_max_pool2d import _adaptive_max_pool2d_tbe
-from .pooling import _pooling_tbe
-from .hard_swish import _hard_swish_tbe
-from .hard_swish_grad import _hard_swish_grad_tbe
-from .hard_swish_ds import _hard_swish_ds_tbe
-from .hard_swish_grad_ds import _hard_swish_grad_ds_tbe
-from .arg_min_with_value_ds import _arg_min_with_value_ds_tbe
-from .im2col import _im2col_tbe
+from .gather_nd import _gather_nd_tbe

mindspore/ops/_op_impl/tbe/assign_add_ds.py

@@ -24,6 +24,7 @@ assign_add_ds_op_info = TBERegOp("AssignAdd") \
     .kernel_name("assign_add") \
     .partial_flag(True) \
     .dynamic_shape(True) \
+    .dynamic_compile_static(True) \
     .input(0, "ref", False, "required", "all") \
     .input(1, "value", False, "required", "all") \
     .output(0, "ref", False, "required", "all") \

mindspore/ops/_op_impl/tbe/atomic_addr_clean.py

@@ -23,7 +23,7 @@ atomic_addr_clean_op_info = TBERegOp("AtomicAddrClean") \
     .compute_cost(10) \
     .kernel_name("atomic_addr_clean") \
     .partial_flag(True) \
-    .attr("automic_add_mem_size", "required", "listUInt64", "all") \
+    .attr("automic_add_mem_size", "required", "listInt64", "all") \
     .get_op_info()
 
 

mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py

@@ -16,7 +16,7 @@
 """AvgPool3DGrad op"""
 from mindspore.ops.op_info_register import op_info_register, TBERegOp, DataType
 
-avg_pool_3d_grad_op_info = TBERegOp("AvgPool3DGrad") \
+avg_pool_3d_grad_op_info = TBERegOp("AvgPool3DGradD") \
     .fusion_type("OPAQUE") \
     .async_flag(False) \
     .binfile_name("avg_pool3d_grad_d.so") \

mindspore/ops/_op_impl/tbe/batch_matmul_ds.py

@@ -31,6 +31,7 @@ batch_matmul_op_info = TBERegOp("BatchMatMul") \
     .input(2, "bias", False, "optional", "all") \
     .output(0, "y", False, "required", "all") \
     .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_Default, DataType.F16_FracNZ) \
+    .dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F32_Default, DataType.F32_FracNZ) \
     .get_op_info()