mindspore 1.10.0__cp37-none-any.whl → 2.0.0rc1__cp37-none-any.whl

mindspore/ops/_op_impl/aicpu/__init__.py

@@ -13,12 +13,43 @@
 # limitations under the License.
 
 """aicpu ops"""
-from .sparse_dense_cwise_add import _sparse_dense_cwise_add_aicpu
-from .sparse_dense_cwise_div import _sparse_dense_cwise_div_aicpu
+from .adaptive_max_pool_3d_grad import _adaptive_max_pool_3d_grad_aicpu
+from .adaptive_max_pool_2d_grad import _adaptive_max_pool_2d_grad_aicpu
+from .adaptive_avg_pool_3d_grad import _adaptiveavgpool3d_grad_aicpu
+from .adaptive_avg_pool_3d import _adaptiveavgpool3d_aicpu
+from .tile import _tile_aicpu
+from .tanh import _tanh_aicpu
+from .less import _less_aicpu
+from .add import _add_aicpu
+from .sparse_matrix_transpose import _sparse_matrix_transpose_aicpu
+from .sparse_matrix_nnz import _sparse_matrix_nnz_aicpu
 from .sparse_dense_cwise_mul import _sparse_dense_cwise_mul_aicpu
+from .sparse_dense_cwise_div import _sparse_dense_cwise_div_aicpu
+from .sparse_dense_cwise_add import _sparse_dense_cwise_add_aicpu
+from .sparse_concat import _sparse_concat_aicpu
+from .sparse_apply_centered_rms_prop import _sparse_apply_centered_rms_prop_aicpu
+from .broadcast_to import _broadcast_to_aicpu
+from .blackman_window import _blackman_window_aicpu
+from .bincount import _bincount_aicpu
+from .asinh_grad import _asinh_grad_aicpu
 from .unique import _unique_aicpu
+from .add_n import _add_n_aicpu
+from .add_v2 import _add_v2_aicpu
+from .adjust_contrastv2 import _adjust_contrastv2_aicpu
+from .adjust_hue import _adjust_hue_aicpu
+from .adjust_saturation import _adjust_saturation_aicpu
+from .affine_grid_grad import _affine_grid_grad_aicpu
+from .angle import _angle_aicpu
+from .arg_max import _arg_max_aicpu
+from .argmax_with_value import _argmax_with_value_aicpu
+from .arg_min import _arg_min_aicpu
+from .argmin_with_value import _argmin_with_value_aicpu
+from .avgpool_v1 import _avgpool_v1_aicpu
+from .avgpool_grad_v1 import _avgpool_grad_v1_aicpu
 from .matrix_solve import _matrix_solve_aicpu
 from .betainc import _betainc_aicpu
+from .bartlett_window import _bartlett_window_aicpu
+from .batch_norm_grad_grad import _batch_norm_grad_grad_aicpu
 from .no_repeat_ngram import _no_repeat_ngram_aicpu
 from .init_data_set_queue import _init_data_set_queue_aicpu
 from .embedding_lookup import _embedding_lookup_aicpu
@@ -43,8 +74,10 @@ from .dynamic_stitch import _dynamic_stitch_aicpu
 from .get_next import _get_next_aicpu
 from .print_tensor import _print_aicpu
 from .topk import _top_k_aicpu
+from .tensor_scatter_update import _tensor_scatter_update_aicpu
 from .log1p import _log1p_aicpu
 from .asin import _asin_aicpu
+from .asin_grad import _asin_grad_aicpu
 from .is_finite import _is_finite_aicpu
 from .is_inf import _is_inf_aicpu
 from .is_nan import _is_nan_aicpu
@@ -54,14 +87,20 @@ from .cosh import _cosh_aicpu
 from .sign import _sign_aicpu
 from .squeeze import _squeeze_aicpu
 from .acos import _acos_aicpu
+from .acos_grad import _acos_grad_aicpu
 from .expand import _expand_aicpu
 from .expand_dims import _expand_dims_aicpu
 from .randperm import _randperm_aicpu
 from .random_choice_with_mask import _random_choice_with_mask_aicpu
 from .rsqrt import _rsqrt_aicpu
+from .sqrt import _sqrt_aicpu
+from .sqrt_grad import _sqrt_grad_aicpu
+from .masked_fill import _masked_fill_aicpu
+from .sort import _sort_aicpu
 from .search_sorted import _search_sorted_aicpu
 from .stack import _stack_aicpu
 from .unstack import _unstack_aicpu
