mindspore 2.0.0a0__cp37-none-any.whl → 2.0.0rc1__cp37-none-any.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/Third_Party_Open_Source_Software_Notice +9064 -0
- mindspore/__init__.py +4 -2
- mindspore/_akg/akg/composite/build_module.py +11 -0
- mindspore/_akg/akg/config/repository_cuda.json +11 -0
- mindspore/_akg/akg/tvm/contrib/nvcc.py +4 -3
- mindspore/_c_dataengine.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-37m-aarch64-linux-gnu.so +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/_mindspore_offline_debug.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/amp.py +52 -57
- mindspore/bin/cache_admin +0 -0
- mindspore/bin/cache_server +0 -0
- 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/config/super_bar_config.json +512 -0
- 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/include/mindapi/base/type_id.h +1 -0
- mindspore/lib/libdnnl.so.2 +0 -0
- mindspore/lib/libicudata.so.69 +0 -0
- mindspore/lib/libicui18n.so.69 +0 -0
- mindspore/lib/libicuuc.so.69 +0 -0
- mindspore/lib/libmindspore.so +0 -0
- mindspore/lib/libmindspore_backend.so +0 -0
- mindspore/lib/libmindspore_common.so +0 -0
- mindspore/lib/libmindspore_core.so +0 -0
- mindspore/lib/libmindspore_glog.so.0 +0 -0
- mindspore/lib/libmindspore_gpr.so.15 +0 -0
- mindspore/lib/libmindspore_grpc++.so.1 +0 -0
- mindspore/lib/libmindspore_grpc.so.15 +0 -0
- mindspore/lib/libmindspore_shared_lib.so +0 -0
- mindspore/lib/libmpi_adapter.so +0 -0
- mindspore/lib/libmpi_collective.so +0 -0
- mindspore/lib/libnnacl.so +0 -0
- mindspore/lib/libopencv_core.so.4.5 +0 -0
- mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
- mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
- mindspore/lib/libps_cache.so +0 -0
- mindspore/lib/plugin/ascend/libakg.so +0 -0
- mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
- mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
- mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_aicpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
- mindspore/lib/plugin/cpu/libakg.so +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.1 +0 -0
- mindspore/lib/plugin/{libmindspore_ascend.so → libmindspore_ascend.so.2} +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/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/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/scipy/linalg.py +10 -114
- mindspore/scipy/ops.py +2 -2
- mindspore/scipy/ops_wrapper.py +1 -1
- mindspore/scipy/optimize/_bfgs.py +1 -1
- mindspore/scipy/optimize/_lagrange.py +200 -0
- mindspore/scipy/optimize/line_search.py +3 -2
- mindspore/scipy/optimize/minimize.py +41 -2
- mindspore/scipy/sparse/__init__.py +2 -2
- mindspore/scipy/sparse/linalg.py +4 -464
- mindspore/scipy/utils.py +1 -1
- mindspore/scipy/utils_const.py +7 -1
- 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/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 +648 -574
- 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
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
s&get_bprop_smooth_l1_loss.3:[CNode]14:1&get_bprop_smooth_l1_loss.3:[CNode]14:1"REF::bprop.5:Default/bprop.5-op0
|
|
4
|
+
:
|
|
5
|
+
value*beta�
|
|
6
|
+
�
|
|
7
|
+
get_bprop_smooth_l1_loss.3:self
|
|
8
|
+
|
|
9
|
+
?
|
|
10
|
+
value* reduction�
|
|
11
|
+
�
|
|
12
|
+
get_bprop_smooth_l1_loss.3:self
|
|
13
|
+
|
|
14
|
+
�
|
|
15
|
+
%get_bprop_smooth_l1_loss.3:[CNode]1:2
|
|
16
|
+
%get_bprop_smooth_l1_loss.3:[CNode]2:4!get_bprop_smooth_l1_loss.3:grad:5!get_bprop_smooth_l1_loss.3:grad:5"CREF::ClassType::mindspore.ops.operations._grad_ops.SmoothL1LossGrad:WDefault/MindIRClassType:class 'mindspore.ops.operations._grad_ops.SmoothL1LossGrad'-op3get_bprop_smooth_l1_loss.3*!
|
|
17
|
+
get_bprop_smooth_l1_loss.3:self*'
|
|
18
|
+
%get_bprop_smooth_l1_loss.3:prediction*#
|
|
19
|
+
!get_bprop_smooth_l1_loss.3:target*
|
|
20
|
+
get_bprop_smooth_l1_loss.3:out*!
|
|
21
|
+
get_bprop_smooth_l1_loss.3:dout2(
|
|
22
|
+
&get_bprop_smooth_l1_loss.3:[CNode]14:1:@314af1f905ccdbb7b74529580bda29b0bf0da9ff41f53edb382ab764cf470067J/grad_nn_ops.pyB�
|
|
23
|
+
�
|
|
24
|
+
%get_bprop_smooth_l1_loss.3:prediction
|
|
25
|
+
!get_bprop_smooth_l1_loss.3:target
|
|
26
|
+
get_bprop_smooth_l1_loss.3:doutbprop.5:dx:6bprop.5:dx:6"&REF::get_bprop_smooth_l1_loss.3:grad:5:4
|
|
27
|
+
�
|
|
28
|
+
!get_bprop_smooth_l1_loss.3:target
|
|
29
|
+
%get_bprop_smooth_l1_loss.3:prediction
|
|
30
|
+
get_bprop_smooth_l1_loss.3:doutbprop.5:dy:7bprop.5:dy:7"&REF::get_bprop_smooth_l1_loss.3:grad:5:5
|
|
31
|
+
{
|
|
32
|
+
bprop.5:dx:6
|
|
33
|
+
bprop.5:dy:7bprop.5:[CNode]4:8bprop.5:[CNode]4:8"REF::S-Prim-MakeTuple:9:Default/S-Prim-MakeTuple-op6bprop.52
|
|
34
|
+
bprop.5:[CNode]4:8Pb
|
|
35
|
+
getattr:3getattrb&
|
|
36
|
+
S-Prim-MakeTuple:9S-Prim-MakeTupleh
|
|
Binary file
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
eget_bprop_softplus.1:[CNode]8:1get_bprop_softplus.1:[CNode]8:1"REF::bprop.3:Default/bprop.3-op0
|
|
4
|
+
�$get_bprop_softplus.1:softplus_grad:2$get_bprop_softplus.1:softplus_grad:2"?REF::ClassType::mindspore.ops.operations._grad_ops.SoftplusGrad:SDefault/MindIRClassType:class 'mindspore.ops.operations._grad_ops.SoftplusGrad'-op1get_bprop_softplus.1*
|
|
5
|
+
get_bprop_softplus.1:self*
|
|
6
|
+
get_bprop_softplus.1:x*
|
|
7
|
+
get_bprop_softplus.1:out*
|
|
8
|
+
get_bprop_softplus.1:dout2!
|
|
9
|
+
get_bprop_softplus.1:[CNode]8:1:@3a7cb85a47fbe925db20ce152e09968888e09f915fa9ab78b0f0a85185434ed3J/grad_nn_ops.pyB�
|
|
10
|
+
}
|
|
11
|
+
get_bprop_softplus.1:dout
|
|
12
|
+
get_bprop_softplus.1:xbprop.3:dx:3bprop.3:dx:3")REF::get_bprop_softplus.1:softplus_grad:2:2
|
|
13
|
+
m
|
|
14
|
+
bprop.3:dx:3bprop.3:[CNode]2:4bprop.3:[CNode]2:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op3bprop.32
|
|
15
|
+
bprop.3:[CNode]2:4Pb&
|
|
16
|
+
S-Prim-MakeTuple:5S-Prim-MakeTupleh
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
g get_bprop_softsign.2:[CNode]18:1 get_bprop_softsign.2:[CNode]18:1"REF::bprop.7:Default/bprop.7-op0
|
|
4
|
+
�get_bprop_softsign.2:mul:2get_bprop_softsign.2:mul:2"5REF::ClassType::mindspore.ops.operations.math_ops.Mul:IDefault/MindIRClassType:class 'mindspore.ops.operations.math_ops.Mul'-op1
|
|
5
|
+
�get_bprop_softsign.2:div:3get_bprop_softsign.2:div:3"5REF::ClassType::mindspore.ops.operations.math_ops.Div:IDefault/MindIRClassType:class 'mindspore.ops.operations.math_ops.Div'-op2
|
|
6
|
+
�get_bprop_softsign.2:square:4get_bprop_softsign.2:square:4"8REF::ClassType::mindspore.ops.operations.math_ops.Square:LDefault/MindIRClassType:class 'mindspore.ops.operations.math_ops.Square'-op3
|
|
7
|
+
�get_bprop_softsign.2:absolute:5get_bprop_softsign.2:absolute:5"5REF::ClassType::mindspore.ops.operations.math_ops.Abs:IDefault/MindIRClassType:class 'mindspore.ops.operations.math_ops.Abs'-op4get_bprop_softsign.2*
|
|
8
|
+
get_bprop_softsign.2:self*
|
|
9
|
+
get_bprop_softsign.2:x*
|
|
10
|
+
get_bprop_softsign.2:out*
|
|
11
|
+
get_bprop_softsign.2:dout2"
|
|
12
|
+
get_bprop_softsign.2:[CNode]18:1:@c01fde5841c04ea3da368f04a0559c317edc5a8328033ec7ca692b8c2ee0805aJ/grad_nn_ops.pyB�
|
|
13
|
+
i
|
|
14
|
+
get_bprop_softsign.2:xbprop.7:[CNode]1:6bprop.7:[CNode]1:6"$REF::get_bprop_softsign.2:absolute:5:5
|
|
15
|
+
6
|
|
16
|
+
value�
|
|
17
|
+
|
|
|
18
|
+
|
|
19
|
+
bprop.7:[CNode]1:6bprop.7:[CNode]3:7bprop.7:[CNode]3:7"REF::MetaFuncGraph::add:Default/S-Prim-add-op6
|
|
20
|
+
c
|
|
21
|
+
bprop.7:[CNode]3:7bprop.7:[CNode]4:8bprop.7:[CNode]4:8""REF::get_bprop_softsign.2:square:4:7
|
|
22
|
+
6
|
|
23
|
+
value�
|
|
24
|
+
o
|
|
25
|
+
|
|
26
|
+
bprop.7:[CNode]4:8bprop.7:[CNode]5:9bprop.7:[CNode]5:9"REF::get_bprop_softsign.2:div:3:8
|
|
27
|
+
q
|
|
28
|
+
get_bprop_softsign.2:dout
|
|
29
|
+
bprop.7:[CNode]5:9
|
|
30
|
+
r
|
|
31
|
+
|
|
32
|
+
bprop.7:[CNode]6:11Pb'
|
|
33
|
+
S-Prim-MakeTuple:12S-Prim-MakeTupleh
|
|
@@ -1,56 +1,54 @@
|
|
|
1
1
|
|
|
2
2
|
0.1.1 MindSpore*2.0.0:�
|
|
3
|
-
j get_bprop_squeeze.
|
|
4
|
-
get_bprop_squeeze.
|
|
5
|
-
get_bprop_squeeze.
|
|
6
|
-
get_bprop_squeeze.
|
|
7
|
-
get_bprop_squeeze.
|
|
8
|
-
get_bprop_squeeze.
|
|
3
|
+
j get_bprop_squeeze.37:[CNode]38:1 get_bprop_squeeze.37:[CNode]38:1"
|
|
4
|
+
get_bprop_squeeze.37:self*
|
|
5
|
+
get_bprop_squeeze.37:x*
|
|
6
|
+
get_bprop_squeeze.37:out*
|
|
7
|
+
get_bprop_squeeze.37:фdout2"
|
|
8
|
+
get_bprop_squeeze.37:[CNode]38:1:@5ea4080e1f189d05653ba505a6a9e0e4e9be103c3bcd5a867ca2f1aaaa82bb49J/grad_array_ops.pyB�
|
|
9
9
|
n
|
|
10
|
-
get_bprop_squeeze.
|
|
11
|
-
�
|
|
12
|
-
bprop.
|
|
10
|
+
get_bprop_squeeze.37:xbprop.39:shapex:2bprop.39:shapex:2"REF::S-Prim-Shape:3:Default/S-Prim-Shape-op29
|
|
11
|
+
�
|
|
12
|
+
bprop.39:shapex:2bprop.39:[CNode]40:4bprop.39:[CNode]40:4"REF::S-Prim-IsShapeUnKnown:5:"Default/S-Prim-IsShapeUnKnown-op30
|
|
13
13
|
h
|
|
14
|
-
bprop.
|
|
14
|
+
bprop.39:[CNode]40:4bprop.39:[CNode]41:6bprop.39:[CNode]41:6"
|
|
15
15
|
�
|
|
16
|
-
bprop.
|
|
17
|
-
✓bprop.
|
|
18
|
-
✗bprop.
|
|
19
|
-
Kbprop.
|
|
16
|
+
bprop.39:[CNode]41:6
|
|
17
|
+
✓bprop.43
|
|
18
|
+
✗bprop.44bprop.39:[CNode]45:7bprop.39:[CNode]45:7"
|
|
19
|
+
Kbprop.39:[CNode]46:9bprop.39:[CNode]46:9"REF::bprop.39:[CNode]45:7:33
|
|
20
20
|
p
|
|
21
|
-
bprop.
|
|
22
|
-
bprop.
|
|
23
|
-
�↓bprop.
|
|
21
|
+
bprop.39:[CNode]46:9bprop.39:[CNode]47:10bprop.39:[CNode]47:10"REF::↓bprop.48:Default/↓bprop.48-op34bprop.392
|
|
22
|
+
bprop.39:[CNode]47:10B�
|
|
23
|
+
�↓bprop.48:49:11↓bprop.48:49:11":REF::ClassType::mindspore.ops.operations.array_ops.Reshape:ODefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Reshape'-op35
|
|
24
24
|
u
|
|
25
|
-
get_bprop_squeeze.
|
|
26
|
-
↓bprop.
|
|
25
|
+
get_bprop_squeeze.37:фdout
|
|
26
|
+
↓bprop.48:фshapex↓bprop.48:49:12↓bprop.48:49:12"REF::↓bprop.48:49:11:36
|
|
27
27
|
�
|
|
28
|
-
↓bprop.
|
|
29
|
-
↓bprop.
|
|
30
|
-
↓bprop.
|
|
31
|
-
after_block� B"✗bprop.
|
|
32
|
-
bprop.
|
|
28
|
+
↓bprop.48:49:12↓bprop.48:[CNode]50:13↓bprop.48:[CNode]50:13"REF::S-Prim-MakeTuple:14:Default/S-Prim-MakeTuple-op37↓bprop.48*
|
|
29
|
+
↓bprop.48:фshapex2
|
|
30
|
+
↓bprop.48:[CNode]50:13B
|
|
31
|
+
after_block� B"✗bprop.442
|
|
32
|
+
bprop.39:shapex:2B�
|
|
33
33
|
�
|
|
34
|
-
get_bprop_squeeze.
|
|
35
|
-
✓bprop.
|
|
36
|
-
>
|
|
34
|
+
get_bprop_squeeze.37:x✓bprop.43:shapex:15✓bprop.43:shapex:15"REF::S-Prim-TensorShape:16:Default/S-Prim-TensorShape-op38✓bprop.432
|
|
35
|
+
✓bprop.43:shapex:15B�
|
|
36
|
+
>
|
|
37
37
|
value*__bool__�
|
|
38
38
|
p
|
|
39
39
|
|
|
40
|
-
bool_.
|
|
41
|
-
|
|
42
|
-
Nbool_.
|
|
40
|
+
bool_.42:x
|
|
41
|
+
|
|
42
|
+
Nbool_.42:[CNode]53:19bool_.42:[CNode]53:19"REF::bool_.42:[CNode]52:17:40bool_.42*
|
|
43
43
|
|
|
44
|
-
bool_.
|
|
45
|
-
bool_.
|
|
44
|
+
bool_.42:x2
|
|
45
|
+
bool_.42:[CNode]53:19Pb
|
|
46
46
|
Switch:8Switchb
|
|
47
47
|
|
|
48
|
-
getattr:18
|
|
49
|
-
S-Prim-Shape:3S-Prim-Shapeb�
|
|
50
|
-
S-Prim-is_shape_unknown:5S-Prim-is_shape_unknown1
|
|
51
|
-
constexpr_module*mindspore.ops._utils.utils�%
|
|
52
|
-
constexpr_name*is_shape_unknown�b'
|
|
53
|
-
S-Prim-MakeTuple:14S-Prim-MakeTuplebm
|
|
48
|
+
getattr:18getattrbm
|
|
54
49
|
S-Prim-TensorShape:16S-Prim-TensorShape
|
|
55
50
|
output_names��Zoutput�
|
|
56
|
-
input_names��Zinput_x�
|
|
51
|
+
input_names��Zinput_x�b'
|
|
52
|
+
S-Prim-MakeTuple:14S-Prim-MakeTupleb0
|
|
53
|
+
S-Prim-IsShapeUnKnown:5S-Prim-IsShapeUnKnownb
|
|
54
|
+
S-Prim-Shape:3S-Prim-Shapeh
|
|
@@ -1,100 +1,98 @@
|
|
|
1
1
|
|
|
2
|
-
0.1.1 MindSpore*2.0.0:�
|
|
3
|
-
}(get_bprop_strided_slice.
|
|
4
|
-
B[ValueNode]
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
}(get_bprop_strided_slice.370:[CNode]371:1(get_bprop_strided_slice.370:[CNode]371:1"REF::bprop.372:Default/bprop.372-op318
|
|
4
|
+
B[ValueNode]382[ValueNode]382"Constant*
|
|
5
5
|
value*
|
|
6
6
|
begin_mask�
|
|
7
7
|
�
|
|
8
|
-
get_bprop_strided_slice.
|
|
9
|
-
[ValueNode]
|
|
10
|
-
@[ValueNode]
|
|
8
|
+
get_bprop_strided_slice.370:self
|
|
9
|
+
[ValueNode]382)get_bprop_strided_slice.370:[CNode]383:11)get_bprop_strided_slice.370:[CNode]383:11"REF::getattr:12:Default/getattr-op325
|
|
10
|
+
@[ValueNode]384[ValueNode]384"Constant*
|
|
11
11
|
value*end_mask�
|
|
12
12
|
�
|
|
13
|
-
get_bprop_strided_slice.
|
|
14
|
-
[ValueNode]
|
|
15
|
-
E[ValueNode]
|
|
13
|
+
get_bprop_strided_slice.370:self
|
|
14
|
+
[ValueNode]384)get_bprop_strided_slice.370:[CNode]385:13)get_bprop_strided_slice.370:[CNode]385:13"REF::getattr:12:Default/getattr-op326
|
|
15
|
+
E[ValueNode]386[ValueNode]386"Constant*
|
|
16
16
|
value*
|
|
17
17
|
�
|
|
18
|
-
get_bprop_strided_slice.
|
|
19
|
-
[ValueNode]
|
|
20
|
-
E[ValueNode]
|
|
18
|
+
get_bprop_strided_slice.370:self
|
|
19
|
+
[ValueNode]386)get_bprop_strided_slice.370:[CNode]387:14)get_bprop_strided_slice.370:[CNode]387:14"REF::getattr:12:Default/getattr-op327
|
|
20
|
+
E[ValueNode]388[ValueNode]388"Constant*
|
|
21
21
|
value*
|
|
22
22
|
�
|
|
23
|
-
get_bprop_strided_slice.
|
|
24
|
-
[ValueNode]
|
|
25
|
-
H[ValueNode]
|
|
23
|
+
get_bprop_strided_slice.370:self
|
|
24
|
+
[ValueNode]388)get_bprop_strided_slice.370:[CNode]389:15)get_bprop_strided_slice.370:[CNode]389:15"REF::getattr:12:Default/getattr-op328
|
|
25
|
+
H[ValueNode]390[ValueNode]390"Constant*
|
|
26
26
|
value*shrink_axis_mask�
|
|
27
27
|
�
|
|
28
|
-
get_bprop_strided_slice.
|
|
29
|
-
[ValueNode]
|
|
30
|
-
�
|
|
31
|
-
)get_bprop_strided_slice.
|
|
32
|
-
)get_bprop_strided_slice.
|
|
33
|
-
)get_bprop_strided_slice.
|
|
34
|
-
)get_bprop_strided_slice.
|
|
35
|
-
)get_bprop_strided_slice.
|
|
36
|
-
get_bprop_strided_slice.
|
|
37
|
-
get_bprop_strided_slice.
|
|
38
|
-
#get_bprop_strided_slice.
|
|
39
|
-
!get_bprop_strided_slice.
|
|
40
|
-
%get_bprop_strided_slice.
|
|
41
|
-
get_bprop_strided_slice.
|
|
42
|
-
"get_bprop_strided_slice.
|
|
43
|
-
(get_bprop_strided_slice.
|
|
28
|
+
get_bprop_strided_slice.370:self
|
|
29
|
+
[ValueNode]390)get_bprop_strided_slice.370:[CNode]391:16)get_bprop_strided_slice.370:[CNode]391:16"REF::getattr:12:Default/getattr-op329
|
|
30
|
+
�
|
|
31
|
+
)get_bprop_strided_slice.370:[CNode]383:11
|
|
32
|
+
)get_bprop_strided_slice.370:[CNode]385:13
|
|
33
|
+
)get_bprop_strided_slice.370:[CNode]387:14
|
|
34
|
+
)get_bprop_strided_slice.370:[CNode]389:15
|
|
35
|
+
)get_bprop_strided_slice.370:[CNode]391:16+get_bprop_strided_slice.370:фinput_grad:17+get_bprop_strided_slice.370:фinput_grad:17"CREF::ClassType::mindspore.ops.operations._grad_ops.StridedSliceGrad:YDefault/MindIRClassType:class 'mindspore.ops.operations._grad_ops.StridedSliceGrad'-op330get_bprop_strided_slice.370*"
|
|
36
|
+
get_bprop_strided_slice.370:self*
|
|
37
|
+
get_bprop_strided_slice.370:x*%
|
|
38
|
+
#get_bprop_strided_slice.370:фbegin*#
|
|
39
|
+
!get_bprop_strided_slice.370:фend*'
|
|
40
|
+
%get_bprop_strided_slice.370:фstrides*!
|
|
41
|
+
get_bprop_strided_slice.370:out*$
|
|
42
|
+
"get_bprop_strided_slice.370:фdout2*
|
|
43
|
+
(get_bprop_strided_slice.370:[CNode]371:1:@10d9df60e7853a61b74b274af84fc8e52796d01df754c8b0b281600ac9347e97J/grad_array_ops.pyB�
|
|
44
44
|
z
|
|
45
|
-
get_bprop_strided_slice.
|
|
46
|
-
�
|
|
47
|
-
bprop.
|
|
45
|
+
get_bprop_strided_slice.370:xbprop.372:x_shape:2bprop.372:x_shape:2"REF::S-Prim-Shape:3:Default/S-Prim-Shape-op319
|
|
46
|
+
�
|
|
47
|
+
bprop.372:x_shape:2bprop.372:[CNode]373:4bprop.372:[CNode]373:4"REF::S-Prim-IsShapeUnKnown:5:#Default/S-Prim-IsShapeUnKnown-op320
|
|
48
48
|
q
|
|
49
|
-
bprop.
|
|
49
|
+
bprop.372:[CNode]373:4bprop.372:[CNode]374:6bprop.372:[CNode]374:6"REF::bool_.375:Default/bool_.375-op321
|
|
50
50
|
�
|
|
51
|
-
bprop.
|
|
52
|
-
✓bprop.
|
|
53
|
-
✗bprop.
|
|
54
|
-
Rbprop.
|
|
51
|
+
bprop.372:[CNode]374:6
|
|
52
|
+
✓bprop.376
|
|
53
|
+
✗bprop.377bprop.372:[CNode]378:7bprop.372:[CNode]378:7"
|
|
54
|
+
Rbprop.372:[CNode]379:9bprop.372:[CNode]379:9"REF::bprop.372:[CNode]378:7:323
|
|
55
55
|
y
|
|
56
|
-
bprop.
|
|
57
|
-
bprop.
|
|
56
|
+
bprop.372:[CNode]379:9bprop.372:[CNode]380:10bprop.372:[CNode]380:10"REF::↓bprop.381:Default/↓bprop.381-op324 bprop.3722
|
|
57
|
+
bprop.372:[CNode]380:10B�
|
|
58
58
|
�
|
|
59
|
-
"get_bprop_strided_slice.
|
|
60
|
-
↓bprop.
|
|
61
|
-
#get_bprop_strided_slice.
|
|
62
|
-
!get_bprop_strided_slice.
|
|
63
|
-
%get_bprop_strided_slice.
|
|
59
|
+
"get_bprop_strided_slice.370:фdout
|
|
60
|
+
↓bprop.381:фx_shape
|
|
61
|
+
#get_bprop_strided_slice.370:фbegin
|
|
62
|
+
!get_bprop_strided_slice.370:фend
|
|
63
|
+
%get_bprop_strided_slice.370:фstrides↓bprop.381:dx:18↓bprop.381:dx:18"0REF::get_bprop_strided_slice.370:фinput_grad:17:331
|
|
64
64
|
�
|
|
65
|
-
#get_bprop_strided_slice.
|
|
65
|
+
#get_bprop_strided_slice.370:фbegin↓bprop.381:[CNode]392:19↓bprop.381:[CNode]392:19".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:/Default/S-Prim-hyper_map[zeros_like_leaf]-op332
|
|
66
66
|
�
|
|
67
|
-
!get_bprop_strided_slice.
|
|
67
|
+
!get_bprop_strided_slice.370:фend↓bprop.381:[CNode]393:20↓bprop.381:[CNode]393:20".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:/Default/S-Prim-hyper_map[zeros_like_leaf]-op333
|
|
68
68
|
�
|
|
69
|
-
%get_bprop_strided_slice.
|
|
69
|
+
%get_bprop_strided_slice.370:фstrides↓bprop.381:[CNode]394:21↓bprop.381:[CNode]394:21".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:/Default/S-Prim-hyper_map[zeros_like_leaf]-op334
|
|
70
70
|
�
|
|
71
|
-
↓bprop.
|
|
72
|
-
↓bprop.
|
|
73
|
-
↓bprop.
|
|
74
|
-
↓bprop.
|
|
75
|
-
↓bprop.
|
|
76
|
-
↓bprop.
|
|
77
|
-
after_block� B%✗bprop.
|
|
78
|
-
bprop.
|
|
71
|
+
↓bprop.381:dx:18
|
|
72
|
+
↓bprop.381:[CNode]392:19
|
|
73
|
+
↓bprop.381:[CNode]393:20
|
|
74
|
+
↓bprop.381:[CNode]394:21↓bprop.381:[CNode]395:22↓bprop.381:[CNode]395:22"REF::S-Prim-MakeTuple:23:Default/S-Prim-MakeTuple-op335↓bprop.381*
|
|
75
|
+
↓bprop.381:фx_shape2
|
|
76
|
+
↓bprop.381:[CNode]395:22B
|
|
77
|
+
after_block� B%✗bprop.3772
|
|
78
|
+
bprop.372:x_shape:2B�
|
|
79
79
|
�
|
|
80
|
-
get_bprop_strided_slice.
|
|
81
|
-
✓bprop.
|
|
82
|
-
@[ValueNode]
|
|
80
|
+
get_bprop_strided_slice.370:x✓bprop.376:x_shape:24✓bprop.376:x_shape:24"REF::S-Prim-TensorShape:25: Default/S-Prim-TensorShape-op336✓bprop.3762
|
|
81
|
+
✓bprop.376:x_shape:24B�
|
|
82
|
+
@[ValueNode]396[ValueNode]396"Constant*
|
|
83
83
|
value*__bool__�
|
|
84
84
|
w
|
|
85
|
-
bool_.
|
|
86
|
-
[ValueNode]
|
|
87
|
-
Ubool_.
|
|
88
|
-
bool_.
|
|
89
|
-
bool_.
|
|
85
|
+
bool_.375:x
|
|
86
|
+
[ValueNode]396bool_.375:[CNode]397:26bool_.375:[CNode]397:26"REF::getattr:12:Default/getattr-op337
|
|
87
|
+
Ubool_.375:[CNode]398:27bool_.375:[CNode]398:27"REF::bool_.375:[CNode]397:26:338 bool_.375*
|
|
88
|
+
bool_.375:x2
|
|
89
|
+
bool_.375:[CNode]398:27Pb
|
|
90
90
|
Switch:8Switchb
|
|
91
91
|
|
|
92
|
-
getattr:12
|
|
92
|
+
getattr:12getattrb
|
|
93
|
+
S-Prim-Shape:3S-Prim-Shapeb'
|
|
94
|
+
S-Prim-MakeTuple:23S-Prim-MakeTuplebm
|
|
93
95
|
S-Prim-TensorShape:25S-Prim-TensorShape
|
|
94
96
|
output_names��Zoutput�
|
|
95
|
-
input_names��Zinput_x�
|
|
96
|
-
|
|
97
|
-
constexpr_module*mindspore.ops._utils.utils�%
|
|
98
|
-
constexpr_name*is_shape_unknown�b
|
|
99
|
-
S-Prim-Shape:3S-Prim-Shapeb'
|
|
100
|
-
S-Prim-MakeTuple:23S-Prim-MakeTupleh
|
|
97
|
+
input_names��Zinput_x�b0
|
|
98
|
+
S-Prim-IsShapeUnKnown:5S-Prim-IsShapeUnKnownh
|
|
Binary file
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
aget_bprop_tanh.9:[CNode]33:1get_bprop_tanh.9:[CNode]33:1"
|
|
4
|
+
�get_bprop_tanh.9:tanh_grad:14get_bprop_tanh.9:tanh_grad:14";REF::ClassType::mindspore.ops.operations._grad_ops.TanhGrad:ODefault/MindIRClassType:class 'mindspore.ops.operations._grad_ops.TanhGrad'-op9
|
|
5
|
+
�get_bprop_tanh.9:conj:16get_bprop_tanh.9:conj:16"6REF::ClassType::mindspore.ops.operations.math_ops.Conj:KDefault/MindIRClassType:class 'mindspore.ops.operations.math_ops.Conj'-op11get_bprop_tanh.9*
|
|
6
|
+
get_bprop_tanh.9:self*
|
|
7
|
+
get_bprop_tanh.9:x*
|
|
8
|
+
get_bprop_tanh.9:out*
|
|
9
|
+
get_bprop_tanh.9:dout2
|
|
10
|
+
get_bprop_tanh.9:[CNode]33:1:@9347c56cf78023e2150873f4c7aa5ee3cd0a267b5a83edb9f7899e9ec5b50988J/grad_nn_ops.pyB�
|
|
11
|
+
;
|
|
12
|
+
value*dtype�
|
|
13
|
+
r
|
|
14
|
+
get_bprop_tanh.9:x
|
|
15
|
+
|
|
16
|
+
@
|
|
17
|
+
valueb
|
|
18
|
+
:value0�
|
|
19
|
+
@
|
|
20
|
+
valueb
|
|
21
|
+
:value0�
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
�
|
|
26
|
+
bprop.10:[CNode]4:2
|
|
27
|
+
bprop.10:[CNode]5:4bprop.10:[CNode]6:6bprop.10:[CNode]6:6"REF::MetaFuncGraph::in:Default/S-Prim-in-op3
|
|
28
|
+
b
|
|
29
|
+
bprop.10:[CNode]6:6bprop.10:[CNode]7:7bprop.10:[CNode]7:7"REF::bool_.8:Default/bool_.8-op4
|
|
30
|
+
~
|
|
31
|
+
bprop.10:[CNode]7:7
|
|
32
|
+
✓bprop.12
|
|
33
|
+
✗bprop.13bprop.10:[CNode]11:8bprop.10:[CNode]11:8"
|
|
34
|
+
Lbprop.10:[CNode]14:10bprop.10:[CNode]14:10"REF::bprop.10:[CNode]11:8:6
|
|
35
|
+
p
|
|
36
|
+
bprop.10:[CNode]14:10bprop.10:[CNode]15:11bprop.10:[CNode]15:11"REF::↓bprop.16:Default/↓bprop.16-op7bprop.102
|
|
37
|
+
bprop.10:[CNode]15:11B�
|
|
38
|
+
|
|
|
39
|
+
↓bprop.16:фdx↓bprop.16:[CNode]1:12↓bprop.16:[CNode]1:12"REF::S-Prim-MakeTuple:13:Default/S-Prim-MakeTuple-op8↓bprop.16*
|
|
40
|
+
↓bprop.16:фdx2
|
|
41
|
+
↓bprop.16:[CNode]1:12B
|
|
42
|
+
after_block� B�
|
|
43
|
+
{
|
|
44
|
+
get_bprop_tanh.9:out
|
|
45
|
+
get_bprop_tanh.9:dout✗bprop.13:dx:15✗bprop.13:dx:15""REF::get_bprop_tanh.9:tanh_grad:14:10✗bprop.132
|
|
46
|
+
✗bprop.13:dx:15B�
|
|
47
|
+
d
|
|
48
|
+
get_bprop_tanh.9:dout✓bprop.12:dout:17✓bprop.12:dout:17"REF::get_bprop_tanh.9:conj:16:12
|
|
49
|
+
y
|
|
50
|
+
get_bprop_tanh.9:out
|
|
51
|
+
✓bprop.12:dout:17✓bprop.12:dx:18✓bprop.12:dx:18""REF::get_bprop_tanh.9:tanh_grad:14:13
|
|
52
|
+
\
|
|
53
|
+
✓bprop.12:dx:18✓bprop.12:dx:19✓bprop.12:dx:19"REF::get_bprop_tanh.9:conj:16:14✓bprop.122
|
|
54
|
+
✓bprop.12:dx:19B�
|
|
55
|
+
>
|
|
56
|
+
value*__bool__�
|
|
57
|
+
j
|
|
58
|
+
bool_.8:x
|
|
59
|
+
|
|
60
|
+
Hbool_.8:[CNode]3:21bool_.8:[CNode]3:21"REF::bool_.8:[CNode]2:20:16bool_.8*
|
|
61
|
+
bool_.8:x2
|
|
62
|
+
bool_.8:[CNode]3:21Pb
|
|
63
|
+
Switch:9Switchb
|
|
64
|
+
getattr:3getattrb'
|
|
65
|
+
S-Prim-MakeTuple:13S-Prim-MakeTupleb&
|
|
66
|
+
S-Prim-MakeTuple:5S-Prim-MakeTupleh
|
|
Binary file
|
|
Binary file
|
|
@@ -1,20 +1,20 @@
|
|
|
1
1
|
|
|
2
|
-
0.1.1 MindSpore*2.0.0
|
|
3
|
-
�
|
|
4
|
-
|
|
5
|
-
�
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
�
|
|
10
|
-
|
|
11
|
-
�
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
�
|
|
4
|
+
bprop_tuple_getitem.1:data bprop_tuple_getitem.1:[CNode]2:1 bprop_tuple_getitem.1:[CNode]2:1".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op0
|
|
5
|
+
�
|
|
6
|
+
bprop_tuple_getitem.1:[CNode]2:1
|
|
7
|
+
bprop_tuple_getitem.1:idx
|
|
8
|
+
bprop_tuple_getitem.1:dout bprop_tuple_getitem.1:[CNode]3:2 bprop_tuple_getitem.1:[CNode]3:2"REF::S-Prim-tuple_setitem:3: Default/S-Prim-tuple_setitem-op1
|
|
9
|
+
�
|
|
10
|
+
bprop_tuple_getitem.1:idx bprop_tuple_getitem.1:[CNode]4:4 bprop_tuple_getitem.1:[CNode]4:4".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op2
|
|
11
|
+
�
|
|
12
|
+
bprop_tuple_getitem.1:[CNode]3:2
|
|
13
|
+
bprop_tuple_getitem.1:[CNode]4:4 bprop_tuple_getitem.1:[CNode]5:5 bprop_tuple_getitem.1:[CNode]5:5"REF::S-Prim-MakeTuple:6:Default/S-Prim-MakeTuple-op3bprop_tuple_getitem.1*
|
|
14
|
+
bprop_tuple_getitem.1:data*
|
|
15
|
+
bprop_tuple_getitem.1:idx*
|
|
16
|
+
bprop_tuple_getitem.1:out*
|
|
17
|
+
bprop_tuple_getitem.1:dout2"
|
|
18
|
+
bprop_tuple_getitem.1:[CNode]5:5:@badb8d2f379272a9bfe28f5e50b98c3de21620b357eb5d206a8abc1d48964179J/grad_implementations.pyPb.
|
|
19
19
|
S-Prim-tuple_setitem:3S-Prim-tuple_setitemb&
|
|
20
20
|
S-Prim-MakeTuple:6S-Prim-MakeTupleh
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
�0get_bprop_upsample_nearest_3d_grad.2:[CNode]13:10get_bprop_upsample_nearest_3d_grad.2:[CNode]13:1"REF::bprop.4:Default/bprop.4-op0
|
|
4
|
+
�0get_bprop_upsample_nearest_3d_grad.2:get_shape:20get_bprop_upsample_nearest_3d_grad.2:get_shape:2"8REF::ClassType::mindspore.ops.operations.array_ops.Shape:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Shape'-op1
|
|
5
|
+
A
|
|
6
|
+
value*output_size�
|
|
7
|
+
�
|
|
8
|
+
)get_bprop_upsample_nearest_3d_grad.2:self
|
|
9
|
+
|
|
10
|
+
<
|
|
11
|
+
value*scales�
|
|
12
|
+
�
|
|
13
|
+
)get_bprop_upsample_nearest_3d_grad.2:self
|
|
14
|
+
|
|
15
|
+
)get_bprop_upsample_nearest_3d_grad.2:self*.
|
|
16
|
+
,get_bprop_upsample_nearest_3d_grad.2:input_x**
|
|
17
|
+
(get_bprop_upsample_nearest_3d_grad.2:out*+
|
|
18
|
+
)get_bprop_upsample_nearest_3d_grad.2:dout22
|
|
19
|
+
0get_bprop_upsample_nearest_3d_grad.2:[CNode]13:1:@89f303661abc4ca7469d2cbbfdc77c210c2a3053877fbd2b8ab19c9b86f6f56aJ/grad_nn_ops.pyB�
|
|
20
|
+
�
|
|
21
|
+
,get_bprop_upsample_nearest_3d_grad.2:input_xbprop.4:[CNode]1:3bprop.4:[CNode]1:3"5REF::get_bprop_upsample_nearest_3d_grad.2:get_shape:2:2
|
|
22
|
+
�
|
|
23
|
+
bprop.4:[CNode]1:3
|
|
24
|
+
2get_bprop_upsample_nearest_3d_grad.2:output_size:4
|
|
25
|
+
-get_bprop_upsample_nearest_3d_grad.2:scales:6bprop.4:input_grad:7bprop.4:input_grad:7"HREF::ClassType::mindspore.ops.operations._grad_ops.UpsampleNearest3DGrad:\Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.UpsampleNearest3DGrad'-op5
|
|
26
|
+
e
|
|
27
|
+
)get_bprop_upsample_nearest_3d_grad.2:doutbprop.4:dx:8bprop.4:dx:8"REF::bprop.4:input_grad:7:6
|
|
28
|
+
n
|
|
29
|
+
bprop.4:dx:8bprop.4:[CNode]3:9bprop.4:[CNode]3:9"REF::S-Prim-MakeTuple:10:Default/S-Prim-MakeTuple-op7bprop.42
|
|
30
|
+
bprop.4:[CNode]3:9Pb
|
|
31
|
+
getattr:5getattrb'
|
|
32
|
+
S-Prim-MakeTuple:10S-Prim-MakeTupleh
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
|
|
2
|
+
0.1.1 MindSpore*2.0.0:�
|
|
3
|
+
�2get_bprop_upsample_trilinear_3d_grad.2:[CNode]13:12get_bprop_upsample_trilinear_3d_grad.2:[CNode]13:1"REF::bprop.4:Default/bprop.4-op0
|
|
4
|
+
�2get_bprop_upsample_trilinear_3d_grad.2:get_shape:22get_bprop_upsample_trilinear_3d_grad.2:get_shape:2"8REF::ClassType::mindspore.ops.operations.array_ops.Shape:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Shape'-op1
|
|
5
|
+
A
|
|
6
|
+
value*output_size�
|
|
7
|
+
�
|
|
8
|
+
+get_bprop_upsample_trilinear_3d_grad.2:self
|
|
9
|
+
|
|
10
|
+
<
|
|
11
|
+
value*scales�
|
|
12
|
+
�
|
|
13
|
+
+get_bprop_upsample_trilinear_3d_grad.2:self
|
|
14
|
+
|
|
15
|
+
C
|
|
16
|
+
value*
|
|
17
|
+
�
|
|
18
|
+
+get_bprop_upsample_trilinear_3d_grad.2:self
|
|
19
|
+
|
|
20
|
+
+get_bprop_upsample_trilinear_3d_grad.2:self*0
|
|
21
|
+
.get_bprop_upsample_trilinear_3d_grad.2:input_x*,
|
|
22
|
+
*get_bprop_upsample_trilinear_3d_grad.2:out*-
|
|
23
|
+
+get_bprop_upsample_trilinear_3d_grad.2:dout24
|
|
24
|
+
2get_bprop_upsample_trilinear_3d_grad.2:[CNode]13:1:@de557b6ab60e23e396148159c8aa5c5a3a0a77cf25b1b19f25b6e067c283d1e4J/grad_nn_ops.pyB�
|
|
25
|
+
�
|
|
26
|
+
.get_bprop_upsample_trilinear_3d_grad.2:input_xbprop.4:[CNode]1:3bprop.4:[CNode]1:3"7REF::get_bprop_upsample_trilinear_3d_grad.2:get_shape:2:2
|
|
27
|
+
�
|
|
28
|
+
bprop.4:[CNode]1:3
|
|
29
|
+
4get_bprop_upsample_trilinear_3d_grad.2:output_size:4
|
|
30
|
+
/get_bprop_upsample_trilinear_3d_grad.2:scales:6
|
|
31
|
+
6get_bprop_upsample_trilinear_3d_grad.2:align_corners:7bprop.4:input_grad:8bprop.4:input_grad:8"JREF::ClassType::mindspore.ops.operations._grad_ops.UpsampleTrilinear3DGrad:^Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.UpsampleTrilinear3DGrad'-op6
|
|
32
|
+
g
|
|
33
|
+
+get_bprop_upsample_trilinear_3d_grad.2:doutbprop.4:dx:9bprop.4:dx:9"REF::bprop.4:input_grad:8:7
|
|
34
|
+
p
|
|
35
|
+
bprop.4:dx:9bprop.4:[CNode]3:10bprop.4:[CNode]3:10"REF::S-Prim-MakeTuple:11:Default/S-Prim-MakeTuple-op8bprop.42
|
|
36
|
+
bprop.4:[CNode]3:10Pb
|
|
37
|
+
getattr:5getattrb'
|
|
38
|
+
S-Prim-MakeTuple:11S-Prim-MakeTupleh
|
|
@@ -28,6 +28,7 @@ os.environ['MS_DEV_EXPORT_BPROP_MINDIR'] = '1'
|
|
|
28
28
|
|
|
29
29
|
|
|
30
30
|
def run_generate(bprop_mindir_install_dir, bprop_map, force_update):
|
|
31
|
+
"""Run the generate process of bprop mindir."""
|
|
31
32
|
for op_name in bprop_map.keys():
|
|
32
33
|
if not isinstance(op_name, str):
|
|
33
34
|
continue
|
|
@@ -36,6 +37,7 @@ def run_generate(bprop_mindir_install_dir, bprop_map, force_update):
|
|
|
36
37
|
|
|
37
38
|
|
|
38
39
|
def run_generate_with_op_name(op_name, force_update):
|
|
40
|
+
"""Run the generate process of bprop mindir for single op."""
|
|
39
41
|
_export_bprop_mindir(op_name, force_update)
|
|
40
42
|
|
|
41
43
|
|
|
@@ -23,14 +23,14 @@ from __future__ import absolute_import
|
|
|
23
23
|
from mindspore.ops.composite.base import GradOperation, _Grad, HyperMap, Map, MultitypeFuncGraph, add_flags, \
|
|
24
24
|
tail, zip_operation, _Vmap, _TaylorOperation
|
|
25
25
|
from mindspore.ops.composite.env_ops import env_get
|
|
26
|
-
from mindspore.ops.
|
|
26
|
+
from mindspore.ops.function.clip_func import clip_by_global_norm
|
|
27
27
|
from mindspore.ops.composite.multitype_ops.add_impl import hyper_add
|
|
28
28
|
from mindspore.ops.composite.multitype_ops.ones_like_impl import ones_like
|
|
29
29
|
from mindspore.ops.composite.multitype_ops.zeros_like_impl import zeros_like
|
|
30
|
-
from mindspore.ops.
|
|
31
|
-
from mindspore.ops.composite.math_ops import count_nonzero, tensor_dot, dot, batch_dot, matmul, cummin
|
|
32
|
-
from mindspore.ops.
|
|
33
|
-
from mindspore.ops.
|
|
30
|
+
from mindspore.ops.function.random_func import normal, laplace, uniform, gamma, poisson, multinomial
|
|
31
|
+
from mindspore.ops.composite.math_ops import count_nonzero, tensor_dot, dot, batch_dot, matmul, cummin, mm
|
|
32
|
+
from mindspore.ops.function.array_func import repeat_interleave, repeat_elements, sequence_mask
|
|
33
|
+
from mindspore.ops.function.vmap_func import _VmapGeneralPreprocess, _VmapGeneralRule
|
|
34
34
|
from mindspore.ops.function.clip_func import clip_by_value
|
|
35
35
|
|
|
36
36
|
|
|
@@ -52,7 +52,6 @@ __all__ = [
|
|
|
52
52
|
'gamma',
|
|
53
53
|
'poisson',
|
|
54
54
|
'multinomial',
|
|
55
|
-
'clip_by_global_norm',
|
|
56
55
|
'count_nonzero',
|
|
57
56
|
'cummin',
|
|
58
57
|
'tensor_dot',
|
|
@@ -62,6 +61,6 @@ __all__ = [
|
|
|
62
61
|
'repeat_interleave',
|
|
63
62
|
'sequence_mask',
|
|
64
63
|
'matmul',
|
|
64
|
+
'mm',
|
|
65
65
|
'_Grad',
|
|
66
|
-
'_Vmap'
|
|
67
|
-
'_VmapGeneralPreprocess']
|
|
66
|
+
'_Vmap']
|