mindspore 2.0.0a0__cp38-cp38-win_amd64.whl → 2.0.0rc1__cp38-cp38-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of mindspore might be problematic. Click here for more details.
- mindspore/.commit_id +1 -1
- mindspore/__init__.py +4 -2
- mindspore/_c_dataengine.cp38-win_amd64.pyd +0 -0
- mindspore/_c_expression.cp38-win_amd64.pyd +0 -0
- mindspore/_c_mindrecord.cp38-win_amd64.pyd +0 -0
- mindspore/_check_jit_forbidden_api.py +102 -0
- mindspore/_checkparam.py +1066 -1001
- mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +4 -3
- mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +50 -48
- mindspore/_extends/parallel_compile/akg_compiler/util.py +9 -4
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +4 -4
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +9 -4
- mindspore/_extends/parse/__init__.py +5 -3
- mindspore/_extends/parse/namespace.py +16 -1
- mindspore/_extends/parse/parser.py +107 -22
- mindspore/_extends/parse/resources.py +0 -7
- mindspore/_extends/parse/standard_method.py +885 -413
- mindspore/amp.py +52 -57
- mindspore/boost/boost.py +2 -2
- mindspore/boost/boost_cell_wrapper.py +38 -20
- mindspore/boost/dim_reduce.py +3 -3
- mindspore/boost/group_loss_scale_manager.py +1 -1
- mindspore/common/__init__.py +4 -6
- mindspore/common/_decorator.py +2 -0
- mindspore/common/_register_for_adapter.py +55 -0
- mindspore/common/_stub_tensor.py +201 -0
- mindspore/common/_utils.py +41 -7
- mindspore/common/api.py +215 -141
- mindspore/common/dtype.py +8 -1
- mindspore/common/dump.py +2 -2
- mindspore/common/initializer.py +4 -2
- mindspore/common/jit_config.py +17 -13
- mindspore/common/mutable.py +33 -13
- mindspore/common/parameter.py +23 -21
- mindspore/common/seed.py +8 -24
- mindspore/common/sparse_tensor.py +62 -41
- mindspore/common/tensor.py +852 -1154
- mindspore/communication/__init__.py +2 -2
- mindspore/communication/_comm_helper.py +11 -4
- mindspore/communication/management.py +22 -21
- mindspore/config/op_info.config +501 -1008
- mindspore/context.py +201 -23
- mindspore/dataset/__init__.py +6 -6
- mindspore/dataset/audio/__init__.py +7 -7
- mindspore/dataset/audio/transforms.py +670 -30
- mindspore/dataset/audio/utils.py +47 -4
- mindspore/dataset/audio/validators.py +223 -1
- mindspore/dataset/callback/ds_callback.py +2 -2
- mindspore/dataset/core/config.py +210 -14
- mindspore/dataset/core/validator_helpers.py +2 -2
- mindspore/{parallel/nn/layers.py → dataset/debug/__init__.py} +7 -8
- mindspore/dataset/debug/debug_hook.py +65 -0
- mindspore/dataset/debug/pre_defined_hook.py +67 -0
- mindspore/dataset/engine/__init__.py +7 -3
- mindspore/dataset/engine/cache_client.py +1 -1
- mindspore/dataset/engine/datasets.py +322 -66
- mindspore/dataset/engine/datasets_audio.py +80 -76
- mindspore/dataset/engine/datasets_standard_format.py +51 -38
- mindspore/dataset/engine/datasets_text.py +232 -118
- mindspore/dataset/engine/datasets_user_defined.py +41 -17
- mindspore/dataset/engine/datasets_vision.py +746 -225
- mindspore/dataset/engine/graphdata.py +75 -10
- mindspore/dataset/engine/iterators.py +45 -5
- mindspore/dataset/engine/offload.py +48 -28
- mindspore/dataset/engine/validators.py +117 -8
- mindspore/dataset/text/__init__.py +6 -5
- mindspore/dataset/text/transforms.py +86 -3
- mindspore/dataset/text/utils.py +6 -4
- mindspore/dataset/text/validators.py +25 -0
- mindspore/dataset/transforms/__init__.py +3 -2
- mindspore/dataset/transforms/c_transforms.py +1 -1
- mindspore/dataset/transforms/transforms.py +2 -2
- mindspore/dataset/utils/__init__.py +2 -1
- mindspore/dataset/utils/line_reader.py +121 -0
- mindspore/dataset/vision/__init__.py +2 -3
- mindspore/dataset/vision/c_transforms.py +9 -9
- mindspore/dataset/vision/py_transforms.py +5 -5
- mindspore/dataset/vision/py_transforms_util.py +2 -0
- mindspore/dataset/vision/transforms.py +160 -161
- mindspore/dataset/vision/utils.py +3 -3
- mindspore/experimental/map_parameter.py +38 -26
- mindspore/include/OWNERS +0 -1
- mindspore/include/api/callback/callback.h +9 -13
- mindspore/include/api/callback/ckpt_saver.h +2 -2
- mindspore/include/api/callback/loss_monitor.h +2 -2
- mindspore/include/api/callback/lr_scheduler.h +5 -5
- mindspore/include/api/callback/time_monitor.h +2 -2
- mindspore/include/api/callback/train_accuracy.h +4 -6
- mindspore/include/api/cfg.h +19 -6
- mindspore/include/api/context.h +44 -9
- mindspore/include/api/delegate.h +1 -1
- mindspore/include/api/metrics/accuracy.h +2 -2
- mindspore/include/api/metrics/metrics.h +4 -3
- mindspore/include/api/model.h +9 -4
- mindspore/include/api/model_parallel_runner.h +2 -2
- mindspore/include/api/net.h +12 -11
- mindspore/include/api/serialization.h +19 -3
- mindspore/include/api/types.h +3 -3
- mindspore/include/dataset/constants.h +7 -0
- mindspore/include/dataset/text.h +59 -0
- mindspore/jpeg62.dll +0 -0
- mindspore/log.py +1 -1
- mindspore/mindrecord/filereader.py +18 -0
- mindspore/mindrecord/filewriter.py +197 -34
- mindspore/mindrecord/shardreader.py +9 -0
- mindspore/mindrecord/shardwriter.py +1 -1
- mindspore/mindrecord/tools/cifar100_to_mr.py +3 -3
- mindspore/mindrecord/tools/cifar10_to_mr.py +3 -3
- mindspore/mindrecord/tools/csv_to_mr.py +3 -3
- mindspore/mindrecord/tools/imagenet_to_mr.py +16 -11
- mindspore/mindrecord/tools/mnist_to_mr.py +2 -2
- mindspore/mindrecord/tools/tfrecord_to_mr.py +6 -6
- mindspore/mindspore_backend.dll +0 -0
- mindspore/mindspore_common.dll +0 -0
- mindspore/mindspore_core.dll +0 -0
- mindspore/mindspore_glog.dll +0 -0
- mindspore/mindspore_shared_lib.dll +0 -0
- mindspore/nn/__init__.py +0 -4
- mindspore/nn/cell.py +204 -132
- mindspore/nn/dynamic_lr.py +1 -1
- mindspore/nn/grad/cell_grad.py +7 -6
- mindspore/nn/layer/__init__.py +5 -4
- mindspore/nn/layer/activation.py +40 -89
- mindspore/nn/layer/basic.py +255 -624
- mindspore/nn/layer/channel_shuffle.py +7 -6
- mindspore/nn/layer/combined.py +1 -1
- mindspore/nn/layer/container.py +41 -4
- mindspore/nn/layer/conv.py +64 -28
- mindspore/nn/layer/dense.py +9 -8
- mindspore/nn/layer/embedding.py +27 -25
- mindspore/nn/layer/image.py +53 -46
- mindspore/nn/layer/math.py +97 -105
- mindspore/nn/layer/normalization.py +117 -86
- mindspore/nn/layer/padding.py +185 -95
- mindspore/nn/layer/pooling.py +817 -414
- mindspore/nn/layer/rnn_cells.py +10 -15
- mindspore/nn/layer/rnns.py +37 -38
- mindspore/nn/layer/thor_layer.py +11 -12
- mindspore/nn/layer/timedistributed.py +5 -5
- mindspore/nn/layer/transformer.py +701 -0
- mindspore/nn/learning_rate_schedule.py +8 -8
- mindspore/nn/loss/__init__.py +5 -4
- mindspore/nn/loss/loss.py +334 -199
- mindspore/nn/optim/ada_grad.py +6 -6
- mindspore/nn/optim/adadelta.py +2 -3
- mindspore/nn/optim/adafactor.py +4 -5
- mindspore/nn/optim/adam.py +126 -62
- mindspore/nn/optim/adamax.py +3 -4
- mindspore/nn/optim/adasum.py +6 -6
- mindspore/nn/optim/asgd.py +2 -2
- mindspore/nn/optim/ftrl.py +67 -38
- mindspore/nn/optim/lamb.py +4 -5
- mindspore/nn/optim/lars.py +2 -2
- mindspore/nn/optim/lazyadam.py +43 -4
- mindspore/nn/optim/momentum.py +6 -5
- mindspore/nn/optim/optimizer.py +3 -1
- mindspore/nn/optim/proximal_ada_grad.py +2 -2
- mindspore/nn/optim/rmsprop.py +1 -1
- mindspore/nn/optim/rprop.py +8 -9
- mindspore/nn/optim/sgd.py +19 -13
- mindspore/nn/optim/thor.py +10 -15
- mindspore/nn/probability/__init__.py +0 -2
- mindspore/nn/probability/bijector/bijector.py +4 -4
- mindspore/nn/probability/bijector/invert.py +1 -1
- mindspore/nn/probability/bijector/softplus.py +2 -2
- mindspore/nn/probability/bnn_layers/dense_variational.py +1 -1
- mindspore/nn/probability/bnn_layers/layer_distribution.py +2 -2
- mindspore/nn/probability/distribution/_utils/utils.py +9 -15
- mindspore/nn/probability/distribution/bernoulli.py +3 -3
- mindspore/nn/probability/distribution/beta.py +1 -1
- mindspore/nn/probability/distribution/categorical.py +5 -7
- mindspore/nn/probability/distribution/cauchy.py +3 -3
- mindspore/nn/probability/distribution/distribution.py +2 -2
- mindspore/nn/probability/distribution/exponential.py +2 -2
- mindspore/nn/probability/distribution/gamma.py +3 -3
- mindspore/nn/probability/distribution/geometric.py +1 -1
- mindspore/nn/probability/distribution/gumbel.py +3 -3
- mindspore/nn/probability/distribution/half_normal.py +15 -11
- mindspore/nn/probability/distribution/laplace.py +16 -13
- mindspore/nn/probability/distribution/logistic.py +2 -2
- mindspore/nn/probability/distribution/normal.py +1 -1
- mindspore/nn/probability/distribution/poisson.py +1 -1
- mindspore/nn/probability/distribution/student_t.py +20 -15
- mindspore/nn/probability/distribution/transformed_distribution.py +4 -4
- mindspore/nn/probability/distribution/uniform.py +2 -2
- mindspore/nn/reinforcement/_tensors_queue.py +3 -3
- mindspore/nn/reinforcement/tensor_array.py +2 -2
- mindspore/nn/sparse/sparse.py +2 -2
- mindspore/nn/wrap/cell_wrapper.py +27 -10
- mindspore/nn/wrap/grad_reducer.py +2 -2
- mindspore/nn/wrap/loss_scale.py +40 -24
- mindspore/numpy/array_creations.py +33 -22
- mindspore/numpy/array_ops.py +35 -30
- mindspore/numpy/logic_ops.py +6 -27
- mindspore/numpy/math_ops.py +22 -19
- mindspore/numpy/utils.py +1 -1
- mindspore/numpy/utils_const.py +108 -58
- mindspore/opencv_core452.dll +0 -0
- mindspore/opencv_imgcodecs452.dll +0 -0
- mindspore/opencv_imgproc452.dll +0 -0
- mindspore/ops/_constants.py +0 -6
- mindspore/ops/_grad/__init__.py +2 -1
- mindspore/ops/_grad/grad_array_ops.py +86 -117
- mindspore/ops/_grad/grad_base.py +23 -1
- mindspore/ops/_grad/grad_clip_ops.py +2 -3
- mindspore/ops/_grad/grad_comm_ops.py +34 -24
- mindspore/ops/_grad/grad_implementations.py +9 -45
- mindspore/ops/_grad/grad_inner_ops.py +47 -4
- mindspore/ops/_grad/grad_math_ops.py +142 -117
- mindspore/ops/_grad/grad_nn_ops.py +71 -165
- mindspore/ops/_grad/grad_sequence_ops.py +296 -0
- mindspore/ops/_grad/grad_sparse.py +7 -6
- mindspore/ops/_grad_experimental/__init__.py +1 -0
- mindspore/ops/_grad_experimental/grad_array_ops.py +150 -15
- mindspore/ops/_grad_experimental/grad_image_ops.py +16 -7
- mindspore/ops/_grad_experimental/grad_inner_ops.py +1 -22
- mindspore/ops/_grad_experimental/grad_linalg_ops.py +4 -11
- mindspore/ops/_grad_experimental/grad_math_ops.py +210 -89
- mindspore/ops/_grad_experimental/grad_nn_ops.py +26 -22
- mindspore/ops/_grad_experimental/grad_scalar_ops.py +112 -0
- mindspore/ops/_grad_experimental/grad_sparse_ops.py +49 -8
- mindspore/ops/_op_impl/_custom_op/batch_matmul_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py +4 -4
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold_grad.py +3 -3
- mindspore/ops/_op_impl/_custom_op/cholesky_trsm_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/correction_mul.py +2 -2
- mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +1 -5
- mindspore/ops/_op_impl/_custom_op/dsd_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fused_abs_max1_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/img2col_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_right_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_impl.py +2 -2
- mindspore/ops/_op_impl/_custom_op/matmul_dds_impl.py +0 -4
- mindspore/ops/_op_impl/_custom_op/matrix_combine_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/transpose02314_impl.py +1 -1
- mindspore/ops/_op_impl/aicpu/__init__.py +236 -4
- mindspore/ops/_op_impl/aicpu/abs.py +36 -0
- mindspore/ops/_op_impl/aicpu/{adaptive_avg_pool_2d_v1.py → adaptive_avg_pool_2d.py} +6 -5
- mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d_grad.py +34 -0
- mindspore/ops/_op_impl/aicpu/add.py +43 -0
- mindspore/ops/_op_impl/aicpu/addcdiv.py +0 -32
- mindspore/ops/_op_impl/aicpu/addcmul.py +0 -84
- mindspore/ops/_op_impl/aicpu/affine_grid_grad.py +35 -0
- mindspore/ops/_op_impl/aicpu/batch_matmul.py +43 -43
- mindspore/ops/_op_impl/aicpu/bernoulli.py +48 -0
- mindspore/{compression/common/__init__.py → ops/_op_impl/aicpu/bessel_i0.py} +15 -8
- mindspore/ops/_op_impl/aicpu/channel_shuffle.py +40 -0
- mindspore/ops/_op_impl/aicpu/conj.py +11 -0
- mindspore/ops/_op_impl/aicpu/cumulative_logsumexp.py +0 -3
- mindspore/ops/_op_impl/aicpu/deformable_offsets.py +38 -0
- mindspore/ops/_op_impl/aicpu/deformable_offsets_grad.py +43 -0
- mindspore/ops/_op_impl/aicpu/{adaptive_avg_pool_2d_grad_v1.py → digamma.py} +7 -9
- mindspore/ops/_op_impl/aicpu/flatten.py +1 -0
- mindspore/ops/_op_impl/aicpu/fmax.py +36 -0
- mindspore/ops/_op_impl/aicpu/fmin.py +37 -0
- mindspore/ops/_op_impl/aicpu/fractional_max_pool3d_with_fixed_ksize.py +1 -1
- mindspore/ops/_op_impl/aicpu/fse_decode.py +43 -0
- mindspore/ops/_op_impl/aicpu/greater.py +41 -0
- mindspore/ops/_op_impl/aicpu/greater_equal.py +41 -0
- mindspore/ops/_op_impl/aicpu/index_put.py +50 -0
- mindspore/ops/_op_impl/aicpu/less.py +41 -0
- mindspore/{nn/probability/infer/variational/__init__.py → ops/_op_impl/aicpu/lgamma.py} +16 -10
- mindspore/ops/_op_impl/aicpu/mirror_pad.py +0 -4
- mindspore/ops/_op_impl/aicpu/mirror_pad_grad.py +0 -4
- mindspore/ops/_op_impl/aicpu/mul.py +3 -1
- mindspore/ops/_op_impl/aicpu/multinomial.py +14 -6
- mindspore/ops/_op_impl/aicpu/nllloss.py +38 -0
- mindspore/ops/_op_impl/aicpu/nllloss_grad.py +39 -0
- mindspore/ops/_op_impl/aicpu/ones_like.py +0 -2
- mindspore/ops/_op_impl/aicpu/polar.py +32 -0
- mindspore/ops/_op_impl/aicpu/polygamma.py +34 -0
- mindspore/ops/_op_impl/aicpu/quant_dtype_cast.py +40 -0
- mindspore/ops/_op_impl/aicpu/quantile.py +35 -0
- mindspore/ops/_op_impl/aicpu/ragged_tensor_to_sparse.py +73 -0
- mindspore/ops/_op_impl/aicpu/randperm_v2.py +41 -0
- mindspore/ops/_op_impl/aicpu/resize_bicubic.py +2 -8
- mindspore/ops/_op_impl/aicpu/resize_bicubic_grad.py +1 -1
- mindspore/ops/_op_impl/aicpu/resize_v2.py +68 -0
- mindspore/ops/_op_impl/aicpu/resize_v2_grad.py +68 -0
- mindspore/ops/_op_impl/aicpu/scatter_elements.py +4 -0
- mindspore/ops/_op_impl/aicpu/scatter_nd_update.py +2 -0
- mindspore/ops/_op_impl/aicpu/sequence_add.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_add_offset.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_addn.py +38 -0
- mindspore/ops/_op_impl/aicpu/smooth_l1_loss.py +35 -0
- mindspore/ops/_op_impl/aicpu/smooth_l1_loss_grad.py +37 -0
- mindspore/ops/_op_impl/aicpu/sparse_apply_adagrad_da.py +0 -24
- mindspore/ops/_op_impl/aicpu/sparse_cross.py +42 -0
- mindspore/ops/_op_impl/aicpu/sparse_slice.py +4 -0
- mindspore/ops/_op_impl/aicpu/sparse_slice_grad.py +6 -0
- mindspore/ops/_op_impl/aicpu/tensor_scatter_update.py +59 -0
- mindspore/ops/_op_impl/aicpu/trans_data.py +1 -0
- mindspore/ops/_op_impl/aicpu/tril_indices.py +34 -0
- mindspore/ops/_op_impl/aicpu/uniform.py +34 -0
- mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +1 -0
- mindspore/ops/_op_impl/aicpu/unique_consecutive.py +10 -2
- mindspore/ops/_op_impl/cpu/dynamic_shape.py +5 -1
- mindspore/ops/_op_impl/cpu/sparse_slice.py +4 -0
- mindspore/ops/_op_impl/cpu/sparse_slice_grad.py +6 -0
- mindspore/ops/_op_impl/cpu/tensor_shape.py +5 -1
- mindspore/ops/_op_impl/tbe/__init__.py +27 -611
- mindspore/ops/_op_impl/tbe/assign_add_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_matmul_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +1 -1
- mindspore/ops/_op_impl/tbe/bn_infer_grad.py +4 -2
- mindspore/ops/_op_impl/tbe/bn_training_update.py +0 -1
- mindspore/ops/_op_impl/tbe/bn_training_update_ds.py +0 -1
- mindspore/ops/_op_impl/tbe/broadcast_to_ds.py +6 -4
- mindspore/ops/_op_impl/tbe/cast.py +0 -2
- mindspore/ops/_op_impl/tbe/cast_ds.py +3 -3
- mindspore/ops/_op_impl/tbe/data_format_dim_map_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +2 -2
- mindspore/ops/_op_impl/tbe/dynamic_atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/gather_nd.py +1 -0
- mindspore/ops/_op_impl/tbe/{index_add.py → inplace_index_add.py} +3 -6
- mindspore/ops/_op_impl/tbe/matmul_ds.py +2 -0
- mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py +35 -0
- mindspore/ops/_op_impl/tbe/npu_get_float_status_v2.py +35 -0
- mindspore/ops/_op_impl/tbe/scatter_mul.py +2 -0
- mindspore/ops/_op_impl/tbe/scatter_nd_add.py +0 -2
- mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
- mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +1 -1
- mindspore/ops/_op_impl/tbe/trans_data_ds.py +15 -5
- mindspore/ops/_register_for_op.py +1 -0
- mindspore/ops/_utils/__init__.py +1 -2
- mindspore/ops/_utils/utils.py +19 -40
- mindspore/ops/_vmap/vmap_array_ops.py +116 -38
- mindspore/ops/_vmap/vmap_base.py +16 -9
- mindspore/ops/_vmap/vmap_convolution_ops.py +7 -10
- mindspore/ops/_vmap/vmap_grad_math_ops.py +4 -4
- mindspore/ops/_vmap/vmap_grad_nn_ops.py +7 -5
- mindspore/ops/_vmap/vmap_image_ops.py +12 -5
- mindspore/ops/_vmap/vmap_math_ops.py +46 -5
- mindspore/ops/_vmap/vmap_nn_ops.py +15 -21
- mindspore/ops/_vmap/vmap_random_ops.py +1 -1
- mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +150 -0
- mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +66 -0
- mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +33 -0
- mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +220 -106
- mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +240 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +247 -0
- mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +247 -0
- mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +315 -0
- mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +278 -0
- mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +58 -0
- mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +138 -0
- mindspore/ops/bprop_mindir/Dropout2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Dropout3D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DropoutDoMask_bprop.mindir +22 -23
- mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +16 -17
- mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +27 -0
- mindspore/ops/bprop_mindir/Dropout_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicGRUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicRNN_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Elu_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +39 -41
- mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Flatten_bprop.mindir +41 -43
- mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +51 -57
- mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/HSwish_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +126 -0
- mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +15 -0
- mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +30 -0
- mindspore/ops/bprop_mindir/LRN_bprop.mindir +43 -0
- mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +23 -0
- mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +74 -0
- mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +74 -0
- mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +75 -0
- mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +65 -0
- mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +27 -0
- mindspore/ops/bprop_mindir/Mish_bprop.mindir +35 -0
- mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/OneHot_bprop.mindir +24 -25
- mindspore/ops/bprop_mindir/PReLU_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Pad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Padding_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/RNNTLoss_bprop.mindir +29 -0
- mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +82 -0
- mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/Reshape_bprop.mindir +53 -53
- mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +29 -0
- mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +77 -85
- mindspore/ops/bprop_mindir/SeLU_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +36 -0
- mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Softplus_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Softsign_bprop.mindir +33 -0
- mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +37 -39
- mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +70 -72
- mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Tanh_bprop.mindir +66 -0
- mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +17 -17
- mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +32 -0
- mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +38 -0
- mindspore/ops/bprop_mindir/generate_mindir.py +2 -0
- mindspore/ops/composite/__init__.py +7 -8
- mindspore/ops/composite/base.py +101 -47
- mindspore/ops/composite/math_ops.py +188 -158
- mindspore/ops/composite/multitype_ops/_compile_utils.py +415 -170
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +142 -87
- mindspore/ops/composite/multitype_ops/add_impl.py +6 -1
- mindspore/ops/composite/multitype_ops/div_impl.py +2 -3
- mindspore/ops/composite/multitype_ops/getitem_impl.py +31 -3
- mindspore/ops/composite/multitype_ops/greater_equal_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/greater_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/in_impl.py +9 -0
- mindspore/ops/composite/multitype_ops/less_equal_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/less_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/mul_impl.py +21 -5
- mindspore/ops/composite/multitype_ops/not_in_impl.py +9 -0
- mindspore/ops/composite/multitype_ops/ones_like_impl.py +2 -4
- mindspore/ops/composite/multitype_ops/setitem_impl.py +21 -3
- mindspore/ops/composite/multitype_ops/sub_impl.py +1 -1
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +35 -4
- mindspore/ops/function/__init__.py +152 -8
- mindspore/ops/function/array_func.py +2555 -674
- mindspore/ops/function/clip_func.py +209 -13
- mindspore/ops/function/debug_func.py +2 -2
- mindspore/ops/function/grad/__init__.py +2 -1
- mindspore/ops/function/grad/grad_func.py +147 -62
- mindspore/ops/function/image_func.py +54 -38
- mindspore/ops/function/linalg_func.py +167 -16
- mindspore/ops/function/math_func.py +4849 -1492
- mindspore/ops/function/nn_func.py +2573 -988
- mindspore/ops/function/other_func.py +115 -0
- mindspore/ops/function/parameter_func.py +3 -3
- mindspore/ops/function/random_func.py +790 -73
- mindspore/ops/function/sparse_func.py +98 -78
- mindspore/ops/function/sparse_unary_func.py +54 -53
- mindspore/ops/function/spectral_func.py +27 -24
- mindspore/ops/function/vmap_func.py +22 -2
- mindspore/ops/functional.py +97 -37
- mindspore/ops/op_info_register.py +70 -28
- mindspore/ops/operations/__init__.py +47 -14
- mindspore/ops/operations/_csr_ops.py +7 -7
- mindspore/ops/operations/_embedding_cache_ops.py +5 -5
- mindspore/ops/operations/_grad_ops.py +276 -187
- mindspore/ops/operations/_inner_ops.py +319 -113
- mindspore/ops/operations/_ms_kernel.py +10 -8
- mindspore/ops/operations/_ocr_ops.py +9 -9
- mindspore/ops/operations/_opaque_predicate_registry.py +4 -0
- mindspore/ops/operations/_quant_ops.py +137 -102
- mindspore/ops/operations/_rl_inner_ops.py +121 -60
- mindspore/ops/operations/_scalar_ops.py +466 -0
- mindspore/ops/operations/_sequence_ops.py +1004 -2
- mindspore/ops/operations/_tensor_array.py +10 -11
- mindspore/ops/operations/_thor_ops.py +1 -1
- mindspore/ops/operations/array_ops.py +801 -466
- mindspore/ops/operations/comm_ops.py +51 -49
- mindspore/ops/operations/control_ops.py +2 -2
- mindspore/ops/operations/custom_ops.py +123 -44
- mindspore/ops/operations/debug_ops.py +24 -24
- mindspore/ops/operations/image_ops.py +240 -153
- mindspore/ops/operations/inner_ops.py +34 -50
- mindspore/ops/operations/linalg_ops.py +31 -9
- mindspore/ops/operations/math_ops.py +988 -757
- mindspore/ops/operations/nn_ops.py +965 -819
- mindspore/ops/operations/other_ops.py +51 -40
- mindspore/ops/operations/random_ops.py +204 -122
- mindspore/ops/operations/rl_ops.py +8 -9
- mindspore/ops/operations/sparse_ops.py +254 -93
- mindspore/ops/operations/spectral_ops.py +35 -3
- mindspore/ops/primitive.py +111 -9
- mindspore/parallel/_auto_parallel_context.py +189 -83
- mindspore/parallel/_offload_context.py +185 -0
- mindspore/parallel/_parallel_serialization.py +99 -7
- mindspore/parallel/_ps_context.py +9 -5
- mindspore/parallel/_recovery_context.py +1 -1
- mindspore/parallel/_tensor.py +7 -1
- mindspore/{nn/transformer → parallel/_transformer}/__init__.py +6 -6
- mindspore/{nn/transformer → parallel/_transformer}/layers.py +6 -37
- mindspore/{nn/transformer → parallel/_transformer}/loss.py +4 -7
- mindspore/{nn/transformer → parallel/_transformer}/moe.py +20 -16
- mindspore/{nn/transformer → parallel/_transformer}/op_parallel_config.py +3 -3
- mindspore/{nn/transformer → parallel/_transformer}/transformer.py +48 -111
- mindspore/parallel/_utils.py +1 -2
- mindspore/parallel/algo_parameter_config.py +1 -1
- mindspore/parallel/checkpoint_transform.py +37 -34
- mindspore/parallel/shard.py +17 -18
- mindspore/profiler/common/validator/validate_path.py +2 -2
- mindspore/profiler/envprofiling.py +69 -47
- mindspore/profiler/parser/ascend_timeline_generator.py +49 -42
- mindspore/profiler/parser/base_timeline_generator.py +49 -56
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +98 -78
- mindspore/profiler/parser/hwts_log_parser.py +1 -1
- mindspore/profiler/parser/integrator.py +15 -14
- mindspore/profiler/parser/minddata_analyzer.py +2 -2
- mindspore/profiler/parser/msadvisor_analyzer.py +12 -25
- mindspore/profiler/parser/msadvisor_parser.py +2 -4
- mindspore/profiler/parser/optime_parser.py +17 -18
- mindspore/profiler/parser/profiler_info.py +2 -1
- mindspore/profiler/profiling.py +218 -186
- mindspore/rewrite/__init__.py +3 -1
- mindspore/rewrite/api/node.py +1 -114
- mindspore/rewrite/api/node_type.py +3 -0
- mindspore/rewrite/api/pattern_engine.py +31 -1
- mindspore/rewrite/api/scoped_value.py +4 -4
- mindspore/rewrite/api/symbol_tree.py +3 -78
- mindspore/rewrite/api/tree_node_helper.py +1 -1
- mindspore/rewrite/ast_creator_register.py +1 -0
- mindspore/rewrite/ast_helpers/__init__.py +2 -2
- mindspore/rewrite/ast_helpers/ast_creator.py +1 -2
- mindspore/rewrite/ast_helpers/ast_finder.py +65 -0
- mindspore/rewrite/ast_helpers/ast_modifier.py +11 -3
- mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +18 -2
- mindspore/rewrite/namespace.py +0 -2
- mindspore/rewrite/node.py +157 -11
- mindspore/rewrite/parsers/assign_parser.py +231 -53
- mindspore/rewrite/parsers/class_def_parser.py +187 -109
- mindspore/rewrite/parsers/for_parser.py +24 -14
- mindspore/rewrite/parsers/function_def_parser.py +21 -4
- mindspore/rewrite/parsers/if_parser.py +6 -2
- mindspore/rewrite/sparsify/__init__.py +0 -0
- mindspore/rewrite/sparsify/sparse_transformer.py +448 -0
- mindspore/rewrite/sparsify/sparsify.py +109 -0
- mindspore/rewrite/sparsify/utils.py +173 -0
- mindspore/rewrite/symbol_tree.py +256 -133
- mindspore/rewrite/symbol_tree_builder.py +38 -1
- mindspore/run_check/_check_version.py +69 -63
- mindspore/run_check/run_check.py +2 -1
- mindspore/tinyxml2.dll +0 -0
- mindspore/train/__init__.py +1 -1
- mindspore/train/_utils.py +28 -5
- mindspore/train/amp.py +273 -102
- mindspore/train/callback/_backup_and_restore.py +5 -5
- mindspore/train/callback/_callback.py +2 -2
- mindspore/train/callback/_checkpoint.py +3 -3
- mindspore/train/callback/_early_stop.py +3 -3
- mindspore/train/callback/_lambda_callback.py +2 -2
- mindspore/train/callback/_landscape.py +29 -31
- mindspore/train/callback/_loss_monitor.py +3 -3
- mindspore/train/callback/_on_request_exit.py +3 -3
- mindspore/train/callback/_reduce_lr_on_plateau.py +4 -4
- mindspore/train/callback/_summary_collector.py +23 -16
- mindspore/train/callback/_time_monitor.py +3 -3
- mindspore/train/checkpoint_pb2.py +68 -8
- mindspore/train/data_sink.py +15 -3
- mindspore/train/dataset_helper.py +10 -15
- mindspore/train/loss_scale_manager.py +8 -11
- mindspore/train/metrics/__init__.py +1 -1
- mindspore/train/metrics/bleu_score.py +1 -1
- mindspore/train/metrics/confusion_matrix.py +1 -1
- mindspore/train/metrics/cosine_similarity.py +1 -1
- mindspore/train/metrics/dice.py +2 -2
- mindspore/train/metrics/fbeta.py +1 -1
- mindspore/train/metrics/hausdorff_distance.py +4 -3
- mindspore/train/metrics/mean_surface_distance.py +2 -2
- mindspore/train/metrics/occlusion_sensitivity.py +1 -1
- mindspore/train/metrics/perplexity.py +1 -1
- mindspore/train/metrics/precision.py +1 -1
- mindspore/train/metrics/recall.py +1 -1
- mindspore/train/metrics/roc.py +2 -2
- mindspore/train/metrics/root_mean_square_surface_distance.py +2 -2
- mindspore/train/mind_ir_pb2.py +116 -37
- mindspore/train/model.py +45 -28
- mindspore/train/serialization.py +295 -188
- mindspore/train/summary/_summary_adapter.py +1 -1
- mindspore/train/summary/summary_record.py +43 -13
- mindspore/train/train_thor/convert_utils.py +2 -2
- mindspore/train/train_thor/dataset_helper.py +3 -3
- mindspore/turbojpeg.dll +0 -0
- mindspore/version.py +1 -1
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/METADATA +3 -2
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/RECORD +610 -541
- mindspore/compression/__init__.py +0 -19
- mindspore/compression/common/constant.py +0 -124
- mindspore/compression/export/__init__.py +0 -19
- mindspore/compression/export/quant_export.py +0 -515
- mindspore/compression/quant/__init__.py +0 -28
- mindspore/compression/quant/qat.py +0 -634
- mindspore/compression/quant/quant_utils.py +0 -462
- mindspore/compression/quant/quantizer.py +0 -68
- mindspore/nn/layer/quant.py +0 -1868
- mindspore/nn/layer/rnn_utils.py +0 -90
- mindspore/nn/probability/dpn/__init__.py +0 -22
- mindspore/nn/probability/dpn/vae/__init__.py +0 -25
- mindspore/nn/probability/dpn/vae/cvae.py +0 -140
- mindspore/nn/probability/dpn/vae/vae.py +0 -124
- mindspore/nn/probability/infer/__init__.py +0 -22
- mindspore/nn/probability/infer/variational/elbo.py +0 -70
- mindspore/nn/probability/infer/variational/svi.py +0 -84
- mindspore/nn/probability/toolbox/__init__.py +0 -22
- mindspore/nn/probability/toolbox/anomaly_detection.py +0 -99
- mindspore/nn/probability/toolbox/uncertainty_evaluation.py +0 -364
- mindspore/nn/probability/transforms/__init__.py +0 -22
- mindspore/nn/probability/transforms/transform_bnn.py +0 -262
- mindspore/nn/probability/zhusuan/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/framework/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/framework/bn.py +0 -95
- mindspore/nn/probability/zhusuan/variational/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/variational/elbo.py +0 -46
- mindspore/ops/_op_impl/aicpu/parallel_concat.py +0 -42
- mindspore/ops/_op_impl/tbe/gather_v2.py +0 -56
- mindspore/ops/bprop_mindir/AssignAdd_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/Cast_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/LogicalOr_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/MatMul_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ReLU_bprop.mindir +0 -17
- mindspore/ops/bprop_mindir/Transpose_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/UpdateState_bprop.mindir +0 -15
- mindspore/ops/composite/array_ops.py +0 -241
- mindspore/ops/composite/clip_ops.py +0 -134
- mindspore/ops/composite/random_ops.py +0 -426
- mindspore/ops/composite/vmap_ops.py +0 -38
- mindspore/parallel/nn/__init__.py +0 -42
- mindspore/parallel/nn/loss.py +0 -22
- mindspore/parallel/nn/moe.py +0 -21
- mindspore/parallel/nn/op_parallel_config.py +0 -22
- mindspore/parallel/nn/transformer.py +0 -31
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/WHEEL +0 -0
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/entry_points.txt +0 -0
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/top_level.txt +0 -0
|
@@ -19,13 +19,13 @@ from __future__ import absolute_import
|
|
|
19
19
|
import mindspore.numpy as mnp
|
|
20
20
|
from mindspore.ops import operations as P
|
|
21
21
|
from mindspore.ops import functional as F
|
|
22
|
-
from mindspore.ops import
|
|
22
|
+
from mindspore.ops.primitive import _primexpr
|
|
23
23
|
from mindspore.common import Tensor
|
|
24
24
|
from mindspore.ops.operations import math_ops
|
|
25
25
|
from mindspore.ops.operations import linalg_ops
|
|
26
26
|
from mindspore.ops.operations import _inner_ops
|
|
27
27
|
from mindspore.ops.primitive import Primitive
|
|
28
|
-
from mindspore.ops.
|
|
28
|
+
from mindspore.ops.function import _VmapGeneralRule
|
|
29
29
|
from mindspore.ops._vmap.vmap_base import vmap_rules_getters, vmap_general_preprocess, get_assign_vmap_rule, \
|
|
30
30
|
get_unop_vmap_rule, _raise_value_error, _bdim_at_front, _broadcast_by_axis, _handle_broadcasting, \
|
|
31
31
|
_vmap_clone_prim, _bdim_at_any, _get_reduce_batch_axis, _get_reduce_out_dim
|
|
@@ -33,7 +33,7 @@ from mindspore.ops.operations.math_ops import Bernoulli, BesselI0, BesselI1, Bes
|
|
|
33
33
|
BesselK0, BesselK0e, BesselY0, BesselY1, BesselK1, BesselK1e, Median
|
|
34
34
|
|
|
35
35
|
|
|
36
|
-
@
|
|
36
|
+
@_primexpr
|
|
37
37
|
def _broadcast_shape(nd, x_ndim, x_shape):
|
|
38
38
|
return x_shape + (1,) * (nd - x_ndim)
|
|
39
39
|
|
|
@@ -111,6 +111,9 @@ def get_addcxxx_vmap_rule(prim, axis_size):
|
|
|
111
111
|
x1, x1_dim = x1_bdim
|
|
112
112
|
x2, x2_dim = x2_bdim
|
|
113
113
|
value, value_dim = value_bdim
|
|
114
|
+
if input_data_dim == x1_dim and x1_dim == x2_dim and x2_dim == value_dim:
|
|
115
|
+
out = prim(input_data, x1, x2, value)
|
|
116
|
+
return out, input_data_dim
|
|
114
117
|
|
|
115
118
|
input_data = _bdim_at_front(input_data, input_data_dim, axis_size)
|
|
116
119
|
x1 = _bdim_at_front(x1, x1_dim, axis_size)
|
|
@@ -433,7 +436,7 @@ def get_median_vmap_rule(prim, axis_size):
|
|
|
433
436
|
axis = prim.axis
|
|
434
437
|
keep_dims = prim.keep_dims
|
|
435
438
|
|
|
436
|
-
@
|
|
439
|
+
@_primexpr
|
|
437
440
|
def trans_axis(axis, rank, dim, keep_dims):
|
|
438
441
|
if axis < 0:
|
|
439
442
|
axis += rank - 1
|
|
@@ -461,7 +464,7 @@ def get_index_add_vmap_rule(prim, axis_size):
|
|
|
461
464
|
"""VmapRule for IndexAdd."""
|
|
462
465
|
axis = prim.axis
|
|
463
466
|
|
|
464
|
-
@
|
|
467
|
+
@_primexpr
|
|
465
468
|
def _get_index_add_batch_axis(axis, x_dim, x_ndim):
|
|
466
469
|
"""get batch_axis for IndexAdd."""
|
|
467
470
|
# case1: batch not exists
|
|
@@ -800,6 +803,44 @@ def get_square_sum_all_vmap_rule(prim, axis_size):
|
|
|
800
803
|
return vmap_rule
|
|
801
804
|
|
|
802
805
|
|
|
806
|
+
@vmap_rules_getters.register(math_ops.FFTWithSize)
|
|
807
|
+
def get_fft_with_size_vmap_rule(prim, axis_size):
|
|
808
|
+
"""VmapRule for `FFTWithSize` operation"""
|
|
809
|
+
if isinstance(prim, str):
|
|
810
|
+
prim_name = prim
|
|
811
|
+
prim = Primitive(prim)
|
|
812
|
+
signal_ndim = 1
|
|
813
|
+
inverse = False
|
|
814
|
+
real = False
|
|
815
|
+
norm = "backward"
|
|
816
|
+
oneside = True
|
|
817
|
+
signal_sizes = ()
|
|
818
|
+
else:
|
|
819
|
+
prim_name = prim.name
|
|
820
|
+
signal_ndim = prim.signal_ndim
|
|
821
|
+
inverse = prim.inverse
|
|
822
|
+
real = prim.real
|
|
823
|
+
norm = prim.norm
|
|
824
|
+
oneside = prim.oneside
|
|
825
|
+
signal_sizes = prim.signal_sizes
|
|
826
|
+
|
|
827
|
+
fft = math_ops.FFTWithSize(signal_ndim, inverse, real, norm, oneside, signal_sizes)
|
|
828
|
+
|
|
829
|
+
def vmap_rule(x_bdim):
|
|
830
|
+
is_all_none, result = vmap_general_preprocess(prim, x_bdim)
|
|
831
|
+
if is_all_none:
|
|
832
|
+
return result
|
|
833
|
+
x, x_dim = x_bdim
|
|
834
|
+
x_ndim = F.rank(x)
|
|
835
|
+
if x_dim < 0 or x_dim >= x_ndim - signal_ndim:
|
|
836
|
+
_raise_value_error("The source axi of `x` in `{} must be`in range of ({} {}), "
|
|
837
|
+
"but got {}.".format(prim_name, 0, x_ndim - signal_ndim, x_dim))
|
|
838
|
+
out = fft(x)
|
|
839
|
+
return (out, x_dim)
|
|
840
|
+
|
|
841
|
+
return vmap_rule
|
|
842
|
+
|
|
843
|
+
|
|
803
844
|
get_assign_vmap_rule = vmap_rules_getters.register(P.AssignAdd)(get_assign_vmap_rule)
|
|
804
845
|
get_assign_vmap_rule = vmap_rules_getters.register(P.AssignSub)(get_assign_vmap_rule)
|
|
805
846
|
|
|
@@ -23,6 +23,7 @@ from mindspore.ops.operations import _grad_ops as G
|
|
|
23
23
|
from mindspore.ops.operations import nn_ops as NN
|
|
24
24
|
from mindspore.ops import functional as F
|
|
25
25
|
from mindspore.ops import constexpr
|
|
26
|
+
from mindspore.ops.primitive import _primexpr
|
|
26
27
|
from mindspore.ops._vmap.vmap_base import vmap_rules_getters, vmap_general_preprocess, get_unop_vmap_rule, \
|
|
27
28
|
_bdim_at_any, _bdim_at_front, _bdim_at_back, _handle_broadcasting, get_unary_grad_vmap_rule, _raise_value_error, \
|
|
28
29
|
_vmap_clone_prim, _get_reduce_batch_axis
|
|
@@ -375,7 +376,7 @@ def get_bias_add_vmap_rule(prim, axis_size):
|
|
|
375
376
|
def get_channal_pos_in_x(d_format):
|
|
376
377
|
return d_format.find('C') + 1
|
|
377
378
|
|
|
378
|
-
@
|
|
379
|
+
@_primexpr
|
|
379
380
|
def get_bias_dst_shape(x_shape, n_dims, d_format, has_b_dim: bool):
|
|
380
381
|
pos = get_channal_pos_in_x(d_format)
|
|
381
382
|
|
|
@@ -430,7 +431,7 @@ def get_bias_add_grad_vmap_rule(prim, axis_size):
|
|
|
430
431
|
def get_channal_pos(d_format):
|
|
431
432
|
return d_format.find('C') + 1
|
|
432
433
|
|
|
433
|
-
@
|
|
434
|
+
@_primexpr
|
|
434
435
|
def get_axis_for_reduce(x_shape_rank, data_format):
|
|
435
436
|
channal_pos = get_channal_pos(data_format)
|
|
436
437
|
axis_list = ()
|
|
@@ -1332,10 +1333,9 @@ def get_adam_rule(prim, axis_size):
|
|
|
1332
1333
|
epsilon, epsilon_dim = epsilon_bdim
|
|
1333
1334
|
grad, grad_dim = grad_bdim
|
|
1334
1335
|
|
|
1336
|
+
all_dim = [m_dim, v_dim, beta1_power_dim, beta2_power_dim, lr_dim, beta1_dim, beta2_dim, epsilon_dim, grad_dim]
|
|
1335
1337
|
if var_dim is None:
|
|
1336
|
-
if any(dim is not None for dim in
|
|
1337
|
-
beta2_power_dim, lr_dim, beta1_dim,
|
|
1338
|
-
beta2_dim, epsilon_dim, grad_dim]):
|
|
1338
|
+
if any(dim is not None for dim in all_dim):
|
|
1339
1339
|
raise ValueError("The source axis of `var` is None, "
|
|
1340
1340
|
"but the source axis of `m/v/vhat/beta1_power/beta2_power/lr/beta1/beta2/epsilon grad"
|
|
1341
1341
|
" is not None. The execution of operator `{}` cannot be guaranteed.".format(prim_name))
|
|
@@ -1516,10 +1516,9 @@ def get_adaptive_max_pool_2d_vmap_rule(prim, axis_size):
|
|
|
1516
1516
|
nchw_index = 4
|
|
1517
1517
|
chw_reverse_index = -3
|
|
1518
1518
|
hw_size = 2
|
|
1519
|
-
return_indices = prim.return_indices
|
|
1520
1519
|
output_size = prim.output_size
|
|
1521
1520
|
|
|
1522
|
-
@
|
|
1521
|
+
@_primexpr
|
|
1523
1522
|
def get_output_shape(x_ori_shape, output_size):
|
|
1524
1523
|
if isinstance(output_size, tuple):
|
|
1525
1524
|
h_out, w_out = output_size
|
|
@@ -1554,20 +1553,14 @@ def get_adaptive_max_pool_2d_vmap_rule(prim, axis_size):
|
|
|
1554
1553
|
x_ori_shape = F.shape(x)
|
|
1555
1554
|
x = F.reshape(x, (-1,) + x_ori_shape[chw_reverse_index:])
|
|
1556
1555
|
output_shape = get_output_shape(x_ori_shape, output_size)
|
|
1557
|
-
if return_indices:
|
|
1558
|
-
out, indices = prim(x)
|
|
1559
|
-
out = F.reshape(out, output_shape)
|
|
1560
|
-
indices = F.reshape(indices, output_shape)
|
|
1561
|
-
return (out, 0), (indices, 0)
|
|
1562
|
-
out = prim(x)
|
|
1563
|
-
out = F.reshape(out, output_shape)
|
|
1564
|
-
return out, 0
|
|
1565
|
-
# for the case of CHW
|
|
1566
|
-
if return_indices:
|
|
1567
1556
|
out, indices = prim(x)
|
|
1557
|
+
out = F.reshape(out, output_shape)
|
|
1558
|
+
indices = F.reshape(indices, output_shape)
|
|
1568
1559
|
return (out, 0), (indices, 0)
|
|
1569
|
-
|
|
1570
|
-
|
|
1560
|
+
|
|
1561
|
+
# for the case of CHW
|
|
1562
|
+
out, indices = prim(x)
|
|
1563
|
+
return (out, 0), (indices, 0)
|
|
1571
1564
|
|
|
1572
1565
|
return vmap_rule
|
|
1573
1566
|
|
|
@@ -1712,6 +1705,7 @@ def get_apply_centered_rmsprop_vmap_rule(prim, axis_size):
|
|
|
1712
1705
|
|
|
1713
1706
|
@vmap_rules_getters.register(P.MaxPool)
|
|
1714
1707
|
@vmap_rules_getters.register(P.MaxPoolWithArgmax)
|
|
1708
|
+
@vmap_rules_getters.register(P.MaxPoolWithArgmaxV2)
|
|
1715
1709
|
def get_max_pool_vmap_rule(prim, axis_size):
|
|
1716
1710
|
"""VmapRule for `MaxPool` operation."""
|
|
1717
1711
|
if isinstance(prim, str):
|
|
@@ -1719,7 +1713,7 @@ def get_max_pool_vmap_rule(prim, axis_size):
|
|
|
1719
1713
|
|
|
1720
1714
|
prim_name = prim.name
|
|
1721
1715
|
|
|
1722
|
-
@
|
|
1716
|
+
@_primexpr
|
|
1723
1717
|
def get_original_shape(x_shape, out_shape):
|
|
1724
1718
|
h_new = out_shape[2]
|
|
1725
1719
|
w_new = out_shape[3]
|
|
@@ -1764,7 +1758,7 @@ def get_layernorm_vmap_rule(prim, axis_size):
|
|
|
1764
1758
|
params_axis = process_attr_axis(prim.begin_params_axis)
|
|
1765
1759
|
batch_prim = P.LayerNorm(norm_axis, params_axis, prim.epsilon)
|
|
1766
1760
|
|
|
1767
|
-
@
|
|
1761
|
+
@_primexpr
|
|
1768
1762
|
def get_logical_shape(var_shape):
|
|
1769
1763
|
return var_shape[1:]
|
|
1770
1764
|
|
|
@@ -18,7 +18,7 @@ from __future__ import absolute_import
|
|
|
18
18
|
|
|
19
19
|
from mindspore.ops.operations.random_ops import UniformCandidateSampler, RandomShuffle, Multinomial, \
|
|
20
20
|
RandomChoiceWithMask
|
|
21
|
-
from mindspore.ops.
|
|
21
|
+
from mindspore.ops.function import _VmapGeneralRule
|
|
22
22
|
from mindspore.ops._vmap.vmap_base import vmap_rules_getters, _bdim_at_front, _vmap_clone_prim, \
|
|
23
23
|
vmap_general_preprocess, _raise_value_error
|
|
24
24
|
|
|
Binary file
|
|
Binary file
|
|
@@ -0,0 +1,150 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�)
|
|
3
|
+
�+get_bprop_avg_pool_3d_grad.822:[CNode]823:1+get_bprop_avg_pool_3d_grad.822:[CNode]823:1"REF::bprop.824:Default/bprop.824-op627
|
|
4
|
+
C[ValueNode]835[ValueNode]835"Constant*
|
|
5
|
+
value*kernel_size�
|
|
6
|
+
?[ValueNode]836[ValueNode]836"Constant*
|
|
7
|
+
value*strides�
|
|
8
|
+
<[ValueNode]837[ValueNode]837"Constant*
|
|
9
|
+
value*pads�
|
|
10
|
+
A[ValueNode]838[ValueNode]838"Constant*
|
|
11
|
+
value* ceil_mode�
|
|
12
|
+
I[ValueNode]839[ValueNode]839"Constant*
|
|
13
|
+
value*count_include_pad�
|
|
14
|
+
H[ValueNode]840[ValueNode]840"Constant*
|
|
15
|
+
value*divisor_override�
|
|
16
|
+
C[ValueNode]841[ValueNode]841"Constant*
|
|
17
|
+
value*data_format�
|
|
18
|
+
@[ValueNode]842[ValueNode]842"Constant*
|
|
19
|
+
value*pad_mode�
|
|
20
|
+
�
|
|
21
|
+
[ValueNode]835
|
|
22
|
+
[ValueNode]836
|
|
23
|
+
[ValueNode]837
|
|
24
|
+
[ValueNode]838
|
|
25
|
+
[ValueNode]839
|
|
26
|
+
[ValueNode]840
|
|
27
|
+
[ValueNode]841
|
|
28
|
+
[ValueNode]842,get_bprop_avg_pool_3d_grad.822:[CNode]843:10,get_bprop_avg_pool_3d_grad.822:[CNode]843:10"REF::S-Prim-MakeTuple:11:Default/S-Prim-MakeTuple-op634
|
|
29
|
+
C[ValueNode]844[ValueNode]844"Constant*
|
|
30
|
+
value*kernel_size�
|
|
31
|
+
�
|
|
32
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
33
|
+
[ValueNode]844,get_bprop_avg_pool_3d_grad.822:[CNode]845:12,get_bprop_avg_pool_3d_grad.822:[CNode]845:12"REF::getattr:13:Default/getattr-op635
|
|
34
|
+
?[ValueNode]846[ValueNode]846"Constant*
|
|
35
|
+
value*strides�
|
|
36
|
+
�
|
|
37
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
38
|
+
[ValueNode]846,get_bprop_avg_pool_3d_grad.822:[CNode]847:14,get_bprop_avg_pool_3d_grad.822:[CNode]847:14"REF::getattr:13:Default/getattr-op636
|
|
39
|
+
E[ValueNode]848[ValueNode]848"Constant*
|
|
40
|
+
value*
|
|
41
|
+
�
|
|
42
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
43
|
+
[ValueNode]848,get_bprop_avg_pool_3d_grad.822:[CNode]849:15,get_bprop_avg_pool_3d_grad.822:[CNode]849:15"REF::getattr:13:Default/getattr-op637
|
|
44
|
+
�,get_bprop_avg_pool_3d_grad.822:[CNode]850:16,get_bprop_avg_pool_3d_grad.822:[CNode]850:16"1REF::get_bprop_avg_pool_3d_grad.822:[CNode]849:15:638
|
|
45
|
+
@[ValueNode]851[ValueNode]851"Constant*
|
|
46
|
+
value*pad_list�
|
|
47
|
+
�
|
|
48
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]850:16
|
|
49
|
+
[ValueNode]851*get_bprop_avg_pool_3d_grad.822:pad_list:17*get_bprop_avg_pool_3d_grad.822:pad_list:17"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op639
|
|
50
|
+
A[ValueNode]852[ValueNode]852"Constant*
|
|
51
|
+
value* ceil_mode�
|
|
52
|
+
�
|
|
53
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
54
|
+
[ValueNode]852,get_bprop_avg_pool_3d_grad.822:[CNode]853:18,get_bprop_avg_pool_3d_grad.822:[CNode]853:18"REF::getattr:13:Default/getattr-op640
|
|
55
|
+
E[ValueNode]854[ValueNode]854"Constant*
|
|
56
|
+
value*
|
|
57
|
+
�
|
|
58
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
59
|
+
[ValueNode]854,get_bprop_avg_pool_3d_grad.822:[CNode]855:19,get_bprop_avg_pool_3d_grad.822:[CNode]855:19"REF::getattr:13:Default/getattr-op641
|
|
60
|
+
�,get_bprop_avg_pool_3d_grad.822:[CNode]856:20,get_bprop_avg_pool_3d_grad.822:[CNode]856:20"1REF::get_bprop_avg_pool_3d_grad.822:[CNode]855:19:642
|
|
61
|
+
I[ValueNode]857[ValueNode]857"Constant*
|
|
62
|
+
value*count_include_pad�
|
|
63
|
+
�
|
|
64
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]856:20
|
|
65
|
+
[ValueNode]8573get_bprop_avg_pool_3d_grad.822:count_include_pad:213get_bprop_avg_pool_3d_grad.822:count_include_pad:21"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op643
|
|
66
|
+
H[ValueNode]858[ValueNode]858"Constant*
|
|
67
|
+
value*divisor_override�
|
|
68
|
+
�
|
|
69
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
70
|
+
[ValueNode]858,get_bprop_avg_pool_3d_grad.822:[CNode]859:22,get_bprop_avg_pool_3d_grad.822:[CNode]859:22"REF::getattr:13:Default/getattr-op644
|
|
71
|
+
C[ValueNode]860[ValueNode]860"Constant*
|
|
72
|
+
value*data_format�
|
|
73
|
+
�
|
|
74
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
75
|
+
[ValueNode]860,get_bprop_avg_pool_3d_grad.822:[CNode]861:23,get_bprop_avg_pool_3d_grad.822:[CNode]861:23"REF::getattr:13:Default/getattr-op645
|
|
76
|
+
@[ValueNode]862[ValueNode]862"Constant*
|
|
77
|
+
value*pad_mode�
|
|
78
|
+
�
|
|
79
|
+
#get_bprop_avg_pool_3d_grad.822:self
|
|
80
|
+
[ValueNode]862,get_bprop_avg_pool_3d_grad.822:[CNode]863:24,get_bprop_avg_pool_3d_grad.822:[CNode]863:24"REF::getattr:13:Default/getattr-op646
|
|
81
|
+
�
|
|
82
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]845:12
|
|
83
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]847:14
|
|
84
|
+
*get_bprop_avg_pool_3d_grad.822:pad_list:17
|
|
85
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]853:18
|
|
86
|
+
3get_bprop_avg_pool_3d_grad.822:count_include_pad:21
|
|
87
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]859:22
|
|
88
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]861:23
|
|
89
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]863:24,get_bprop_avg_pool_3d_grad.822:[CNode]864:25,get_bprop_avg_pool_3d_grad.822:[CNode]864:25"REF::S-Prim-MakeTuple:26:Default/S-Prim-MakeTuple-op647
|
|
90
|
+
�
|
|
91
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]843:10
|
|
92
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]864:25,get_bprop_avg_pool_3d_grad.822:[CNode]865:27,get_bprop_avg_pool_3d_grad.822:[CNode]865:27"REF::S-Prim-make_dict:28:Default/S-Prim-make_dict-op648
|
|
93
|
+
R866866"Constant*<
|
|
94
|
+
value*0mindspore.ops.operations._grad_ops.AvgPool3DGrad�!
|
|
95
|
+
�
|
|
96
|
+
866
|
|
97
|
+
,get_bprop_avg_pool_3d_grad.822:[CNode]865:272get_bprop_avg_pool_3d_grad.822:фavgpool3d_grad:292get_bprop_avg_pool_3d_grad.822:фavgpool3d_grad:29"REF::MetaFuncGraph::unpack_call:Default/unpack_call.867-op649get_bprop_avg_pool_3d_grad.822*%
|
|
98
|
+
#get_bprop_avg_pool_3d_grad.822:self*"
|
|
99
|
+
get_bprop_avg_pool_3d_grad.822:x*$
|
|
100
|
+
"get_bprop_avg_pool_3d_grad.822:out*'
|
|
101
|
+
%get_bprop_avg_pool_3d_grad.822:фdout2-
|
|
102
|
+
+get_bprop_avg_pool_3d_grad.822:[CNode]823:1:@bf8b8248fd9c9db80b0d31d687666512880861acf459f9165036db6d5b9bb2a0J/grad_nn_ops.pyB�
|
|
103
|
+
u
|
|
104
|
+
get_bprop_avg_pool_3d_grad.822:xbprop.824:x_shape:2bprop.824:x_shape:2"REF::shape.825:Default/shape.825-op628
|
|
105
|
+
�
|
|
106
|
+
bprop.824:x_shape:2bprop.824:[CNode]826:3bprop.824:[CNode]826:3"REF::S-Prim-IsShapeUnKnown:4:#Default/S-Prim-IsShapeUnKnown-op629
|
|
107
|
+
q
|
|
108
|
+
bprop.824:[CNode]826:3bprop.824:[CNode]827:5bprop.824:[CNode]827:5"REF::bool_.828:Default/bool_.828-op630
|
|
109
|
+
�
|
|
110
|
+
bprop.824:[CNode]827:5
|
|
111
|
+
✓bprop.829
|
|
112
|
+
✗bprop.830bprop.824:[CNode]831:6bprop.824:[CNode]831:6"
|
|
113
|
+
Rbprop.824:[CNode]832:8bprop.824:[CNode]832:8"REF::bprop.824:[CNode]831:6:632
|
|
114
|
+
w
|
|
115
|
+
bprop.824:[CNode]832:8bprop.824:[CNode]833:9bprop.824:[CNode]833:9"REF::↓bprop.834:Default/↓bprop.834-op633 bprop.8242
|
|
116
|
+
bprop.824:[CNode]833:9B�
|
|
117
|
+
�
|
|
118
|
+
↓bprop.834:фx_shape
|
|
119
|
+
%get_bprop_avg_pool_3d_grad.822:фdout↓bprop.834:dx:30↓bprop.834:dx:30"7REF::get_bprop_avg_pool_3d_grad.822:фavgpool3d_grad:29:650
|
|
120
|
+
�
|
|
121
|
+
↓bprop.834:dx:30↓bprop.834:[CNode]868:31↓bprop.834:[CNode]868:31"REF::S-Prim-MakeTuple:32:Default/S-Prim-MakeTuple-op651↓bprop.834*
|
|
122
|
+
↓bprop.834:фx_shape2
|
|
123
|
+
↓bprop.834:[CNode]868:31B
|
|
124
|
+
after_block� B%✗bprop.8302
|
|
125
|
+
bprop.824:x_shape:2B�
|
|
126
|
+
�✓bprop.829:[CNode]869:33✓bprop.829:[CNode]869:33">REF::ClassType::mindspore.ops.operations.array_ops.TensorShape:TDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.TensorShape'-op652
|
|
127
|
+
z
|
|
128
|
+
get_bprop_avg_pool_3d_grad.822:x✓bprop.829:x_shape:34✓bprop.829:x_shape:34"REF::✓bprop.829:[CNode]869:33:653✓bprop.8292
|
|
129
|
+
✓bprop.829:x_shape:34B�
|
|
130
|
+
@[ValueNode]870[ValueNode]870"Constant*
|
|
131
|
+
value*__bool__�
|
|
132
|
+
w
|
|
133
|
+
bool_.828:x
|
|
134
|
+
[ValueNode]870bool_.828:[CNode]871:35bool_.828:[CNode]871:35"REF::getattr:13:Default/getattr-op654
|
|
135
|
+
Ubool_.828:[CNode]872:36bool_.828:[CNode]872:36"REF::bool_.828:[CNode]871:35:655 bool_.828*
|
|
136
|
+
bool_.828:x2
|
|
137
|
+
bool_.828:[CNode]872:36B�
|
|
138
|
+
w
|
|
139
|
+
shape.825:input_xshape.825:[CNode]873:37shape.825:[CNode]873:37"REF::S-Prim-Shape:38:Default/S-Prim-Shape-op656 shape.825*
|
|
140
|
+
shape.825:input_x2
|
|
141
|
+
shape.825:[CNode]873:37Pb
|
|
142
|
+
Switch:7Switchb
|
|
143
|
+
|
|
144
|
+
getattr:13getattrb
|
|
145
|
+
S-Prim-Shape:38S-Prim-Shapeb'
|
|
146
|
+
S-Prim-MakeTuple:26S-Prim-MakeTupleb'
|
|
147
|
+
S-Prim-MakeTuple:32S-Prim-MakeTupleb0
|
|
148
|
+
S-Prim-IsShapeUnKnown:4S-Prim-IsShapeUnKnownb'
|
|
149
|
+
S-Prim-MakeTuple:11S-Prim-MakeTupleb'
|
|
150
|
+
S-Prim-make_dict:28S-Prim-make_dicth
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
s%get_bprop_avg_pool_grad.9:[CNode]16:1%get_bprop_avg_pool_grad.9:[CNode]16:1"
|
|
4
|
+
A
|
|
5
|
+
value*kernel_size�
|
|
6
|
+
=
|
|
7
|
+
value*strides�
|
|
8
|
+
>
|
|
9
|
+
value*pad_mode�
|
|
10
|
+
A
|
|
11
|
+
value*data_format�
|
|
12
|
+
�
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
A
|
|
18
|
+
value*kernel_size�
|
|
19
|
+
�
|
|
20
|
+
get_bprop_avg_pool_grad.9:self
|
|
21
|
+
|
|
22
|
+
=
|
|
23
|
+
value*strides�
|
|
24
|
+
�
|
|
25
|
+
get_bprop_avg_pool_grad.9:self
|
|
26
|
+
|
|
27
|
+
>
|
|
28
|
+
value*pad_mode�
|
|
29
|
+
�
|
|
30
|
+
get_bprop_avg_pool_grad.9:self
|
|
31
|
+
|
|
32
|
+
<
|
|
33
|
+
value*format�
|
|
34
|
+
�
|
|
35
|
+
get_bprop_avg_pool_grad.9:self
|
|
36
|
+
|
|
37
|
+
�
|
|
38
|
+
$get_bprop_avg_pool_grad.9:[CNode]2:4
|
|
39
|
+
$get_bprop_avg_pool_grad.9:[CNode]3:6
|
|
40
|
+
$get_bprop_avg_pool_grad.9:[CNode]4:7
|
|
41
|
+
$get_bprop_avg_pool_grad.9:[CNode]5:8$get_bprop_avg_pool_grad.9:[CNode]6:9$get_bprop_avg_pool_grad.9:[CNode]6:9"REF::S-Prim-MakeTuple:10:Default/S-Prim-MakeTuple-op6
|
|
42
|
+
�
|
|
43
|
+
$get_bprop_avg_pool_grad.9:[CNode]1:2
|
|
44
|
+
$get_bprop_avg_pool_grad.9:[CNode]6:9%get_bprop_avg_pool_grad.9:[CNode]7:11%get_bprop_avg_pool_grad.9:[CNode]7:11"REF::S-Prim-make_dict:12:Default/S-Prim-make_dict-op7
|
|
45
|
+
N2525"Constant*:
|
|
46
|
+
value*.mindspore.ops.operations._grad_ops.AvgPoolGrad�!
|
|
47
|
+
�
|
|
48
|
+
25
|
|
49
|
+
%get_bprop_avg_pool_grad.9:[CNode]7:11)get_bprop_avg_pool_grad.9:avgpool_grad:13)get_bprop_avg_pool_grad.9:avgpool_grad:13"REF::MetaFuncGraph::unpack_call:Default/unpack_call.8-op8get_bprop_avg_pool_grad.9*
|
|
50
|
+
get_bprop_avg_pool_grad.9:self*
|
|
51
|
+
get_bprop_avg_pool_grad.9:x*
|
|
52
|
+
get_bprop_avg_pool_grad.9:out*
|
|
53
|
+
get_bprop_avg_pool_grad.9:dout2'
|
|
54
|
+
%get_bprop_avg_pool_grad.9:[CNode]16:1:@4a74a0ba73d5078e776e116148e7145421a39dc8a84d9ba3d7c495f451ee6680J/grad_nn_ops.pyB�
|
|
55
|
+
�
|
|
56
|
+
get_bprop_avg_pool_grad.9:x
|
|
57
|
+
get_bprop_avg_pool_grad.9:out
|
|
58
|
+
get_bprop_avg_pool_grad.9:doutbprop.11:dx:14bprop.11:dx:14".REF::get_bprop_avg_pool_grad.9:avgpool_grad:13:9
|
|
59
|
+
w
|
|
60
|
+
bprop.11:dx:14bprop.11:[CNode]10:15bprop.11:[CNode]10:15"REF::S-Prim-MakeTuple:16:Default/S-Prim-MakeTuple-op10bprop.112
|
|
61
|
+
bprop.11:[CNode]10:15Pb
|
|
62
|
+
getattr:5getattrb'
|
|
63
|
+
S-Prim-make_dict:12S-Prim-make_dictb'
|
|
64
|
+
S-Prim-MakeTuple:16S-Prim-MakeTupleb&
|
|
65
|
+
S-Prim-MakeTuple:3S-Prim-MakeTupleb'
|
|
66
|
+
S-Prim-MakeTuple:10S-Prim-MakeTupleh
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
,get_bprop_binary_cross_entropy.2:[CNode]17:1,get_bprop_binary_cross_entropy.2:[CNode]17:1"REF::bprop.6:Default/bprop.6-op0
|
|
4
|
+
?
|
|
5
|
+
value* reduction�
|
|
6
|
+
�
|
|
7
|
+
%get_bprop_binary_cross_entropy.2:self
|
|
8
|
+
|
|
9
|
+
�
|
|
10
|
+
+get_bprop_binary_cross_entropy.2:[CNode]1:2'get_bprop_binary_cross_entropy.2:grad:4'get_bprop_binary_cross_entropy.2:grad:4"IREF::ClassType::mindspore.ops.operations._grad_ops.BinaryCrossEntropyGrad:]Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.BinaryCrossEntropyGrad'-op2 get_bprop_binary_cross_entropy.2*'
|
|
11
|
+
%get_bprop_binary_cross_entropy.2:self*$
|
|
12
|
+
"get_bprop_binary_cross_entropy.2:x*$
|
|
13
|
+
"get_bprop_binary_cross_entropy.2:y*)
|
|
14
|
+
'get_bprop_binary_cross_entropy.2:weight*&
|
|
15
|
+
$get_bprop_binary_cross_entropy.2:out*'
|
|
16
|
+
%get_bprop_binary_cross_entropy.2:dout2.
|
|
17
|
+
,get_bprop_binary_cross_entropy.2:[CNode]17:1:@ae043af2d528e56aa13b09b1bb8b1799c25e61a31aed38bbb97885ccd3c4ccf4J/grad_nn_ops.pyB�
|
|
18
|
+
�
|
|
19
|
+
"get_bprop_binary_cross_entropy.2:x
|
|
20
|
+
"get_bprop_binary_cross_entropy.2:y
|
|
21
|
+
%get_bprop_binary_cross_entropy.2:dout
|
|
22
|
+
'get_bprop_binary_cross_entropy.2:weightbprop.6:dx:5bprop.6:dx:5",REF::get_bprop_binary_cross_entropy.2:grad:4:3
|
|
23
|
+
�
|
|
24
|
+
"get_bprop_binary_cross_entropy.2:ybprop.6:[CNode]3:6bprop.6:[CNode]3:6".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op4
|
|
25
|
+
�
|
|
26
|
+
'get_bprop_binary_cross_entropy.2:weightbprop.6:[CNode]4:7bprop.6:[CNode]4:7".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op5
|
|
27
|
+
�
|
|
28
|
+
bprop.6:dx:5
|
|
29
|
+
bprop.6:[CNode]3:6
|
|
30
|
+
bprop.6:[CNode]4:7bprop.6:[CNode]5:8bprop.6:[CNode]5:8"REF::S-Prim-MakeTuple:9:Default/S-Prim-MakeTuple-op6bprop.62
|
|
31
|
+
bprop.6:[CNode]5:8Pb
|
|
32
|
+
getattr:3getattrb&
|
|
33
|
+
S-Prim-MakeTuple:9S-Prim-MakeTupleh
|