+from .unsorted_segment_sum import _unsorted_segment_sum_aicpu
 from .addcmul import _addcmul_aicpu
 from .uniform_candidate_sampler import _uniform_candidate_sampler_aicpu
 from .log_uniform_candidate_sampler import _log_uniform_candidate_sampler_aicpu
@@ -71,7 +110,10 @@ from .reverse_sequence import _reverse_sequence_aicpu
 from .log_matrix_determinant import _log_matrix_determinant_aicpu
 from .crop_and_resize import _crop_and_resize_aicpu
 from .acosh import _acosh_aicpu
+from .acosh_grad import _acosh_grad_aicpu
 from .rnnt_loss import _rnnt_loss_aicpu
+from .greater import _greater_aicpu
+from .greater_equal import _greater_equal_aicpu
 from .random_categorical import _random_categorical_aicpu
 from .tanh_grad import _tanh_grad_aicpu
 from .cast import _cast_aicpu
@@ -87,6 +129,7 @@ from .sub import _sub_aicpu
 from .not_equal import _not_equal_aicpu
 from .poisson import _poisson_aicpu
 from .update_cache import _update_cache_aicpu
+from .upper_bound import _upper_bound_aicpu
 from .cache_swap_table import _cache_swap_table_aicpu
 from .uniform_int import _uniform_int_aicpu
 from .uniform_real import _uniform_real_aicpu
@@ -98,6 +141,20 @@ from .end_of_sequence import _end_of_sequence_aicpu
 from .fused_sparse_adam import _fused_sparse_adam_aicpu
 from .fused_sparse_lazy_adam import _fused_sparse_lazy_adam_aicpu
 from .fused_sparse_ftrl import _fused_sparse_ftrl_aicpu
+from .sparse_fill_empty_rows_grad import _sparse_fill_empty_rows_grad_aicpu
+from .sparse_reshape import _sparse_reshape_aicpu
+from .sparse_segment_sqrt_n_grad import _sparse_segment_sqrt_n_grad_aicpu
+from .sparse_segment_sum import _sparse_segment_sum_aicpu
+from .sparse_segment_sum_with_num_segments import _sparse_segment_sum_with_num_segments_aicpu
+from .sparse_softmax_cross_entropy_with_logits_v2 import _sparse_softmax_cross_entropy_with_logits_v2_aicpu
+from .sparsesparsemaximum import _sparsesparsemaximum_aicpu
+from .split import _split_aicpu
+from .transpose import _transpose_aicpu
+from .tril_indices import _tril_indices_aicpu
+from .triu_indices import _triu_indices_aicpu
+from .unravel_index import _unravel_index_aicpu
+from .xlogy import _xlogy_aicpu
+from .xdivy import _xdivy_aicpu
 from .fused_sparse_proximal_adagrad import _fused_sparse_proximal_adagrad_aicpu
 from .meshgrid import _meshgrid_aicpu
 from .div import _div_aicpu
@@ -113,6 +170,9 @@ from .tan import _tan_aicpu
 from .ctc_greedy_decoder import _ctc_greedy_decoder_aicpu
 from .resize_bilinear import _resize_bilinear_aicpu
 from .resize_bilinear_grad import _resize_bilinear_grad_aicpu
+from .resize_bicubic_grad import _resize_bicubic_grad_aicpu
+from .resize_nearest_neighbor_v2 import _resize_nearest_neighbor_v2_aicpu
+from .resize_nearest_neighbor_v2_grad import _resize_nearest_neighbor_v2_grad_aicpu
 from .scatter_elements import _scatter_elements_aicpu
 from .non_max_suppression import _non_max_suppression_aicpu
 from .square import _square_aicpu
@@ -148,6 +208,8 @@ from .zeta import _zeta_aicpu
 from .grid_sampler_2d import _grid_sampler_2d_aicpu
 from .grid_sampler_2d_grad import _grid_sampler_2d_grad_aicpu
 from .sparse_segment_mean_grad import _sparse_segment_mean_grad_aicpu
+from .scatter_nd import _scatter_nd_aicpu
+from .scatter_nd_update import _scatter_nd_update_aicpu
 from .scatter_nd_max import _scatter_nd_max_aicpu
 from .conj import _conj_aicpu
 from .scatter_nd_min import _scatter_nd_min_aicpu
@@ -162,4 +224,177 @@ from .reservoir_replay_buffer import _rrb_create_op_cpu
 from .reservoir_replay_buffer import _rrb_push_op_cpu
 from .reservoir_replay_buffer import _rrb_sample_op_cpu
 from .reservoir_replay_buffer import _rrb_destroy_op_cpu
