mindspore 2.1.0__cp38-none-any.whl → 2.2.0__cp38-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/__init__.py +4 -1
- mindspore/_akg/akg/build_module.py +5 -6
- mindspore/_akg/akg/composite/build_module.py +49 -16
- mindspore/_akg/akg/composite/split_stitch.py +10 -11
- mindspore/_akg/akg/ms/info_version_adapt.py +67 -1
- mindspore/_akg/akg/tvm/api.py +4 -3
- mindspore/_akg/akg/tvm/autotvm/__init__.py +1 -2
- mindspore/_akg/akg/tvm/autotvm/graph_tuner/base_graph_tuner.py +1 -5
- mindspore/_akg/akg/tvm/autotvm/measure/__init__.py +1 -1
- mindspore/_akg/akg/tvm/autotvm/measure/measure.py +1 -10
- mindspore/_akg/akg/tvm/autotvm/measure/measure_methods.py +1 -372
- mindspore/_akg/akg/tvm/build_module.py +16 -1
- mindspore/_akg/akg/tvm/contrib/graph_runtime.py +0 -53
- mindspore/_akg/akg/tvm/hybrid/parser.py +7 -6
- mindspore/_akg/akg/tvm/ir_builder.py +1 -1
- mindspore/_akg/akg/tvm/module.py +1 -2
- mindspore/_akg/akg/tvm/stmt.py +2 -2
- mindspore/_akg/akg/utils/composite_op_helper.py +9 -10
- mindspore/_akg/akg/utils/kernel_exec.py +58 -260
- mindspore/_akg/akg/utils/result_analysis.py +4 -24
- mindspore/_akg/akg/utils/tbe_codegen_utils.py +198 -0
- mindspore/_c_dataengine.cpython-38-aarch64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-38-aarch64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-38-aarch64-linux-gnu.so +0 -0
- mindspore/_check_jit_forbidden_api.py +3 -1
- mindspore/_checkparam.py +26 -32
- mindspore/_extends/graph_kernel/__init__.py +0 -1
- mindspore/_extends/graph_kernel/model/model_builder.py +9 -50
- mindspore/_extends/graph_kernel/splitter.py +1 -9
- mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +122 -15
- mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +2 -2
- mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +4 -2
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +2 -2
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +4 -4
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job.py +1 -1
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -1
- mindspore/_extends/parse/__init__.py +12 -15
- mindspore/_extends/parse/namespace.py +7 -33
- mindspore/_extends/parse/parser.py +61 -71
- mindspore/_extends/parse/resources.py +1 -1
- mindspore/_extends/parse/standard_method.py +72 -95
- mindspore/_extends/parse/trope.py +1 -1
- mindspore/_extends/remote/kernel_build_server.py +24 -7
- mindspore/_extends/remote/kernel_build_server_akg_v2.py +55 -0
- mindspore/_install_custom.py +43 -0
- mindspore/_mindspore_offline_debug.cpython-38-aarch64-linux-gnu.so +0 -0
- mindspore/amp.py +47 -11
- mindspore/bin/cache_admin +0 -0
- mindspore/bin/cache_server +0 -0
- mindspore/boost/boost.py +1 -8
- mindspore/boost/boost_cell_wrapper.py +3 -2
- mindspore/boost/grad_accumulation.py +1 -1
- mindspore/boost/group_loss_scale_manager.py +8 -7
- mindspore/common/__init__.py +5 -3
- mindspore/common/_jit_fallback_utils.py +6 -0
- mindspore/common/_register_for_adapter.py +2 -0
- mindspore/common/_register_for_tensor.py +2 -2
- mindspore/common/_stub_tensor.py +13 -0
- mindspore/common/_utils.py +13 -0
- mindspore/common/api.py +173 -258
- mindspore/common/auto_dynamic_shape.py +498 -0
- mindspore/common/dtype.py +18 -11
- mindspore/common/dump.py +6 -4
- mindspore/common/initializer.py +14 -14
- mindspore/common/jit_config.py +33 -15
- mindspore/common/lazy_inline.py +126 -7
- mindspore/common/mindir_util.py +101 -0
- mindspore/common/parameter.py +51 -41
- mindspore/common/seed.py +4 -4
- mindspore/common/sparse_tensor.py +13 -14
- mindspore/common/tensor.py +240 -145
- mindspore/communication/__init__.py +7 -4
- mindspore/communication/_comm_helper.py +83 -4
- mindspore/communication/management.py +152 -84
- mindspore/config/op_info.config +13 -2
- mindspore/config/super_bar_config.json +4 -2
- mindspore/context.py +143 -59
- mindspore/dataset/__init__.py +5 -5
- mindspore/dataset/audio/__init__.py +2 -2
- mindspore/dataset/audio/transforms.py +52 -52
- mindspore/dataset/callback/ds_callback.py +16 -2
- mindspore/dataset/core/config.py +68 -51
- mindspore/dataset/engine/cache_client.py +28 -5
- mindspore/dataset/engine/datasets.py +250 -112
- mindspore/dataset/engine/datasets_audio.py +43 -211
- mindspore/dataset/engine/datasets_standard_format.py +11 -35
- mindspore/dataset/engine/datasets_text.py +43 -67
- mindspore/dataset/engine/datasets_user_defined.py +86 -100
- mindspore/dataset/engine/datasets_vision.py +219 -1029
- mindspore/dataset/engine/iterators.py +11 -4
- mindspore/dataset/engine/obs/obs_mindrecord_dataset.py +4 -0
- mindspore/dataset/engine/obs/util.py +3 -0
- mindspore/dataset/engine/samplers.py +1 -1
- mindspore/dataset/engine/validators.py +19 -5
- mindspore/dataset/text/__init__.py +3 -3
- mindspore/dataset/text/transforms.py +101 -127
- mindspore/dataset/text/utils.py +205 -138
- mindspore/dataset/transforms/__init__.py +1 -1
- mindspore/dataset/transforms/py_transforms_util.py +40 -12
- mindspore/dataset/transforms/transforms.py +95 -40
- mindspore/dataset/utils/browse_dataset.py +8 -2
- mindspore/dataset/utils/line_reader.py +17 -19
- mindspore/dataset/vision/__init__.py +3 -3
- mindspore/dataset/vision/c_transforms.py +6 -3
- mindspore/dataset/vision/transforms.py +409 -287
- mindspore/dataset/vision/utils.py +13 -14
- mindspore/dataset/vision/validators.py +11 -1
- mindspore/experimental/map_parameter.py +14 -0
- mindspore/{nn/optim_ex → experimental/optim}/__init__.py +30 -29
- mindspore/{nn/optim_ex → experimental/optim}/adam.py +59 -66
- mindspore/{nn/optim_ex → experimental/optim}/adamw.py +181 -203
- mindspore/experimental/optim/lr_scheduler.py +1427 -0
- mindspore/{nn/optim_ex → experimental/optim}/optimizer.py +252 -259
- mindspore/{nn/optim_ex → experimental/optim}/sgd.py +147 -152
- mindspore/gen_ops.py +273 -0
- mindspore/include/OWNERS +0 -1
- mindspore/include/api/data_type.h +2 -1
- mindspore/include/api/graph.h +0 -15
- mindspore/include/api/kernel.h +2 -0
- mindspore/include/api/kernel_api.h +37 -12
- mindspore/include/api/model.h +0 -14
- mindspore/include/api/types.h +37 -4
- mindspore/include/c_api/ms/abstract.h +67 -0
- mindspore/include/c_api/ms/attribute.h +197 -0
- mindspore/include/c_api/ms/base/handle_types.h +43 -0
- mindspore/include/c_api/ms/base/macros.h +32 -0
- mindspore/include/c_api/ms/base/status.h +33 -0
- mindspore/include/c_api/ms/base/types.h +282 -0
- mindspore/include/c_api/ms/context.h +102 -0
- mindspore/include/c_api/ms/graph.h +160 -0
- mindspore/include/c_api/ms/node.h +606 -0
- mindspore/include/c_api/ms/tensor.h +161 -0
- mindspore/include/c_api/ms/value.h +84 -0
- mindspore/include/dataset/constants.h +6 -5
- mindspore/include/dataset/execute.h +23 -13
- mindspore/include/dataset/text.h +26 -26
- mindspore/include/dataset/transforms.h +13 -13
- mindspore/include/dataset/vision.h +60 -60
- mindspore/include/dataset/vision_ascend.h +5 -6
- mindspore/include/dataset/vision_lite.h +17 -17
- mindspore/include/mindapi/base/type_id.h +1 -0
- mindspore/include/mindapi/base/types.h +1 -0
- mindspore/lib/libdnnl.so.2 +0 -0
- mindspore/lib/libjemalloc.so.2 +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/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/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_aicpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_cpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/config/cust_aicpu_kernel.json +9000 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.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.2 +0 -0
- mindspore/mindrecord/tools/imagenet_to_mr.py +1 -1
- mindspore/mindrecord/tools/mnist_to_mr.py +2 -2
- mindspore/nn/__init__.py +0 -2
- mindspore/nn/cell.py +316 -74
- mindspore/nn/dynamic_lr.py +21 -21
- mindspore/nn/layer/activation.py +21 -28
- mindspore/nn/layer/basic.py +15 -13
- mindspore/nn/layer/channel_shuffle.py +1 -1
- mindspore/nn/layer/container.py +271 -9
- mindspore/nn/layer/conv.py +310 -207
- mindspore/nn/layer/dense.py +8 -5
- mindspore/nn/layer/embedding.py +33 -27
- mindspore/nn/layer/flash_attention.py +82 -41
- mindspore/nn/layer/image.py +8 -6
- mindspore/nn/layer/math.py +13 -18
- mindspore/nn/layer/normalization.py +107 -66
- mindspore/nn/layer/padding.py +1 -1
- mindspore/nn/layer/pooling.py +131 -109
- mindspore/nn/layer/rnn_cells.py +22 -17
- mindspore/nn/layer/rnns.py +13 -16
- mindspore/nn/layer/thor_layer.py +1 -1
- mindspore/nn/layer/transformer.py +221 -154
- mindspore/nn/learning_rate_schedule.py +9 -1
- mindspore/nn/loss/loss.py +235 -174
- mindspore/nn/optim/ada_grad.py +2 -1
- mindspore/nn/optim/adadelta.py +1 -0
- mindspore/nn/optim/adafactor.py +2 -1
- mindspore/nn/optim/adam.py +7 -4
- mindspore/nn/optim/adamax.py +3 -2
- mindspore/nn/optim/adasum.py +2 -2
- mindspore/nn/optim/asgd.py +2 -3
- mindspore/nn/optim/ftrl.py +6 -5
- mindspore/nn/optim/lamb.py +7 -4
- mindspore/nn/optim/lars.py +1 -1
- mindspore/nn/optim/lazyadam.py +5 -3
- mindspore/nn/optim/momentum.py +2 -1
- mindspore/nn/optim/optimizer.py +53 -4
- mindspore/nn/optim/proximal_ada_grad.py +3 -4
- mindspore/nn/optim/rmsprop.py +4 -3
- mindspore/nn/optim/rprop.py +23 -12
- mindspore/nn/optim/sgd.py +26 -11
- mindspore/nn/optim/thor.py +9 -7
- mindspore/nn/probability/bijector/bijector.py +5 -5
- mindspore/nn/probability/bijector/power_transform.py +27 -27
- mindspore/nn/probability/bijector/softplus.py +3 -3
- mindspore/nn/probability/distribution/_utils/custom_ops.py +3 -3
- mindspore/nn/probability/distribution/bernoulli.py +5 -5
- mindspore/nn/probability/distribution/beta.py +3 -3
- mindspore/nn/probability/distribution/categorical.py +7 -7
- mindspore/nn/probability/distribution/cauchy.py +0 -1
- mindspore/nn/probability/distribution/distribution.py +3 -3
- mindspore/nn/probability/distribution/gamma.py +3 -3
- mindspore/nn/probability/distribution/geometric.py +4 -4
- mindspore/nn/probability/distribution/gumbel.py +4 -4
- mindspore/nn/probability/distribution/log_normal.py +2 -2
- mindspore/nn/probability/distribution/logistic.py +2 -2
- mindspore/nn/probability/distribution/poisson.py +4 -4
- mindspore/nn/probability/distribution/transformed_distribution.py +3 -3
- mindspore/nn/probability/distribution/uniform.py +6 -6
- mindspore/nn/wrap/cell_wrapper.py +78 -34
- mindspore/nn/wrap/grad_reducer.py +8 -5
- mindspore/nn/wrap/loss_scale.py +105 -42
- mindspore/numpy/array_creations.py +1 -2
- mindspore/numpy/array_ops.py +3 -2
- mindspore/offline_debug/convert_async.py +2 -2
- mindspore/ops/_grad_experimental/__init__.py +0 -5
- mindspore/ops/_grad_experimental/grad_array_ops.py +1 -2
- mindspore/ops/_grad_experimental/grad_comm_ops.py +15 -2
- mindspore/ops/_grad_experimental/grad_debug_ops.py +0 -37
- mindspore/ops/_grad_experimental/grad_implementations.py +10 -0
- mindspore/ops/_grad_experimental/grad_inner_ops.py +2 -216
- mindspore/ops/_grad_experimental/grad_math_ops.py +0 -181
- mindspore/ops/_grad_experimental/grad_sparse.py +15 -0
- mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/flash_attention/attention.py +165 -109
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_bwd.py +144 -86
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_fwd.py +172 -187
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_impl.py +51 -57
- mindspore/ops/_op_impl/_custom_op/flash_attention/tik_ops_utils.py +6 -17
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/wukong_tiling.py +1 -1
- mindspore/ops/_op_impl/aicpu/__init__.py +14 -2
- mindspore/ops/_op_impl/aicpu/bias_add_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/count_nonzero.py +43 -0
- mindspore/ops/_op_impl/aicpu/eps.py +32 -0
- mindspore/ops/_op_impl/aicpu/gamma.py +2 -2
- mindspore/ops/_op_impl/aicpu/log_uniform_candidate_sampler.py +6 -3
- mindspore/ops/_op_impl/aicpu/lu_unpack_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/multinomial.py +3 -3
- mindspore/ops/_op_impl/aicpu/parameterized_truncated_normal.py +15 -7
- mindspore/ops/_op_impl/aicpu/random_categorical.py +39 -19
- mindspore/ops/_op_impl/aicpu/random_choice_with_mask.py +5 -2
- mindspore/ops/_op_impl/aicpu/random_poisson.py +103 -52
- mindspore/ops/_op_impl/aicpu/random_shuffle.py +17 -15
- mindspore/ops/_op_impl/aicpu/{sparseaddmm.py → sparse_addmm.py} +2 -2
- mindspore/ops/_op_impl/aicpu/{sparsesparsemaximum.py → sparse_sparse_maximum.py} +4 -4
- mindspore/ops/_op_impl/aicpu/standard_laplace.py +5 -5
- mindspore/ops/_op_impl/aicpu/standard_normal.py +5 -5
- mindspore/ops/_op_impl/aicpu/truncated_normal.py +9 -7
- mindspore/ops/_op_impl/aicpu/uniform.py +5 -3
- mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +8 -4
- mindspore/ops/_op_impl/aicpu/uniform_int.py +5 -5
- mindspore/ops/_op_impl/aicpu/uniform_real.py +4 -4
- mindspore/ops/_op_impl/tbe/__init__.py +4 -4
- mindspore/ops/_op_impl/tbe/inplace_index_add.py +7 -3
- mindspore/ops/_op_impl/tbe/trans_data_ds.py +2 -0
- mindspore/ops/_primitive_cache.py +1 -1
- mindspore/ops/_tracefunc.py +45 -13
- mindspore/ops/_utils/utils.py +4 -1
- mindspore/ops/_vmap/vmap_array_ops.py +3 -3
- mindspore/ops/_vmap/vmap_base.py +3 -3
- mindspore/ops/_vmap/vmap_convolution_ops.py +1 -1
- mindspore/ops/_vmap/vmap_grad_math_ops.py +6 -4
- mindspore/ops/_vmap/vmap_math_ops.py +5 -2
- mindspore/ops/_vmap/vmap_nn_ops.py +61 -7
- mindspore/ops/arg_dtype_cast.py +54 -0
- mindspore/ops/composite/base.py +37 -10
- mindspore/ops/composite/math_ops.py +5 -4
- mindspore/ops/composite/multitype_ops/_compile_utils.py +273 -72
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +16 -9
- mindspore/ops/composite/multitype_ops/add_impl.py +43 -4
- mindspore/ops/composite/multitype_ops/getitem_impl.py +40 -2
- mindspore/ops/composite/multitype_ops/ones_like_impl.py +6 -0
- mindspore/ops/composite/multitype_ops/setitem_impl.py +2 -1
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +9 -0
- mindspore/ops/deprecated.py +304 -0
- mindspore/ops/function/__init__.py +4 -1
- mindspore/ops/function/array_func.py +167 -189
- mindspore/ops/function/clip_func.py +81 -13
- mindspore/ops/function/debug_func.py +1 -1
- mindspore/ops/function/grad/grad_func.py +18 -8
- mindspore/ops/function/image_func.py +10 -4
- mindspore/ops/function/linalg_func.py +5 -5
- mindspore/ops/function/math_func.py +575 -386
- mindspore/ops/function/nn_func.py +470 -251
- mindspore/ops/function/random_func.py +86 -56
- mindspore/ops/function/sparse_func.py +1 -1
- mindspore/ops/function/sparse_unary_func.py +14 -12
- mindspore/ops/function/vmap_func.py +6 -5
- mindspore/ops/functional.py +15 -10
- mindspore/ops/op_info_register.py +235 -19
- mindspore/ops/operations/__init__.py +25 -17
- mindspore/ops/operations/_grad_ops.py +52 -7
- mindspore/ops/operations/_inner_ops.py +213 -12
- mindspore/ops/operations/_quant_ops.py +4 -8
- mindspore/ops/operations/_sequence_ops.py +42 -0
- mindspore/ops/operations/array_ops.py +64 -280
- mindspore/ops/operations/comm_ops.py +105 -57
- mindspore/ops/operations/custom_ops.py +10 -3
- mindspore/ops/operations/debug_ops.py +8 -4
- mindspore/ops/operations/image_ops.py +18 -12
- mindspore/ops/operations/math_ops.py +185 -138
- mindspore/ops/operations/nn_ops.py +716 -492
- mindspore/ops/operations/other_ops.py +0 -22
- mindspore/ops/operations/random_ops.py +53 -111
- mindspore/ops/operations/sparse_ops.py +3 -1
- mindspore/ops/primitive.py +24 -18
- mindspore/parallel/_auto_parallel_context.py +68 -8
- mindspore/parallel/_cost_model_context.py +2 -2
- mindspore/parallel/_offload_context.py +17 -3
- mindspore/parallel/_parallel_serialization.py +2 -2
- mindspore/parallel/_ps_context.py +12 -0
- mindspore/parallel/_tensor.py +14 -12
- mindspore/parallel/_transformer/layers.py +5 -3
- mindspore/parallel/_transformer/loss.py +1 -0
- mindspore/parallel/_transformer/moe.py +2 -2
- mindspore/parallel/_transformer/op_parallel_config.py +12 -1
- mindspore/parallel/_transformer/transformer.py +23 -3
- mindspore/parallel/_utils.py +11 -7
- mindspore/parallel/algo_parameter_config.py +85 -5
- mindspore/parallel/checkpoint_transform.py +6 -10
- mindspore/parallel/shard.py +4 -4
- mindspore/profiler/common/struct_type.py +3 -3
- mindspore/profiler/common/util.py +3 -2
- mindspore/profiler/envprofiling.py +1 -1
- mindspore/profiler/parser/aicpu_data_parser.py +5 -3
- mindspore/profiler/parser/ascend_flops_generator.py +2 -2
- mindspore/profiler/parser/ascend_fpbp_generator.py +1 -1
- mindspore/profiler/parser/ascend_hccl_generator.py +17 -12
- mindspore/profiler/parser/ascend_msprof_exporter.py +104 -252
- mindspore/profiler/parser/ascend_msprof_generator.py +8 -8
- mindspore/profiler/parser/ascend_op_generator.py +5 -5
- mindspore/profiler/parser/ascend_steptrace_generator.py +6 -4
- mindspore/profiler/parser/ascend_timeline_generator.py +9 -6
- mindspore/profiler/parser/base_timeline_generator.py +9 -7
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +14 -10
- mindspore/profiler/parser/flops_parser.py +15 -11
- mindspore/profiler/parser/framework_parser.py +37 -21
- mindspore/profiler/parser/hccl_parser.py +16 -12
- mindspore/profiler/parser/integrator.py +22 -11
- mindspore/profiler/parser/memory_usage_parser.py +2 -2
- mindspore/profiler/parser/minddata_analyzer.py +12 -14
- mindspore/profiler/parser/minddata_pipeline_parser.py +1 -1
- mindspore/profiler/parser/msadvisor_parser.py +8 -4
- mindspore/profiler/parser/op_intermediate_parser.py +5 -2
- mindspore/profiler/parser/optime_parser.py +1 -1
- mindspore/profiler/parser/profiler_info.py +2 -2
- mindspore/profiler/parser/step_trace_parser.py +11 -14
- mindspore/profiler/profiling.py +139 -71
- mindspore/rewrite/api/node.py +102 -19
- mindspore/rewrite/api/node_type.py +5 -1
- mindspore/rewrite/api/scoped_value.py +9 -17
- mindspore/rewrite/api/symbol_tree.py +131 -47
- mindspore/rewrite/ast_helpers/__init__.py +2 -1
- mindspore/rewrite/ast_helpers/ast_finder.py +129 -0
- mindspore/rewrite/ast_helpers/ast_modifier.py +116 -104
- mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +93 -46
- mindspore/rewrite/common/rewrite_elog.py +5 -1
- mindspore/rewrite/namer.py +33 -24
- mindspore/rewrite/namespace.py +14 -5
- mindspore/{_extends/graph_kernel/expanders/complex → rewrite/node}/__init__.py +9 -9
- mindspore/rewrite/node/call_function.py +79 -0
- mindspore/rewrite/node/cell_container.py +135 -0
- mindspore/rewrite/node/control_flow.py +88 -0
- mindspore/rewrite/{node.py → node/node.py} +273 -234
- mindspore/rewrite/node/node_manager.py +254 -0
- mindspore/rewrite/{topological_manager.py → node/node_topological_manager.py} +13 -46
- mindspore/rewrite/parsers/arguments_parser.py +22 -21
- mindspore/rewrite/parsers/assign_parser.py +216 -221
- mindspore/rewrite/parsers/attribute_parser.py +9 -7
- mindspore/rewrite/parsers/class_def_parser.py +174 -113
- mindspore/rewrite/parsers/constant_parser.py +9 -6
- mindspore/rewrite/parsers/container_parser.py +9 -7
- mindspore/rewrite/parsers/for_parser.py +36 -15
- mindspore/rewrite/parsers/function_def_parser.py +24 -16
- mindspore/rewrite/parsers/if_parser.py +28 -24
- mindspore/rewrite/parsers/module_parser.py +196 -25
- mindspore/rewrite/{parser.py → parsers/parser.py} +4 -2
- mindspore/rewrite/{parser_register.py → parsers/parser_register.py} +1 -1
- mindspore/rewrite/parsers/return_parser.py +6 -6
- mindspore/rewrite/sparsify/sparse_transformer.py +12 -3
- mindspore/rewrite/sparsify/utils.py +1 -1
- mindspore/rewrite/symbol_tree.py +525 -577
- mindspore/rewrite/symbol_tree_builder.py +9 -193
- mindspore/rewrite/symbol_tree_dumper.py +2 -2
- mindspore/run_check/_check_version.py +2 -2
- mindspore/{ops/bprop_mindir → safeguard}/__init__.py +4 -3
- mindspore/safeguard/rewrite_obfuscation.py +517 -0
- mindspore/scipy/linalg.py +1 -1
- mindspore/scipy/optimize/minimize.py +7 -3
- mindspore/train/_utils.py +7 -3
- mindspore/train/amp.py +323 -123
- mindspore/train/anf_ir_pb2.py +14 -2
- mindspore/train/callback/_backup_and_restore.py +2 -12
- mindspore/train/callback/_callback.py +29 -4
- mindspore/train/callback/_checkpoint.py +23 -8
- mindspore/train/callback/_early_stop.py +2 -2
- mindspore/train/callback/_landscape.py +4 -4
- mindspore/train/callback/_loss_monitor.py +2 -2
- mindspore/train/callback/_on_request_exit.py +2 -2
- mindspore/train/callback/_reduce_lr_on_plateau.py +3 -4
- mindspore/train/callback/_summary_collector.py +14 -7
- mindspore/train/callback/_time_monitor.py +58 -5
- mindspore/train/data_sink.py +5 -11
- mindspore/train/dataset_helper.py +83 -57
- mindspore/train/loss_scale_manager.py +2 -2
- mindspore/train/metrics/__init__.py +3 -3
- mindspore/train/metrics/cosine_similarity.py +1 -1
- mindspore/train/metrics/hausdorff_distance.py +3 -2
- mindspore/train/metrics/mean_surface_distance.py +3 -2
- mindspore/train/metrics/metric.py +39 -19
- mindspore/train/metrics/roc.py +2 -2
- mindspore/train/metrics/root_mean_square_surface_distance.py +4 -3
- mindspore/train/mind_ir_pb2.py +85 -36
- mindspore/train/model.py +185 -45
- mindspore/train/serialization.py +390 -150
- mindspore/train/summary/_writer_pool.py +3 -2
- mindspore/train/summary/summary_record.py +14 -10
- mindspore/train/train_thor/convert_utils.py +3 -3
- mindspore/train/train_thor/dataset_helper.py +1 -1
- mindspore/version.py +1 -1
- {mindspore-2.1.0.dist-info → mindspore-2.2.0.dist-info}/METADATA +6 -7
- {mindspore-2.1.0.dist-info → mindspore-2.2.0.dist-info}/RECORD +447 -507
- {mindspore-2.1.0.dist-info → mindspore-2.2.0.dist-info}/entry_points.txt +0 -1
- mindspore/_akg/akg/tvm/contrib/debugger/__init__.py +0 -16
- mindspore/_akg/akg/tvm/contrib/debugger/debug_result.py +0 -274
- mindspore/_akg/akg/tvm/contrib/debugger/debug_runtime.py +0 -259
- mindspore/_akg/akg/tvm/contrib/peak.py +0 -341
- mindspore/_akg/akg/tvm/contrib/rpc.py +0 -25
- mindspore/_akg/akg/tvm/contrib/xcode.py +0 -257
- mindspore/_akg/akg/tvm/exec/__init__.py +0 -17
- mindspore/_akg/akg/tvm/exec/autotvm_log_editor.py +0 -60
- mindspore/_akg/akg/tvm/exec/measure_peak.py +0 -48
- mindspore/_akg/akg/tvm/exec/query_rpc_tracker.py +0 -48
- mindspore/_akg/akg/tvm/exec/rpc_proxy.py +0 -98
- mindspore/_akg/akg/tvm/exec/rpc_server.py +0 -88
- mindspore/_akg/akg/tvm/exec/rpc_tracker.py +0 -62
- mindspore/_akg/akg/tvm/rpc/__init__.py +0 -29
- mindspore/_akg/akg/tvm/rpc/base.py +0 -182
- mindspore/_akg/akg/tvm/rpc/client.py +0 -436
- mindspore/_akg/akg/tvm/rpc/proxy.py +0 -595
- mindspore/_akg/akg/tvm/rpc/server.py +0 -413
- mindspore/_akg/akg/tvm/rpc/tornado_util.py +0 -121
- mindspore/_akg/akg/tvm/rpc/tracker.py +0 -431
- mindspore/_extends/graph_kernel/expander.py +0 -80
- mindspore/_extends/graph_kernel/expanders/__init__.py +0 -54
- mindspore/_extends/graph_kernel/expanders/_utils.py +0 -269
- mindspore/_extends/graph_kernel/expanders/addn.py +0 -33
- mindspore/_extends/graph_kernel/expanders/batchnorm.py +0 -152
- mindspore/_extends/graph_kernel/expanders/batchnorm_grad.py +0 -105
- mindspore/_extends/graph_kernel/expanders/clip_by_norm_no_div_sum.py +0 -33
- mindspore/_extends/graph_kernel/expanders/complex/abs.py +0 -30
- mindspore/_extends/graph_kernel/expanders/complex/add.py +0 -44
- mindspore/_extends/graph_kernel/expanders/complex/div.py +0 -62
- mindspore/_extends/graph_kernel/expanders/complex/mul.py +0 -52
- mindspore/_extends/graph_kernel/expanders/complex/real_div.py +0 -62
- mindspore/_extends/graph_kernel/expanders/complex/sub.py +0 -45
- mindspore/_extends/graph_kernel/expanders/conv2d.py +0 -200
- mindspore/_extends/graph_kernel/expanders/dropout_grad.py +0 -30
- mindspore/_extends/graph_kernel/expanders/equal_count.py +0 -50
- mindspore/_extends/graph_kernel/expanders/erfc.py +0 -35
- mindspore/_extends/graph_kernel/expanders/expand_dims.py +0 -50
- mindspore/_extends/graph_kernel/expanders/fused_adam.py +0 -44
- mindspore/_extends/graph_kernel/expanders/fused_adam_weight_decay.py +0 -47
- mindspore/_extends/graph_kernel/expanders/fused_mul_add.py +0 -28
- mindspore/_extends/graph_kernel/expanders/gelu_grad.py +0 -70
- mindspore/_extends/graph_kernel/expanders/gkdropout.py +0 -40
- mindspore/_extends/graph_kernel/expanders/identity.py +0 -25
- mindspore/_extends/graph_kernel/expanders/layernorm.py +0 -93
- mindspore/_extends/graph_kernel/expanders/layernorm_grad.py +0 -113
- mindspore/_extends/graph_kernel/expanders/logsoftmax.py +0 -46
- mindspore/_extends/graph_kernel/expanders/logsoftmax_grad.py +0 -36
- mindspore/_extends/graph_kernel/expanders/matmul.py +0 -80
- mindspore/_extends/graph_kernel/expanders/maximum_grad.py +0 -59
- mindspore/_extends/graph_kernel/expanders/minimum_grad.py +0 -80
- mindspore/_extends/graph_kernel/expanders/oneslike.py +0 -26
- mindspore/_extends/graph_kernel/expanders/reduce_mean.py +0 -43
- mindspore/_extends/graph_kernel/expanders/relu_grad.py +0 -32
- mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits.py +0 -41
- mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits_grad.py +0 -35
- mindspore/_extends/graph_kernel/expanders/sigmoid_grad.py +0 -31
- mindspore/_extends/graph_kernel/expanders/slice.py +0 -35
- mindspore/_extends/graph_kernel/expanders/softmax_cross_entropy_with_logits.py +0 -42
- mindspore/_extends/graph_kernel/expanders/softmax_grad_ext.py +0 -41
- mindspore/_extends/graph_kernel/expanders/softsign.py +0 -28
- mindspore/_extends/graph_kernel/expanders/sqrt_grad.py +0 -29
- mindspore/_extends/graph_kernel/expanders/square_sum_all.py +0 -44
- mindspore/_extends/graph_kernel/expanders/square_sum_v1.py +0 -37
- mindspore/_extends/graph_kernel/expanders/squared_difference.py +0 -43
- mindspore/_extends/graph_kernel/expanders/tanh_grad.py +0 -31
- mindspore/_extends/graph_kernel/model/op_infer.py +0 -506
- mindspore/dataset/datapreprocess/__init__.py +0 -20
- mindspore/dataset/datapreprocess/preprocess_imagenet_validate_dataset.py +0 -54
- mindspore/include/api/net.h +0 -142
- mindspore/nn/lr_scheduler.py +0 -262
- mindspore/ops/_grad_experimental/grad_image_ops.py +0 -248
- mindspore/ops/_grad_experimental/grad_linalg_ops.py +0 -181
- mindspore/ops/_grad_experimental/grad_other_ops.py +0 -72
- mindspore/ops/_grad_experimental/grad_scalar_ops.py +0 -112
- mindspore/ops/_grad_experimental/grad_sequence_ops.py +0 -351
- mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Depend_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +0 -138
- mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Load_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Switch_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TransShape_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Unique_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Unstack_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/generate_mindir.py +0 -114
- mindspore/rewrite/node_visitor.py +0 -44
- {mindspore-2.1.0.dist-info → mindspore-2.2.0.dist-info}/WHEEL +0 -0
- {mindspore-2.1.0.dist-info → mindspore-2.2.0.dist-info}/top_level.txt +0 -0
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
# Copyright 2019-
|
|
1
|
+
# Copyright 2019-2023 Huawei Technologies Co., Ltd
|
|
2
2
|
#
|
|
3
3
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
4
|
# you may not use this file except in compliance with the License.
|
|
@@ -63,7 +63,7 @@ class CMUArcticDataset(MappableDataset, AudioBaseDataset):
|
|
|
63
63
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None``, will use ``0``. This
|
|
64
64
|
argument can only be specified when `num_shards` is also specified.
|
|
65
65
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
66
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
66
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
67
67
|
Default: ``None``, which means no cache is used.
|
|
68
68
|
|
|
69
69
|
Raises:
|
|
@@ -77,38 +77,14 @@ class CMUArcticDataset(MappableDataset, AudioBaseDataset):
|
|
|
77
77
|
|
|
78
78
|
Tutorial Examples:
|
|
79
79
|
- `Load & Process Data With Dataset Pipeline
|
|
80
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
80
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
81
81
|
|
|
82
82
|
Note:
|
|
83
83
|
- Not support :class:`mindspore.dataset.PKSampler` for `sampler` parameter yet.
|
|
84
|
-
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
..
|
|
88
|
-
:widths: 25 25 50
|
|
89
|
-
:header-rows: 1
|
|
90
|
-
|
|
91
|
-
* - Parameter `sampler`
|
|
92
|
-
- Parameter `shuffle`
|
|
93
|
-
- Expected Order Behavior
|
|
94
|
-
* - None
|
|
95
|
-
- None
|
|
96
|
-
- random order
|
|
97
|
-
* - None
|
|
98
|
-
- True
|
|
99
|
-
- random order
|
|
100
|
-
* - None
|
|
101
|
-
- False
|
|
102
|
-
- sequential order
|
|
103
|
-
* - Sampler object
|
|
104
|
-
- None
|
|
105
|
-
- order defined by sampler
|
|
106
|
-
* - Sampler object
|
|
107
|
-
- True
|
|
108
|
-
- not allowed
|
|
109
|
-
* - Sampler object
|
|
110
|
-
- False
|
|
111
|
-
- not allowed
|
|
84
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
85
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
86
|
+
|
|
87
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
112
88
|
|
|
113
89
|
Examples:
|
|
114
90
|
>>> import mindspore.dataset as ds
|
|
@@ -204,7 +180,7 @@ class GTZANDataset(MappableDataset, AudioBaseDataset):
|
|
|
204
180
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None`` . This
|
|
205
181
|
argument can only be specified when `num_shards` is also specified.
|
|
206
182
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
207
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
183
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
208
184
|
Default: ``None`` , which means no cache is used.
|
|
209
185
|
|
|
210
186
|
Raises:
|
|
@@ -218,38 +194,14 @@ class GTZANDataset(MappableDataset, AudioBaseDataset):
|
|
|
218
194
|
|
|
219
195
|
Tutorial Examples:
|
|
220
196
|
- `Load & Process Data With Dataset Pipeline
|
|
221
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
197
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
222
198
|
|
|
223
199
|
Note:
|
|
224
200
|
- Not support :class:`mindspore.dataset.PKSampler` for `sampler` parameter yet.
|
|
225
|
-
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
..
|
|
229
|
-
:widths: 25 25 50
|
|
230
|
-
:header-rows: 1
|
|
231
|
-
|
|
232
|
-
* - Parameter `sampler`
|
|
233
|
-
- Parameter `shuffle`
|
|
234
|
-
- Expected Order Behavior
|
|
235
|
-
* - None
|
|
236
|
-
- None
|
|
237
|
-
- random order
|
|
238
|
-
* - None
|
|
239
|
-
- True
|
|
240
|
-
- random order
|
|
241
|
-
* - None
|
|
242
|
-
- False
|
|
243
|
-
- sequential order
|
|
244
|
-
* - Sampler object
|
|
245
|
-
- None
|
|
246
|
-
- order defined by sampler
|
|
247
|
-
* - Sampler object
|
|
248
|
-
- True
|
|
249
|
-
- not allowed
|
|
250
|
-
* - Sampler object
|
|
251
|
-
- False
|
|
252
|
-
- not allowed
|
|
201
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
202
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
203
|
+
|
|
204
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
253
205
|
|
|
254
206
|
Examples:
|
|
255
207
|
>>> import mindspore.dataset as ds
|
|
@@ -346,7 +298,7 @@ class LibriTTSDataset(MappableDataset, AudioBaseDataset):
|
|
|
346
298
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None`` . This
|
|
347
299
|
argument can only be specified when `num_shards` is also specified.
|
|
348
300
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
349
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
301
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
350
302
|
Default: ``None`` , which means no cache is used.
|
|
351
303
|
|
|
352
304
|
Raises:
|
|
@@ -360,38 +312,14 @@ class LibriTTSDataset(MappableDataset, AudioBaseDataset):
|
|
|
360
312
|
|
|
361
313
|
Tutorial Examples:
|
|
362
314
|
- `Load & Process Data With Dataset Pipeline
|
|
363
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
315
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
364
316
|
|
|
365
317
|
Note:
|
|
366
318
|
- Not support :class:`mindspore.dataset.PKSampler` for `sampler` parameter yet.
|
|
367
|
-
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
..
|
|
371
|
-
:widths: 25 25 50
|
|
372
|
-
:header-rows: 1
|
|
373
|
-
|
|
374
|
-
* - Parameter `sampler`
|
|
375
|
-
- Parameter `shuffle`
|
|
376
|
-
- Expected Order Behavior
|
|
377
|
-
* - None
|
|
378
|
-
- None
|
|
379
|
-
- random order
|
|
380
|
-
* - None
|
|
381
|
-
- True
|
|
382
|
-
- random order
|
|
383
|
-
* - None
|
|
384
|
-
- False
|
|
385
|
-
- sequential order
|
|
386
|
-
* - Sampler object
|
|
387
|
-
- None
|
|
388
|
-
- order defined by sampler
|
|
389
|
-
* - Sampler object
|
|
390
|
-
- True
|
|
391
|
-
- not allowed
|
|
392
|
-
* - Sampler object
|
|
393
|
-
- False
|
|
394
|
-
- not allowed
|
|
319
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
320
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
321
|
+
|
|
322
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
395
323
|
|
|
396
324
|
Examples:
|
|
397
325
|
>>> import mindspore.dataset as ds
|
|
@@ -497,7 +425,7 @@ class LJSpeechDataset(MappableDataset, AudioBaseDataset):
|
|
|
497
425
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None`` . This
|
|
498
426
|
argument can only be specified when `num_shards` is also specified.
|
|
499
427
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
500
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
428
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
501
429
|
Default: ``None`` , which means no cache is used.
|
|
502
430
|
|
|
503
431
|
Raises:
|
|
@@ -511,37 +439,13 @@ class LJSpeechDataset(MappableDataset, AudioBaseDataset):
|
|
|
511
439
|
|
|
512
440
|
Tutorial Examples:
|
|
513
441
|
- `Load & Process Data With Dataset Pipeline
|
|
514
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
442
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
515
443
|
|
|
516
444
|
Note:
|
|
517
|
-
-
|
|
518
|
-
|
|
519
|
-
|
|
520
|
-
..
|
|
521
|
-
:widths: 25 25 50
|
|
522
|
-
:header-rows: 1
|
|
523
|
-
|
|
524
|
-
* - Parameter `sampler`
|
|
525
|
-
- Parameter `shuffle`
|
|
526
|
-
- Expected Order Behavior
|
|
527
|
-
* - None
|
|
528
|
-
- None
|
|
529
|
-
- random order
|
|
530
|
-
* - None
|
|
531
|
-
- True
|
|
532
|
-
- random order
|
|
533
|
-
* - None
|
|
534
|
-
- False
|
|
535
|
-
- sequential order
|
|
536
|
-
* - Sampler object
|
|
537
|
-
- None
|
|
538
|
-
- order defined by sampler
|
|
539
|
-
* - Sampler object
|
|
540
|
-
- True
|
|
541
|
-
- not allowed
|
|
542
|
-
* - Sampler object
|
|
543
|
-
- False
|
|
544
|
-
- not allowed
|
|
445
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
446
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
447
|
+
|
|
448
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
545
449
|
|
|
546
450
|
Examples:
|
|
547
451
|
>>> import mindspore.dataset as ds
|
|
@@ -644,7 +548,7 @@ class SpeechCommandsDataset(MappableDataset, AudioBaseDataset):
|
|
|
644
548
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None`` .
|
|
645
549
|
This argument can only be specified when `num_shards` is also specified.
|
|
646
550
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
647
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
551
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
648
552
|
Default: ``None`` , which means no cache is used.
|
|
649
553
|
|
|
650
554
|
Raises:
|
|
@@ -658,37 +562,13 @@ class SpeechCommandsDataset(MappableDataset, AudioBaseDataset):
|
|
|
658
562
|
|
|
659
563
|
Tutorial Examples:
|
|
660
564
|
- `Load & Process Data With Dataset Pipeline
|
|
661
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
565
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
662
566
|
|
|
663
567
|
Note:
|
|
664
|
-
-
|
|
665
|
-
|
|
666
|
-
|
|
667
|
-
..
|
|
668
|
-
:widths: 25 25 50
|
|
669
|
-
:header-rows: 1
|
|
670
|
-
|
|
671
|
-
* - Parameter `sampler`
|
|
672
|
-
- Parameter `shuffle`
|
|
673
|
-
- Expected Order Behavior
|
|
674
|
-
* - None
|
|
675
|
-
- None
|
|
676
|
-
- random order
|
|
677
|
-
* - None
|
|
678
|
-
- True
|
|
679
|
-
- random order
|
|
680
|
-
* - None
|
|
681
|
-
- False
|
|
682
|
-
- sequential order
|
|
683
|
-
* - Sampler object
|
|
684
|
-
- None
|
|
685
|
-
- order defined by sampler
|
|
686
|
-
* - Sampler object
|
|
687
|
-
- True
|
|
688
|
-
- not allowed
|
|
689
|
-
* - Sampler object
|
|
690
|
-
- False
|
|
691
|
-
- not allowed
|
|
568
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
569
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
570
|
+
|
|
571
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
692
572
|
|
|
693
573
|
Examples:
|
|
694
574
|
>>> import mindspore.dataset as ds
|
|
@@ -781,7 +661,7 @@ class TedliumDataset(MappableDataset, AudioBaseDataset):
|
|
|
781
661
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None`` . This
|
|
782
662
|
argument can only be specified when `num_shards` is also specified.
|
|
783
663
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
784
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
664
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
785
665
|
Default: ``None`` , which means no cache is used.
|
|
786
666
|
|
|
787
667
|
Raises:
|
|
@@ -795,37 +675,13 @@ class TedliumDataset(MappableDataset, AudioBaseDataset):
|
|
|
795
675
|
|
|
796
676
|
Tutorial Examples:
|
|
797
677
|
- `Load & Process Data With Dataset Pipeline
|
|
798
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
678
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
799
679
|
|
|
800
680
|
Note:
|
|
801
|
-
-
|
|
802
|
-
|
|
803
|
-
|
|
804
|
-
..
|
|
805
|
-
:widths: 25 25 50
|
|
806
|
-
:header-rows: 1
|
|
807
|
-
|
|
808
|
-
* - Parameter `sampler`
|
|
809
|
-
- Parameter `shuffle`
|
|
810
|
-
- Expected Order Behavior
|
|
811
|
-
* - None
|
|
812
|
-
- None
|
|
813
|
-
- random order
|
|
814
|
-
* - None
|
|
815
|
-
- True
|
|
816
|
-
- random order
|
|
817
|
-
* - None
|
|
818
|
-
- False
|
|
819
|
-
- sequential order
|
|
820
|
-
* - Sampler object
|
|
821
|
-
- None
|
|
822
|
-
- order defined by sampler
|
|
823
|
-
* - Sampler object
|
|
824
|
-
- True
|
|
825
|
-
- not allowed
|
|
826
|
-
* - Sampler object
|
|
827
|
-
- False
|
|
828
|
-
- not allowed
|
|
681
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
682
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
683
|
+
|
|
684
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
829
685
|
|
|
830
686
|
Examples:
|
|
831
687
|
>>> import mindspore.dataset as ds
|
|
@@ -985,7 +841,7 @@ class YesNoDataset(MappableDataset, AudioBaseDataset):
|
|
|
985
841
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None`` . This argument can only
|
|
986
842
|
be specified when `num_shards` is also specified.
|
|
987
843
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
988
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
844
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
989
845
|
Default: ``None`` , which means no cache is used.
|
|
990
846
|
|
|
991
847
|
Raises:
|
|
@@ -999,37 +855,13 @@ class YesNoDataset(MappableDataset, AudioBaseDataset):
|
|
|
999
855
|
|
|
1000
856
|
Tutorial Examples:
|
|
1001
857
|
- `Load & Process Data With Dataset Pipeline
|
|
1002
|
-
<https://www.mindspore.cn/docs/en/r2.
|
|
858
|
+
<https://www.mindspore.cn/docs/en/r2.2/api_python/samples/dataset/dataset_gallery.html>`_
|
|
1003
859
|
|
|
1004
860
|
Note:
|
|
1005
|
-
-
|
|
1006
|
-
|
|
1007
|
-
|
|
1008
|
-
..
|
|
1009
|
-
:widths: 25 25 50
|
|
1010
|
-
:header-rows: 1
|
|
1011
|
-
|
|
1012
|
-
* - Parameter `sampler`
|
|
1013
|
-
- Parameter `shuffle`
|
|
1014
|
-
- Expected Order Behavior
|
|
1015
|
-
* - None
|
|
1016
|
-
- None
|
|
1017
|
-
- random order
|
|
1018
|
-
* - None
|
|
1019
|
-
- True
|
|
1020
|
-
- random order
|
|
1021
|
-
* - None
|
|
1022
|
-
- False
|
|
1023
|
-
- sequential order
|
|
1024
|
-
* - Sampler object
|
|
1025
|
-
- None
|
|
1026
|
-
- order defined by sampler
|
|
1027
|
-
* - Sampler object
|
|
1028
|
-
- True
|
|
1029
|
-
- not allowed
|
|
1030
|
-
* - Sampler object
|
|
1031
|
-
- False
|
|
1032
|
-
- not allowed
|
|
861
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
862
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
863
|
+
|
|
864
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
1033
865
|
|
|
1034
866
|
Examples:
|
|
1035
867
|
>>> import mindspore.dataset as ds
|
|
@@ -76,7 +76,7 @@ class CSVDataset(SourceDataset, UnionBaseDataset):
|
|
|
76
76
|
shard_id (int, optional): The shard ID within `num_shards` . Default: ``None``. This
|
|
77
77
|
argument can only be specified when `num_shards` is also specified.
|
|
78
78
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
79
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
79
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
80
80
|
Default: ``None``, which means no cache is used.
|
|
81
81
|
|
|
82
82
|
Raises:
|
|
@@ -149,13 +149,13 @@ class MindDataset(MappableDataset, UnionBaseDataset):
|
|
|
149
149
|
:class:`mindspore.dataset.PKSampler`, :class:`mindspore.dataset.RandomSampler`,
|
|
150
150
|
:class:`mindspore.dataset.SequentialSampler`, :class:`mindspore.dataset.DistributedSampler`.
|
|
151
151
|
padded_sample (dict, optional): Samples will be appended to dataset, where
|
|
152
|
-
keys are the same as
|
|
152
|
+
keys are the same as columns_list. Default: ``None``.
|
|
153
153
|
num_padded (int, optional): Number of padding samples. Dataset size
|
|
154
154
|
plus num_padded should be divisible by num_shards. Default: ``None``.
|
|
155
155
|
num_samples (int, optional): The number of samples to be included in the dataset.
|
|
156
156
|
Default: ``None`` , all samples.
|
|
157
157
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
158
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
158
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
159
159
|
Default: ``None`` , which means no cache is used.
|
|
160
160
|
|
|
161
161
|
Raises:
|
|
@@ -166,34 +166,10 @@ class MindDataset(MappableDataset, UnionBaseDataset):
|
|
|
166
166
|
ValueError: If `shard_id` is not in range of [0, `num_shards` ).
|
|
167
167
|
|
|
168
168
|
Note:
|
|
169
|
-
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
..
|
|
173
|
-
:widths: 25 25 50
|
|
174
|
-
:header-rows: 1
|
|
175
|
-
|
|
176
|
-
* - Parameter `sampler`
|
|
177
|
-
- Parameter `shuffle`
|
|
178
|
-
- Expected Order Behavior
|
|
179
|
-
* - None
|
|
180
|
-
- None
|
|
181
|
-
- random order
|
|
182
|
-
* - None
|
|
183
|
-
- True
|
|
184
|
-
- random order
|
|
185
|
-
* - None
|
|
186
|
-
- False
|
|
187
|
-
- sequential order
|
|
188
|
-
* - Sampler object
|
|
189
|
-
- None
|
|
190
|
-
- order defined by sampler
|
|
191
|
-
* - Sampler object
|
|
192
|
-
- True
|
|
193
|
-
- not allowed
|
|
194
|
-
* - Sampler object
|
|
195
|
-
- False
|
|
196
|
-
- not allowed
|
|
169
|
+
- The parameters `num_samples` , `shuffle` , `num_shards` , `shard_id` can be used to control the sampler
|
|
170
|
+
used in the dataset, and their effects when combined with parameter `sampler` are as follows.
|
|
171
|
+
|
|
172
|
+
.. include:: mindspore.dataset.sampler.txt
|
|
197
173
|
|
|
198
174
|
Examples:
|
|
199
175
|
>>> import mindspore.dataset as ds
|
|
@@ -298,7 +274,7 @@ class TFRecordDataset(SourceDataset, UnionBaseDataset):
|
|
|
298
274
|
When `compression_type` is not ``None``, and `num_samples` or numRows (parsed from `schema` ) is provided,
|
|
299
275
|
`shard_equal_rows` will be implied as ``True``.
|
|
300
276
|
cache (DatasetCache, optional): Use tensor caching service to speed up dataset processing. More details:
|
|
301
|
-
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.
|
|
277
|
+
`Single-Node Data Cache <https://www.mindspore.cn/tutorials/experts/en/r2.2/dataset/cache.html>`_ .
|
|
302
278
|
Default: ``None`` , which means no cache is used.
|
|
303
279
|
compression_type (str, optional): The type of compression used for all files, must be either ``''``,
|
|
304
280
|
``'GZIP'``, or ``'ZLIB'``. Default: ``None`` , as in empty string. It is highly recommended to
|
|
@@ -376,11 +352,11 @@ class OBSMindDataset(GeneratorDataset):
|
|
|
376
352
|
dataset_files (list[str]): List of files in cloud storage to be read and file path is in
|
|
377
353
|
the format of s3://bucketName/objectKey.
|
|
378
354
|
server (str): Endpoint for accessing cloud storage.
|
|
379
|
-
If it's OBS Service of Huawei Cloud
|
|
355
|
+
If it's OBS Service of Huawei Cloud, the endpoint is
|
|
380
356
|
like ``<obs.cn-north-4.myhuaweicloud.com>`` (Region cn-north-4).
|
|
381
357
|
If it's Minio which starts locally, the endpoint is like ``<https://127.0.0.1:9000>``.
|
|
382
|
-
ak (str):
|
|
383
|
-
sk (str):
|
|
358
|
+
ak (str): The access key ID used to access the OBS data.
|
|
359
|
+
sk (str): The secret access key used to access the OBS data.
|
|
384
360
|
sync_obs_path (str): Remote dir path used for synchronization, users need to
|
|
385
361
|
create it on cloud storage in advance. Path is in the format of s3://bucketName/objectKey.
|
|
386
362
|
columns_list (list[str], optional): List of columns to be read. Default: ``None`` , read all columns.
|