-from .deformable_offsets_grad import _deformable_offsets_grad_op_info_aicpu
+from .concat_offset import _concat_offset_aicpu
+from .range import _range_aicpu
+from .slice_grad import _slice_grad_aicpu
+from .median import _median_aicpu
+from .median_grad import _median_grad_aicpu
+from .reduce_sum import _reduce_sum_aicpu
+from .adaptive_avg_pool_2d import _adaptive_avg_pool_2d_aicpu
+from .adaptive_avg_pool_2d_grad import _adaptive_avg_pool_2d_grad_aicpu
+from .fill_v2 import _fill_v2_aicpu
+from .data_format_vec_permute import _data_format_vec_permute_aicpu
+from .multinomial import _multinomial_aicpu
+from .fft_with_size import _fft_with_size_aicpu
+from .histogram import _histogram_aicpu
+from .matrix_determinant import _matrix_determinant_aicpu
+from .matrix_set_diag_v3 import _matrix_set_diag_v3_aicpu
+from .nan_to_num import _nan_to_num_aicpu
+from .qr import _qr_aicpu
+from .col2im import _col2im_aicpu
+from .matrix_solve_ls import _matrix_solve_ls_aicpu
+from .cauchy import _cauchy_aicpu
+from .bucketize import _bucketize_aicpu
+from .channel_shuffle import _channel_shuffle_aicpu
+from .choleskygrad import _choleskygrad_aicpu
+from .cholesky_inverse import _cholesky_inverse_aicpu
+from .cholesky_solve import _cholesky_solve_aicpu
+from .combined_non_max_suppression import _combined_non_max_suppression_aicpu
+from .complex import _complex_aicpu
+from .complex_abs import _complex_abs_aicpu
+from .concat import _concat_aicpu
+from .cos import _cos_aicpu
+from .csr_sparse_matrix_to_dense import _csr_sparse_matrix_to_dense_aicpu
+from .cumprod import _cumprod_aicpu
+from .exp import _exp_aicpu
+from .matrix_triangular_solve import _matrix_triangular_solve_aicpu
+from .maximum_grad_grad import _maximum_grad_grad_aicpu
+from .maxpool_v1 import _maxpool_v1_aicpu
+from .minimum_grad_grad import _minimum_grad_grad_aicpu
+from .mul_no_nan import _mul_no_nan_aicpu
+from .multilabel_margin_loss_grad import _multilabel_margin_loss_grad_aicpu
+from .nth_element import _nth_element_aicpu
+from .non_max_suppression_with_overlaps import _non_max_suppression_with_overlaps_aicpu
+from .one_hot import _one_hot_aicpu
+from .orgqr import _orgqr_aicpu
+from .parameterized_truncated_normal import _parameterized_truncated_normal_aicpu
+from .polar import _polar_aicpu
+from .pdist_grad import _pdist_grad_aicpu
+from .ragged_range import _raggedrange_aicpu
+from .ragged_tensor_to_sparse import _ragged_tensor_to_sparse_aicpu
+from .ragged_tensor_to_tensor import _ragged_tensor_to_tensor_aicpu
+from .reciprocal import _reciprocal_aicpu
+from .reciprocal_grad import _reciprocal_grad_aicpu
+from .reduce_mean import _reduce_mean_aicpu
+from .reduce_prod import _reduce_prod_aicpu
+from .relu_v3 import _relu_v3_aicpu
+from .reversev2 import _reversev2_aicpu
+from .rgb_to_hsv import _rgb_to_hsv_aicpu
+from .rsqrt_grad import _rsqrt_grad_aicpu
+from .sample_distorted_bounding_box_v2 import _sample_distorted_bounding_box_v2_aicpu
+from .scale_and_translate_grad import _scale_and_translate_grad_aicpu
+from .select import _select_aicpu
+from .self_adjoint_eig import _self_adjoint_eig_aicpu
+from .sin import _sin_aicpu
+from .sinc import _sinc_aicpu
+from .sinh import _sinh_aicpu
+from .smooth_l1_loss_grad import _smooth_l1_loss_grad_aicpu
+from .smooth_l1_loss import _smooth_l1_loss_aicpu
+from .cumulative_logsumexp import _cumulative_logsumexp_aicpu
+from .sparse_segment_sqrt_n import _sparse_segment_sqrt_n_aicpu
+from .scale_and_translate import _scale_and_translate_aicpu
+from .quant_dtype_cast import _quant_dtype_cast_aicpu
+from .fse_decode import _fse_decode_aicpu
+from .dense_to_csr_sparse_matrix import _dense_to_csr_sparse_matrix_aicpu
+from .dense_to_sparse_set_operation import _dense_to_sparse_set_operation_aicpu
+from .diag import _diag_aicpu
+from .diagonal import _diagonal_aicpu
+from .diag_part import _diag_part_aicpu
+from .bias_add import _bias_add_aicpu
+from .bias_add_grad import _bias_add_grad_aicpu
+from .eig import _eig_aicpu
+from .eye import _eye_aicpu
+from .fmin import _fmin_aicpu
+from .fractional_avg_pool import _fractional_avg_pool_aicpu
+from .fractional_avg_pool_grad import _fractional_avg_pool_grad_aicpu
+from .fractional_max_pool import _fractional_max_pool_aicpu
+from .fractional_max_pool_grad import _fractional_max_pool_grad_aicpu
+from .fractional_max_pool_grad_with_fixed_ksize import _fractional_max_pool_grad_with_fixed_ksize_aicpu
+from .gcd import _gcd_aicpu
+from .geqrf import _geqrf_aicpu
+from .hard_sigmoid import _hard_sigmoid_aicpu
+from .hard_sigmoid_grad import _hard_sigmoid_grad_aicpu
+from .heaviside import _heaviside_aicpu
+from .hypot import _hypot_aicpu
+from .identity_n import _identity_n_aicpu
+from .index_fill import _index_fill_aicpu
+from .kldivloss import _kldiv_loss_aicpu
+from .kldivlossgrad import _kldiv_loss_grad_aicpu
+from .lcm import _lcm_aicpu
+from .less_equal import _less_equal_aicpu
+from .logical_xor import _logical_xor_aicpu
+from .logit import _logit_aicpu
+from .logit_grad import _logit_grad_aicpu
+from .log_normal_reverse import _log_normal_reverse_aicpu
+from .lower_bound import _lower_bound_aicpu
+from .lu_unpack_grad import _lu_unpack_grad_aicpu
+from .pad_v3_grad import _pad_v3_grad_aicpu
+from .pad_v3 import _pad_v3_aicpu
+from .cholesky import _cholesky_aicpu
+from .hsv_to_rgb import _hsv_to_rgb_aicpu
+from .im2col import _im2col_aicpu
+from .lu_solve import _lu_solve_aicpu
+from .relu_grad_v3 import _relu_grad_v3_aicpu
+from .resize_bicubic import _resize_bicubic_aicpu
+from .extract_glimpse import _extract_glimpse_aicpu
+from .real_div import _real_div_aicpu
+from .multinomial_with_replacement import _multinomial_with_replacement_aicpu
+from .coalesce import _coalesce_aicpu
+from .crop_and_resize_grad_boxes import _crop_and_resize_grad_boxes_aicpu
+from .crop_and_resize_grad_image import _crop_and_resize_grad_image_aicpu
+from .dense_to_dense_set_operation import _dense_to_dense_set_operation_aicpu
+from .div_no_nan import _div_no_nan_aicpu
+from .expm1 import _expm1_aicpu
+from .fractional_max_pool3d_with_fixed_ksize import _fractional_max_pool3d_with_fixed_ksize_aicpu
+from .fractional_max_pool3d_grad_with_fixed_ksize import _fractional_max_pool3d_grad_with_fixed_ksize_aicpu
+from .fractional_max_pool_with_fixed_ksize import _fractional_max_pool_with_fixed_ksize_aicpu
+from .hamming_window import _hamming_window_aicpu
+from .igamma import _igamma_aicpu
+from .igammac import _igammac_aicpu
+from .igammagrada import _igammagrada_aicpu
+from .imag import _imag_aicpu
+from .instance_norm_v2 import _instance_norm_v2_aicpu
+from .instance_norm_v2_grad import _instance_norm_v2_grad_aicpu
+from .layer_norm_grad_grad import _layernorm_grad_grad_aicpu
+from .list_diff import _list_diff_aicpu
+from .log import _log_aicpu
+from .logspace import _logspace_aicpu
+from .matrix_inverse import _matrix_inverse_aicpu
+from .matrix_power import _matrix_power_aicpu
+from .max_pool3d_grad_with_argmax import _max_pool3d_grad_with_argmax_aicpu
+from .max_pool3d_with_argmax import _max_pool3d_with_argmax_aicpu
+from .maxpool_grad_v1 import _maxpool_grad_v1_aicpu
+from .max_unpool2d import _max_unpool2d_aicpu
+from .max_unpool2d_grad import _max_unpool2d_grad_aicpu
+from .max_unpool3d import _max_unpool3d_aicpu
+from .max_unpool3d_grad import _max_unpool3d_grad_aicpu
+from .mvlgamma import _mvlgamma_aicpu
+from .mvlgamma_grad import _mvlgamma_grad_aicpu
+from .nextafter import _nextafter_aicpu
+from .non_deterministic_ints import _non_deterministic_ints_aicpu
+from .pow import _pow_aicpu
+from .real import _real_aicpu
+from .resize_area import _resize_area_aicpu
+from .segment_sum import _segment_sum_aicpu
+from .set_size import _set_size_aicpu
+from .slice import _slice_aicpu
+from .sparse_cross import _sparse_cross_aicpu
+from .sparse_slice import _sparse_slice_aicpu
+from .sparse_softmax import _sparse_softmax_aicpu
+from .sparse_tensor_dense_add import _sparse_tensor_dense_add_aicpu
+from .sparse_tensor_dense_mat_mul import _sparse_tensor_dense_mat_mul_aicpu
+from .trace import _trace_aicpu
+from .tracegrad import _tracegrad_aicpu
+from .tridiagonal_solve import _tridiagonal_solve_aicpu
+from .truncated_normal import _truncated_normal_aicpu
+from .glu import _glu_aicpu
+from .deformable_offsets import _deformable_offsets_aicpu
+from .deformable_offsets_grad import _deformable_offsets_grad_aicpu
+from .multi_margin_loss import _multi_margin_loss_aicpu
+from .multi_margin_loss_grad import _multi_margin_loss_grad_aicpu
+from .sparse_to_dense_v2 import _sparse_to_dense_v2_aicpu
+from .bernoulli import _bernoulli_aicpu
+from .glu_grad import _glu_grad_aicpu
+from .sspaddmm import _sspaddmm_aicpu
+from .sequence_addn import _sequence_addn_aicpu
+from .affine_grid import _affine_grid_aicpu

mindspore/ops/_op_impl/aicpu/abs.py

@@ -0,0 +1,36 @@
+# Copyright 2021 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+"""Abs op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+abs_op_info = AiCPURegOp("Abs") \
+    .fusion_type("ELEMWISE") \
+    .input(0, "x", "required") \
+    .output(0, "y", "required") \
+    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
+    .dtype_format(DataType.I16_Default, DataType.I16_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.F64_Default) \
+    .get_op_info()
+
+
+@op_info_register(abs_op_info)
+def _abs_aicpu():
+    """Abs AiCPU register"""
+    return

mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d.py

@@ -0,0 +1,34 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+
+"""AdaptiveAvgPool2D op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+adaptive_avg_pool_2d_op_info = AiCPURegOp("AdaptiveAvgPool2D") \
+    .fusion_type("OPAQUE") \
+    .input(0, "x", "required")\
+    .output(0, "y", "required") \
+    .attr("output_size", "listInt") \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.F64_Default) \
+    .get_op_info()
+
+
+@op_info_register(adaptive_avg_pool_2d_op_info)
+def _adaptive_avg_pool_2d_aicpu():
+    """AdaptiveAvgPool2D aicpu register"""
+    return

mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d_grad.py

@@ -0,0 +1,34 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+
+"""AdaptiveAvgPool2DGrad op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+adaptive_avg_pool_2d_grad_op_info = AiCPURegOp("AdaptiveAvgPool2DGrad") \
+    .fusion_type("OPAQUE") \
+    .input(0, "input_grad", "required")\
+    .input(1, "orig_input_shape", "required") \
+    .output(0, "output_grad", "required") \
+    .dtype_format(DataType.F16_Default, DataType.I64_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.I64_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.I64_Default, DataType.F64_Default) \
+    .get_op_info()
+
+
+@op_info_register(adaptive_avg_pool_2d_grad_op_info)
+def _adaptive_avg_pool_2d_grad_aicpu():
+    """AdaptiveAvgPool2DGrad aicpu register"""
+    return

mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d.py

@@ -0,0 +1,39 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+
+"""AdaptiveAvgPool3D op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+adaptiveavgpool3d_op_info = AiCPURegOp("AdaptiveAvgPool3D") \
+    .fusion_type("OPAQUE") \
+    .input(0, "x", "required")\
+    .output(0, "y", "required") \
+    .attr("output_size", "listInt") \
+    .dtype_format(DataType.I8_Default, DataType.I8_Default) \
+    .dtype_format(DataType.I16_Default, DataType.I16_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.I64_Default, DataType.I64_Default) \
+    .dtype_format(DataType.U8_Default, DataType.U8_Default) \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.F64_Default) \
+    .get_op_info()
+
+
+@op_info_register(adaptiveavgpool3d_op_info)
+def _adaptiveavgpool3d_aicpu():
+    """AdaptiveAvgPool3D aicpu register"""
+    return

mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d_grad.py

@@ -0,0 +1,39 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+
+"""AdaptiveAvgPool3DGrad op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+adaptiveavgpool3d_grad_op_info = AiCPURegOp("AdaptiveAvgPool3DGrad") \
+    .fusion_type("OPAQUE") \
+    .input(0, "input_grad", "required")\
+    .input(1, "orig_input_shape", "required") \
+    .output(0, "output_grad", "required") \
+    .dtype_format(DataType.I8_Default, DataType.I32_Default, DataType.I8_Default) \
+    .dtype_format(DataType.I16_Default, DataType.I32_Default, DataType.I16_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I64_Default) \
+    .dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.U8_Default) \
+    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.I32_Default, DataType.F64_Default) \
+    .get_op_info()
+
+
+@op_info_register(adaptiveavgpool3d_grad_op_info)
+def _adaptiveavgpool3d_grad_aicpu():
+    """AdaptiveAvgPool3DGrad aicpu register"""
+    return

mindspore/ops/_op_impl/aicpu/adaptive_max_pool_2d_grad.py

@@ -0,0 +1,37 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+"""AdaptiveMaxPool2DGrad op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+adaptive_max_pool_2d_grad_op_info = AiCPURegOp("AdaptiveMaxPool2DGrad") \
+    .fusion_type("OPAQUE") \
+    .input(0, "y_grad", "required") \
+    .input(1, "x", "required") \
+    .input(2, "argmax", "required") \
+    .output(0, "x_grad", "required") \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I32_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.I32_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.I64_Default, DataType.F16_Default) \
+    .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.I64_Default, DataType.F32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.I64_Default, DataType.F32_Default) \
+    .get_op_info()
+
+
+@op_info_register(adaptive_max_pool_2d_grad_op_info)
+def _adaptive_max_pool_2d_grad_aicpu():
+    """AdaptiveMaxPool2DGrad aicpu register"""
+    return

mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d.py

@@ -0,0 +1,42 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+"""AdaptiveMaxPool3D op"""
+from mindspore.ops.op_info_register import op_info_register, AiCPURegOp, DataType
+
+adaptive_max_pool_3d_op_info = AiCPURegOp("AdaptiveMaxPool3D")                                            \
+    .fusion_type("OPAQUE")                                                                                \
+    .input(0, "x", "required")                                                                            \
+    .input(1, "output_size", "required")                                                                  \
+    .output(0, "y", "required")                                                                           \
+    .output(1, "argmax", "required")                                                                      \
+    .dtype_format(DataType.F16_Default, DataType.I32_Default, DataType.F16_Default, DataType.I32_Default) \
+    .dtype_format(DataType.F32_Default, DataType.I32_Default, DataType.F32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.F64_Default, DataType.I32_Default, DataType.F64_Default, DataType.I32_Default) \
+    .dtype_format(DataType.I8_Default, DataType.I32_Default, DataType.I8_Default, DataType.I32_Default)   \
+    .dtype_format(DataType.I16_Default, DataType.I32_Default, DataType.I16_Default, DataType.I32_Default) \
+    .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.I64_Default, DataType.I32_Default, DataType.I64_Default, DataType.I32_Default) \
+    .dtype_format(DataType.U8_Default, DataType.I32_Default, DataType.U8_Default, DataType.I32_Default)   \
+    .dtype_format(DataType.U16_Default, DataType.I32_Default, DataType.U16_Default, DataType.I32_Default) \
+    .dtype_format(DataType.U32_Default, DataType.I32_Default, DataType.U32_Default, DataType.I32_Default) \
+    .dtype_format(DataType.U64_Default, DataType.I32_Default, DataType.U64_Default, DataType.I32_Default) \
+    .get_op_info()
+
+
+@op_info_register(adaptive_max_pool_3d_op_info)
+def _adaptive_max_pool_3d_aicpu():
+    """AdaptiveMaxPool3D aicpu register"""
+    return