mindspore 2.0.0rc1__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.

Files changed (870) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/Third_Party_Open_Source_Software_Notice +2 -2
  3. mindspore/__init__.py +5 -2
  4. mindspore/_akg/akg/build_module.py +5 -6
  5. mindspore/_akg/akg/composite/build_module.py +49 -16
  6. mindspore/_akg/akg/composite/split_stitch.py +10 -11
  7. mindspore/_akg/akg/config/repository.json +195 -0
  8. mindspore/_akg/akg/global_configs.py +5 -1
  9. mindspore/_akg/akg/ms/info_version_adapt.py +67 -1
  10. mindspore/_akg/akg/tvm/api.py +4 -3
  11. mindspore/_akg/akg/tvm/autotvm/__init__.py +1 -2
  12. mindspore/_akg/akg/tvm/autotvm/graph_tuner/base_graph_tuner.py +1 -5
  13. mindspore/_akg/akg/tvm/autotvm/measure/__init__.py +1 -1
  14. mindspore/_akg/akg/tvm/autotvm/measure/measure.py +1 -10
  15. mindspore/_akg/akg/tvm/autotvm/measure/measure_methods.py +1 -372
  16. mindspore/_akg/akg/tvm/build_module.py +16 -1
  17. mindspore/_akg/akg/tvm/contrib/graph_runtime.py +0 -53
  18. mindspore/_akg/akg/tvm/hybrid/parser.py +7 -6
  19. mindspore/_akg/akg/tvm/ir_builder.py +1 -1
  20. mindspore/_akg/akg/tvm/module.py +1 -2
  21. mindspore/_akg/akg/tvm/stmt.py +2 -2
  22. mindspore/_akg/akg/utils/composite_op_helper.py +9 -10
  23. mindspore/_akg/akg/utils/kernel_exec.py +58 -260
  24. mindspore/_akg/akg/utils/op_dsl.py +17 -1
  25. mindspore/_akg/akg/utils/result_analysis.py +4 -24
  26. mindspore/_akg/akg/utils/tbe_codegen_utils.py +198 -0
  27. mindspore/_c_dataengine.cpython-38-aarch64-linux-gnu.so +0 -0
  28. mindspore/_c_expression.cpython-38-aarch64-linux-gnu.so +0 -0
  29. mindspore/_c_mindrecord.cpython-38-aarch64-linux-gnu.so +0 -0
  30. mindspore/_check_jit_forbidden_api.py +5 -1
  31. mindspore/_checkparam.py +79 -62
  32. mindspore/_extends/graph_kernel/__init__.py +0 -1
  33. mindspore/_extends/graph_kernel/model/graph_split.py +2 -0
  34. mindspore/_extends/graph_kernel/model/model_builder.py +9 -50
  35. mindspore/_extends/graph_kernel/splitter.py +1 -9
  36. mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +128 -21
  37. mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +2 -2
  38. mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +4 -2
  39. mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +18 -13
  40. mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +13 -9
  41. mindspore/_extends/parallel_compile/tbe_compiler/tbe_job.py +1 -1
  42. mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -1
  43. mindspore/_extends/parse/__init__.py +19 -17
  44. mindspore/_extends/parse/namespace.py +7 -36
  45. mindspore/_extends/parse/parser.py +375 -189
  46. mindspore/_extends/parse/resources.py +36 -41
  47. mindspore/_extends/parse/standard_method.py +350 -245
  48. mindspore/_extends/parse/trope.py +2 -12
  49. mindspore/_extends/remote/kernel_build_server.py +24 -7
  50. mindspore/_extends/remote/kernel_build_server_akg_v2.py +55 -0
  51. mindspore/_install_custom.py +43 -0
  52. mindspore/_mindspore_offline_debug.cpython-38-aarch64-linux-gnu.so +0 -0
  53. mindspore/amp.py +85 -19
  54. mindspore/bin/cache_admin +0 -0
  55. mindspore/bin/cache_server +0 -0
  56. mindspore/boost/base.py +2 -2
  57. mindspore/boost/boost.py +27 -32
  58. mindspore/boost/boost_cell_wrapper.py +37 -13
  59. mindspore/boost/grad_accumulation.py +1 -1
  60. mindspore/boost/grad_freeze.py +34 -6
  61. mindspore/boost/group_loss_scale_manager.py +15 -14
  62. mindspore/boost/less_batch_normalization.py +28 -3
  63. mindspore/common/__init__.py +15 -11
  64. mindspore/common/_auto_dynamic.py +68 -0
  65. mindspore/common/_jit_fallback_utils.py +111 -0
  66. mindspore/common/_register_for_adapter.py +17 -5
  67. mindspore/common/_register_for_tensor.py +2 -2
  68. mindspore/common/_stub_tensor.py +18 -15
  69. mindspore/common/_utils.py +31 -7
  70. mindspore/common/api.py +269 -101
  71. mindspore/common/auto_dynamic_shape.py +498 -0
  72. mindspore/common/dtype.py +61 -21
  73. mindspore/common/dump.py +9 -7
  74. mindspore/common/initializer.py +106 -76
  75. mindspore/common/jit_config.py +35 -14
  76. mindspore/common/lazy_inline.py +187 -0
  77. mindspore/common/mindir_util.py +101 -0
  78. mindspore/common/mutable.py +10 -13
  79. mindspore/common/parameter.py +246 -55
  80. mindspore/common/seed.py +13 -7
  81. mindspore/common/sparse_tensor.py +29 -33
  82. mindspore/common/tensor.py +907 -251
  83. mindspore/communication/__init__.py +7 -4
  84. mindspore/communication/_comm_helper.py +84 -4
  85. mindspore/communication/management.py +160 -88
  86. mindspore/config/op_info.config +99 -75
  87. mindspore/config/super_bar_config.json +36 -4
  88. mindspore/context.py +526 -219
  89. mindspore/dataset/__init__.py +9 -46
  90. mindspore/dataset/audio/__init__.py +4 -19
  91. mindspore/dataset/audio/transforms.py +545 -233
  92. mindspore/dataset/audio/utils.py +21 -18
  93. mindspore/dataset/callback/ds_callback.py +42 -13
  94. mindspore/dataset/core/config.py +158 -100
  95. mindspore/dataset/core/validator_helpers.py +1 -63
  96. mindspore/dataset/debug/debug_hook.py +45 -13
  97. mindspore/dataset/debug/pre_defined_hook.py +5 -5
  98. mindspore/dataset/engine/__init__.py +0 -5
  99. mindspore/dataset/engine/cache_client.py +38 -15
  100. mindspore/dataset/engine/datasets.py +615 -278
  101. mindspore/dataset/engine/datasets_audio.py +154 -283
  102. mindspore/dataset/engine/datasets_standard_format.py +104 -116
  103. mindspore/dataset/engine/datasets_text.py +443 -326
  104. mindspore/dataset/engine/datasets_user_defined.py +251 -164
  105. mindspore/dataset/engine/datasets_vision.py +839 -1443
  106. mindspore/dataset/engine/iterators.py +11 -4
  107. mindspore/dataset/engine/obs/obs_mindrecord_dataset.py +7 -3
  108. mindspore/dataset/engine/obs/util.py +3 -0
  109. mindspore/dataset/engine/offload.py +6 -6
  110. mindspore/dataset/engine/queue.py +15 -14
  111. mindspore/dataset/engine/samplers.py +39 -23
  112. mindspore/dataset/engine/serializer_deserializer.py +22 -6
  113. mindspore/dataset/engine/validators.py +21 -331
  114. mindspore/dataset/text/__init__.py +5 -33
  115. mindspore/dataset/text/transforms.py +334 -165
  116. mindspore/dataset/text/utils.py +215 -145
  117. mindspore/dataset/transforms/__init__.py +1 -1
  118. mindspore/dataset/transforms/c_transforms.py +3 -2
  119. mindspore/dataset/transforms/py_transforms_util.py +40 -12
  120. mindspore/dataset/transforms/transforms.py +174 -71
  121. mindspore/dataset/utils/browse_dataset.py +25 -17
  122. mindspore/dataset/utils/line_reader.py +24 -21
  123. mindspore/dataset/vision/__init__.py +5 -26
  124. mindspore/dataset/vision/c_transforms.py +177 -165
  125. mindspore/dataset/vision/py_transforms.py +114 -119
  126. mindspore/dataset/vision/py_transforms_util.py +54 -51
  127. mindspore/dataset/vision/transforms.py +1127 -381
  128. mindspore/dataset/vision/utils.py +54 -38
  129. mindspore/dataset/vision/validators.py +12 -2
  130. mindspore/experimental/map_parameter.py +38 -4
  131. mindspore/{dataset/datapreprocess → experimental/optim}/__init__.py +14 -4
  132. mindspore/experimental/optim/adam.py +192 -0
  133. mindspore/experimental/optim/adamw.py +181 -0
  134. mindspore/experimental/optim/lr_scheduler.py +1427 -0
  135. mindspore/experimental/optim/optimizer.py +252 -0
  136. mindspore/experimental/optim/sgd.py +147 -0
  137. mindspore/gen_ops.py +273 -0
  138. mindspore/include/OWNERS +1 -2
  139. mindspore/include/api/context.h +21 -1
  140. mindspore/include/api/data_type.h +2 -1
  141. mindspore/include/api/graph.h +0 -15
  142. mindspore/include/api/kernel.h +2 -0
  143. mindspore/include/api/kernel_api.h +37 -12
  144. mindspore/include/api/model.h +29 -42
  145. mindspore/include/api/model_group.h +14 -3
  146. mindspore/include/api/model_parallel_runner.h +18 -2
  147. mindspore/include/api/serialization.h +26 -0
  148. mindspore/include/api/status.h +1 -0
  149. mindspore/include/api/types.h +38 -4
  150. mindspore/include/c_api/ms/abstract.h +67 -0
  151. mindspore/include/c_api/ms/attribute.h +197 -0
  152. mindspore/include/c_api/ms/base/handle_types.h +43 -0
  153. mindspore/include/c_api/ms/base/macros.h +32 -0
  154. mindspore/include/c_api/ms/base/status.h +33 -0
  155. mindspore/include/c_api/ms/base/types.h +282 -0
  156. mindspore/include/c_api/ms/context.h +102 -0
  157. mindspore/include/c_api/ms/graph.h +160 -0
  158. mindspore/include/c_api/ms/node.h +606 -0
  159. mindspore/include/c_api/ms/tensor.h +161 -0
  160. mindspore/include/c_api/ms/value.h +84 -0
  161. mindspore/include/c_api/status_c.h +3 -0
  162. mindspore/include/dataset/constants.h +6 -12
  163. mindspore/include/dataset/execute.h +23 -13
  164. mindspore/include/dataset/text.h +26 -26
  165. mindspore/include/dataset/transforms.h +25 -31
  166. mindspore/include/dataset/vision.h +60 -60
  167. mindspore/include/dataset/vision_ascend.h +5 -6
  168. mindspore/include/dataset/vision_lite.h +17 -17
  169. mindspore/include/mindapi/base/format.h +0 -1
  170. mindspore/include/mindapi/base/type_id.h +2 -1
  171. mindspore/include/mindapi/base/types.h +5 -1
  172. mindspore/lib/libdnnl.so.2 +0 -0
  173. mindspore/lib/libjemalloc.so.2 +0 -0
  174. mindspore/lib/libmindspore.so +0 -0
  175. mindspore/lib/libmindspore_backend.so +0 -0
  176. mindspore/lib/libmindspore_common.so +0 -0
  177. mindspore/lib/libmindspore_core.so +0 -0
  178. mindspore/lib/libmindspore_glog.so.0 +0 -0
  179. mindspore/lib/libmindspore_gpr.so.15 +0 -0
  180. mindspore/lib/libmindspore_grpc++.so.1 +0 -0
  181. mindspore/lib/libmindspore_grpc.so.15 +0 -0
  182. mindspore/lib/libmindspore_shared_lib.so +0 -0
  183. mindspore/lib/libmpi_adapter.so +0 -0
  184. mindspore/lib/libnnacl.so +0 -0
  185. mindspore/lib/libopencv_core.so.4.5 +0 -0
  186. mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
  187. mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
  188. mindspore/lib/libps_cache.so +0 -0
  189. mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_aicpu_kernels.so +0 -0
  190. mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_cpu_kernels.so +0 -0
  191. mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/config/cust_aicpu_kernel.json +9000 -0
  192. mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.so +0 -0
  193. mindspore/lib/plugin/ascend/libakg.so +0 -0
  194. mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
  195. mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
  196. mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
  197. mindspore/lib/plugin/ascend/libmindspore_aicpu_kernels.so +0 -0
  198. mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
  199. mindspore/lib/plugin/cpu/libakg.so +0 -0
  200. mindspore/lib/plugin/libmindspore_ascend.so.1 +0 -0
  201. mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
  202. mindspore/log.py +9 -6
  203. mindspore/mindrecord/filereader.py +33 -4
  204. mindspore/mindrecord/filewriter.py +70 -35
  205. mindspore/mindrecord/mindpage.py +40 -34
  206. mindspore/mindrecord/shardreader.py +1 -1
  207. mindspore/mindrecord/shardsegment.py +1 -1
  208. mindspore/mindrecord/tools/cifar100_to_mr.py +25 -18
  209. mindspore/mindrecord/tools/cifar10_to_mr.py +25 -18
  210. mindspore/mindrecord/tools/csv_to_mr.py +29 -13
  211. mindspore/mindrecord/tools/imagenet_to_mr.py +24 -10
  212. mindspore/mindrecord/tools/mnist_to_mr.py +24 -11
  213. mindspore/mindrecord/tools/tfrecord_to_mr.py +31 -26
  214. mindspore/nn/cell.py +463 -169
  215. mindspore/nn/dynamic_lr.py +47 -43
  216. mindspore/nn/layer/activation.py +225 -82
  217. mindspore/nn/layer/basic.py +121 -79
  218. mindspore/nn/layer/channel_shuffle.py +21 -21
  219. mindspore/nn/layer/combined.py +33 -26
  220. mindspore/nn/layer/container.py +277 -22
  221. mindspore/nn/layer/conv.py +441 -304
  222. mindspore/nn/layer/dense.py +19 -13
  223. mindspore/nn/layer/embedding.py +62 -49
  224. mindspore/nn/layer/flash_attention.py +264 -0
  225. mindspore/nn/layer/image.py +50 -39
  226. mindspore/nn/layer/math.py +62 -51
  227. mindspore/nn/layer/normalization.py +219 -167
  228. mindspore/nn/layer/padding.py +58 -70
  229. mindspore/nn/layer/pooling.py +334 -287
  230. mindspore/nn/layer/rnn_cells.py +53 -38
  231. mindspore/nn/layer/rnns.py +59 -56
  232. mindspore/nn/layer/thor_layer.py +52 -44
  233. mindspore/nn/layer/timedistributed.py +6 -4
  234. mindspore/nn/layer/transformer.py +284 -164
  235. mindspore/nn/learning_rate_schedule.py +34 -25
  236. mindspore/nn/loss/__init__.py +3 -2
  237. mindspore/nn/loss/loss.py +554 -311
  238. mindspore/nn/optim/ada_grad.py +12 -9
  239. mindspore/nn/optim/adadelta.py +14 -11
  240. mindspore/nn/optim/adafactor.py +19 -16
  241. mindspore/nn/optim/adam.py +62 -47
  242. mindspore/nn/optim/adamax.py +13 -10
  243. mindspore/nn/optim/adasum.py +12 -8
  244. mindspore/nn/optim/asgd.py +10 -9
  245. mindspore/nn/optim/ftrl.py +20 -17
  246. mindspore/nn/optim/lamb.py +16 -12
  247. mindspore/nn/optim/lars.py +8 -6
  248. mindspore/nn/optim/lazyadam.py +25 -20
  249. mindspore/nn/optim/momentum.py +10 -7
  250. mindspore/nn/optim/optimizer.py +61 -9
  251. mindspore/nn/optim/proximal_ada_grad.py +14 -13
  252. mindspore/nn/optim/rmsprop.py +17 -13
  253. mindspore/nn/optim/rprop.py +30 -17
  254. mindspore/nn/optim/sgd.py +40 -23
  255. mindspore/nn/optim/thor.py +24 -26
  256. mindspore/nn/probability/bijector/bijector.py +11 -11
  257. mindspore/nn/probability/bijector/exp.py +1 -1
  258. mindspore/nn/probability/bijector/gumbel_cdf.py +3 -3
  259. mindspore/nn/probability/bijector/invert.py +1 -1
  260. mindspore/nn/probability/bijector/power_transform.py +29 -29
  261. mindspore/nn/probability/bijector/scalar_affine.py +3 -3
  262. mindspore/nn/probability/bijector/softplus.py +5 -5
  263. mindspore/nn/probability/bnn_layers/bnn_cell_wrapper.py +4 -2
  264. mindspore/nn/probability/bnn_layers/conv_variational.py +13 -13
  265. mindspore/nn/probability/bnn_layers/dense_variational.py +12 -12
  266. mindspore/nn/probability/bnn_layers/layer_distribution.py +9 -8
  267. mindspore/nn/probability/distribution/_utils/custom_ops.py +19 -3
  268. mindspore/nn/probability/distribution/_utils/utils.py +1 -1
  269. mindspore/nn/probability/distribution/bernoulli.py +9 -9
  270. mindspore/nn/probability/distribution/beta.py +8 -8
  271. mindspore/nn/probability/distribution/categorical.py +23 -15
  272. mindspore/nn/probability/distribution/cauchy.py +5 -6
  273. mindspore/nn/probability/distribution/distribution.py +3 -3
  274. mindspore/nn/probability/distribution/exponential.py +4 -4
  275. mindspore/nn/probability/distribution/gamma.py +10 -10
  276. mindspore/nn/probability/distribution/geometric.py +8 -8
  277. mindspore/nn/probability/distribution/gumbel.py +8 -9
  278. mindspore/nn/probability/distribution/half_normal.py +5 -5
  279. mindspore/nn/probability/distribution/laplace.py +5 -5
  280. mindspore/nn/probability/distribution/log_normal.py +12 -11
  281. mindspore/nn/probability/distribution/logistic.py +8 -8
  282. mindspore/nn/probability/distribution/normal.py +6 -5
  283. mindspore/nn/probability/distribution/poisson.py +10 -11
  284. mindspore/nn/probability/distribution/student_t.py +8 -9
  285. mindspore/nn/probability/distribution/transformed_distribution.py +5 -5
  286. mindspore/nn/probability/distribution/uniform.py +11 -11
  287. mindspore/nn/reinforcement/tensor_array.py +2 -2
  288. mindspore/nn/sparse/sparse.py +9 -9
  289. mindspore/nn/wrap/cell_wrapper.py +188 -63
  290. mindspore/nn/wrap/grad_reducer.py +21 -12
  291. mindspore/nn/wrap/loss_scale.py +136 -49
  292. mindspore/numpy/__init__.py +4 -4
  293. mindspore/numpy/array_creations.py +55 -56
  294. mindspore/numpy/array_ops.py +134 -35
  295. mindspore/numpy/logic_ops.py +66 -20
  296. mindspore/numpy/math_ops.py +142 -139
  297. mindspore/numpy/utils_const.py +2 -2
  298. mindspore/offline_debug/convert_async.py +2 -2
  299. mindspore/ops/_grad_experimental/__init__.py +7 -5
  300. mindspore/ops/_grad_experimental/grad_array_ops.py +231 -348
  301. mindspore/ops/{_grad → _grad_experimental}/grad_base.py +1 -33
  302. mindspore/ops/{_grad → _grad_experimental}/grad_comm_ops.py +25 -13
  303. mindspore/ops/{_grad/__init__.py → _grad_experimental/grad_debug_ops.py} +15 -7
  304. mindspore/ops/{_grad → _grad_experimental}/grad_implementations.py +17 -11
  305. mindspore/ops/_grad_experimental/grad_inner_ops.py +33 -52
  306. mindspore/ops/_grad_experimental/grad_math_ops.py +151 -1224
  307. mindspore/ops/_grad_experimental/grad_nn_ops.py +141 -414
  308. mindspore/ops/{_grad → _grad_experimental}/grad_quant_ops.py +10 -6
  309. mindspore/ops/_grad_experimental/grad_sparse.py +317 -2
  310. mindspore/ops/_grad_experimental/grad_sparse_ops.py +3 -13
  311. mindspore/ops/{_grad → _grad_experimental}/taylor_rule.py +1 -1
  312. mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +1 -1
  313. mindspore/ops/_op_impl/_custom_op/flash_attention/__init__.py +0 -0
  314. mindspore/ops/_op_impl/_custom_op/flash_attention/attention.py +406 -0
  315. mindspore/{_extends/graph_kernel/expanders/complex/__init__.py → ops/_op_impl/_custom_op/flash_attention/constants.py} +27 -8
  316. mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_bwd.py +467 -0
  317. mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_fwd.py +563 -0
  318. mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_impl.py +193 -0
  319. mindspore/ops/_op_impl/_custom_op/flash_attention/tik_ops_utils.py +435 -0
  320. mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/__init__.py +0 -0
  321. mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/sparse_tiling.py +45 -0
  322. mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/strategy.py +67 -0
  323. mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/wukong_tiling.py +62 -0
  324. mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
  325. mindspore/ops/_op_impl/aicpu/__init__.py +41 -1
  326. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_2d.py +37 -0
  327. mindspore/ops/_op_impl/aicpu/bias_add_grad.py +0 -1
  328. mindspore/ops/_op_impl/aicpu/cast.py +52 -0
  329. mindspore/ops/_op_impl/aicpu/coalesce.py +2 -0
  330. mindspore/ops/_op_impl/aicpu/col2im.py +3 -1
  331. mindspore/ops/_op_impl/aicpu/count_nonzero.py +43 -0
  332. mindspore/ops/_op_impl/aicpu/dropout_genmask.py +6 -0
  333. mindspore/ops/_op_impl/aicpu/eps.py +32 -0
  334. mindspore/ops/_op_impl/aicpu/eye.py +4 -4
  335. mindspore/ops/_op_impl/aicpu/fft_with_size.py +6 -0
  336. mindspore/ops/_op_impl/aicpu/fill_diagonal.py +5 -0
  337. mindspore/ops/_op_impl/aicpu/gamma.py +2 -2
  338. mindspore/ops/_op_impl/aicpu/im2col.py +3 -5
  339. mindspore/ops/_op_impl/aicpu/lgamma.py +1 -0
  340. mindspore/ops/_op_impl/aicpu/log_uniform_candidate_sampler.py +6 -3
  341. mindspore/ops/_op_impl/aicpu/lu.py +39 -0
  342. mindspore/ops/_op_impl/aicpu/lu_unpack_grad.py +0 -1
  343. mindspore/ops/_op_impl/aicpu/masked_scatter.py +1 -0
  344. mindspore/ops/_op_impl/aicpu/masked_select_grad.py +3 -0
  345. mindspore/ops/_op_impl/aicpu/matrix_band_part.py +59 -0
  346. mindspore/ops/_op_impl/aicpu/matrix_power.py +6 -1
  347. mindspore/ops/_op_impl/aicpu/median.py +1 -0
  348. mindspore/ops/_op_impl/aicpu/multinomial.py +9 -9
  349. mindspore/ops/_op_impl/aicpu/not_equal.py +0 -5
  350. mindspore/ops/_op_impl/aicpu/pad_v3.py +3 -1
  351. mindspore/ops/_op_impl/aicpu/pad_v3_grad.py +2 -0
  352. mindspore/ops/_op_impl/aicpu/parameterized_truncated_normal.py +15 -7
  353. mindspore/ops/_op_impl/aicpu/random_categorical.py +39 -19
  354. mindspore/ops/_op_impl/aicpu/random_choice_with_mask.py +5 -2
  355. mindspore/ops/_op_impl/aicpu/random_poisson.py +103 -52
  356. mindspore/ops/_op_impl/aicpu/random_shuffle.py +17 -15
  357. mindspore/ops/_op_impl/aicpu/resize_bilinear_grad.py +0 -1
  358. mindspore/ops/_op_impl/aicpu/resize_nearest_neighbor_v2.py +0 -6
  359. mindspore/ops/_op_impl/aicpu/resize_nearest_neighbor_v2_grad.py +0 -7
  360. mindspore/ops/_op_impl/aicpu/scatter_nd.py +2 -0
  361. mindspore/ops/_op_impl/aicpu/sequence_concat.py +40 -0
  362. mindspore/ops/_op_impl/aicpu/sequence_stack.py +40 -0
  363. mindspore/ops/_op_impl/aicpu/{sparseaddmm.py → sparse_addmm.py} +2 -2
  364. mindspore/ops/_op_impl/aicpu/{sparsesparsemaximum.py → sparse_sparse_maximum.py} +4 -4
  365. mindspore/ops/_op_impl/aicpu/standard_laplace.py +5 -4
  366. mindspore/ops/_op_impl/aicpu/standard_normal.py +5 -4
  367. mindspore/ops/_op_impl/aicpu/truncated_normal.py +9 -7
  368. mindspore/ops/_op_impl/aicpu/uniform.py +5 -3
  369. mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +8 -4
  370. mindspore/ops/_op_impl/aicpu/uniform_int.py +5 -5
  371. mindspore/ops/_op_impl/aicpu/uniform_real.py +4 -4
  372. mindspore/ops/_op_impl/aicpu/upsample_nearest_3d.py +14 -6
  373. mindspore/ops/_op_impl/aicpu/upsample_nearest_3d_grad.py +22 -8
  374. mindspore/ops/_op_impl/aicpu/upsample_trilinear_3d.py +11 -6
  375. mindspore/ops/_op_impl/aicpu/upsample_trilinear_3d_grad.py +21 -10
  376. mindspore/ops/_op_impl/tbe/__init__.py +6 -4
  377. mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
  378. mindspore/ops/_op_impl/tbe/avg_pool.py +2 -2
  379. mindspore/ops/_op_impl/tbe/avg_pool_3d.py +3 -3
  380. mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +4 -4
  381. mindspore/ops/_op_impl/tbe/avg_pool_ds.py +2 -2
  382. mindspore/ops/_op_impl/tbe/avg_pool_grad.py +3 -3
  383. mindspore/ops/_op_impl/tbe/avg_pool_grad_vm.py +3 -3
  384. mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
  385. mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +2 -2
  386. mindspore/ops/_op_impl/tbe/bn_infer.py +2 -2
  387. mindspore/ops/_op_impl/tbe/bn_infer_ds.py +3 -2
  388. mindspore/ops/_op_impl/tbe/broadcast_to.py +1 -1
  389. mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +3 -3
  390. mindspore/ops/_op_impl/tbe/expand_dims.py +1 -1
  391. mindspore/ops/_op_impl/tbe/gather_v2.py +56 -0
  392. mindspore/ops/_op_impl/tbe/im2col.py +4 -4
  393. mindspore/ops/_op_impl/tbe/inplace_index_add.py +7 -3
  394. mindspore/ops/_op_impl/tbe/mem_set.py +38 -0
  395. mindspore/ops/_op_impl/tbe/scatter_nd_add.py +3 -0
  396. mindspore/ops/_op_impl/tbe/scatter_nd_d.py +1 -1
  397. mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
  398. mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +2 -2
  399. mindspore/ops/_op_impl/tbe/trans_data_ds.py +2 -0
  400. mindspore/ops/_primitive_cache.py +1 -1
  401. mindspore/ops/_tracefunc.py +241 -0
  402. mindspore/ops/_utils/utils.py +10 -2
  403. mindspore/ops/_vmap/vmap_array_ops.py +5 -3
  404. mindspore/ops/_vmap/vmap_base.py +5 -4
  405. mindspore/ops/_vmap/vmap_convolution_ops.py +1 -1
  406. mindspore/ops/_vmap/vmap_grad_math_ops.py +6 -4
  407. mindspore/ops/_vmap/vmap_grad_nn_ops.py +11 -6
  408. mindspore/ops/_vmap/vmap_math_ops.py +5 -2
  409. mindspore/ops/_vmap/vmap_nn_ops.py +135 -11
  410. mindspore/ops/arg_dtype_cast.py +54 -0
  411. mindspore/ops/composite/__init__.py +7 -5
  412. mindspore/ops/composite/base.py +78 -34
  413. mindspore/ops/composite/math_ops.py +5 -695
  414. mindspore/ops/composite/multitype_ops/_compile_utils.py +403 -97
  415. mindspore/ops/composite/multitype_ops/_constexpr_utils.py +28 -22
  416. mindspore/ops/composite/multitype_ops/add_impl.py +69 -7
  417. mindspore/ops/composite/multitype_ops/bitwise_and_impl.py +2 -1
  418. mindspore/ops/composite/multitype_ops/bitwise_or_impl.py +2 -1
  419. mindspore/ops/composite/multitype_ops/bitwise_xor_impl.py +2 -0
  420. mindspore/ops/composite/multitype_ops/div_impl.py +1 -0
  421. mindspore/ops/composite/multitype_ops/floordiv_impl.py +1 -0
  422. mindspore/ops/composite/multitype_ops/getitem_impl.py +48 -10
  423. mindspore/ops/composite/multitype_ops/greater_equal_impl.py +2 -0
  424. mindspore/ops/composite/multitype_ops/greater_impl.py +2 -0
  425. mindspore/ops/composite/multitype_ops/left_shift_impl.py +2 -0
  426. mindspore/ops/composite/multitype_ops/less_equal_impl.py +2 -0
  427. mindspore/ops/composite/multitype_ops/less_impl.py +2 -0
  428. mindspore/ops/composite/multitype_ops/logic_not_impl.py +2 -2
  429. mindspore/ops/composite/multitype_ops/mod_impl.py +1 -0
  430. mindspore/ops/composite/multitype_ops/mul_impl.py +1 -0
  431. mindspore/ops/composite/multitype_ops/negative_impl.py +1 -0
  432. mindspore/ops/composite/multitype_ops/not_in_impl.py +1 -0
  433. mindspore/ops/composite/multitype_ops/ones_like_impl.py +6 -0
  434. mindspore/ops/composite/multitype_ops/pow_impl.py +1 -0
  435. mindspore/ops/composite/multitype_ops/right_shift_impl.py +2 -0
  436. mindspore/ops/composite/multitype_ops/setitem_impl.py +10 -7
  437. mindspore/ops/composite/multitype_ops/sub_impl.py +1 -0
  438. mindspore/ops/composite/multitype_ops/uadd_impl.py +2 -0
  439. mindspore/ops/composite/multitype_ops/zeros_like_impl.py +9 -0
  440. mindspore/ops/deprecated.py +304 -0
  441. mindspore/ops/function/__init__.py +41 -4
  442. mindspore/ops/function/array_func.py +1108 -467
  443. mindspore/ops/function/clip_func.py +94 -27
  444. mindspore/ops/function/debug_func.py +3 -1
  445. mindspore/ops/function/grad/grad_func.py +82 -73
  446. mindspore/ops/function/image_func.py +28 -12
  447. mindspore/ops/function/linalg_func.py +135 -39
  448. mindspore/ops/function/math_func.py +3779 -894
  449. mindspore/ops/function/nn_func.py +1584 -657
  450. mindspore/ops/function/parameter_func.py +13 -3
  451. mindspore/ops/function/random_func.py +247 -153
  452. mindspore/ops/function/sparse_func.py +14 -11
  453. mindspore/ops/function/sparse_unary_func.py +173 -47
  454. mindspore/ops/function/spectral_func.py +8 -4
  455. mindspore/ops/function/vmap_func.py +8 -7
  456. mindspore/ops/functional.py +47 -16
  457. mindspore/ops/op_info_register.py +346 -86
  458. mindspore/ops/operations/__init__.py +38 -22
  459. mindspore/ops/operations/_grad_ops.py +145 -149
  460. mindspore/ops/operations/_inner_ops.py +298 -56
  461. mindspore/ops/operations/_ms_kernel.py +3 -3
  462. mindspore/ops/operations/_quant_ops.py +24 -28
  463. mindspore/ops/operations/_rl_inner_ops.py +9 -7
  464. mindspore/ops/operations/_scalar_ops.py +115 -0
  465. mindspore/ops/operations/_sequence_ops.py +148 -10
  466. mindspore/ops/operations/_tensor_array.py +1 -1
  467. mindspore/ops/operations/_thor_ops.py +2 -2
  468. mindspore/ops/operations/array_ops.py +1239 -561
  469. mindspore/ops/operations/comm_ops.py +166 -90
  470. mindspore/ops/operations/control_ops.py +3 -3
  471. mindspore/ops/operations/custom_ops.py +124 -102
  472. mindspore/ops/operations/debug_ops.py +24 -11
  473. mindspore/ops/operations/image_ops.py +86 -71
  474. mindspore/ops/operations/inner_ops.py +18 -13
  475. mindspore/ops/operations/linalg_ops.py +30 -11
  476. mindspore/ops/operations/math_ops.py +1730 -435
  477. mindspore/ops/operations/nn_ops.py +1953 -943
  478. mindspore/ops/operations/other_ops.py +65 -43
  479. mindspore/ops/operations/random_ops.py +258 -98
  480. mindspore/ops/operations/rl_ops.py +4 -36
  481. mindspore/ops/operations/sparse_ops.py +38 -33
  482. mindspore/ops/operations/spectral_ops.py +8 -4
  483. mindspore/ops/primitive.py +66 -44
  484. mindspore/ops/signature.py +5 -5
  485. mindspore/parallel/_auto_parallel_context.py +80 -19
  486. mindspore/parallel/_cost_model_context.py +42 -0
  487. mindspore/parallel/_offload_context.py +162 -72
  488. mindspore/parallel/_parallel_serialization.py +2 -2
  489. mindspore/parallel/_ps_context.py +16 -4
  490. mindspore/parallel/_recovery_context.py +2 -1
  491. mindspore/parallel/_tensor.py +15 -13
  492. mindspore/parallel/_transformer/layers.py +8 -6
  493. mindspore/parallel/_transformer/loss.py +1 -0
  494. mindspore/parallel/_transformer/moe.py +7 -7
  495. mindspore/parallel/_transformer/op_parallel_config.py +12 -1
  496. mindspore/parallel/_transformer/transformer.py +34 -14
  497. mindspore/parallel/_utils.py +36 -14
  498. mindspore/parallel/algo_parameter_config.py +114 -20
  499. mindspore/parallel/checkpoint_transform.py +16 -18
  500. mindspore/parallel/shard.py +16 -13
  501. mindspore/profiler/__init__.py +1 -1
  502. mindspore/profiler/common/struct_type.py +3 -3
  503. mindspore/profiler/common/util.py +3 -2
  504. mindspore/profiler/envprofiling.py +11 -4
  505. mindspore/profiler/parser/aicpu_data_parser.py +5 -3
  506. mindspore/profiler/parser/ascend_flops_generator.py +94 -0
  507. mindspore/profiler/parser/ascend_fpbp_generator.py +76 -0
  508. mindspore/profiler/parser/ascend_hccl_generator.py +288 -0
  509. mindspore/profiler/parser/ascend_msprof_exporter.py +213 -0
  510. mindspore/profiler/parser/ascend_msprof_generator.py +199 -0
  511. mindspore/profiler/parser/ascend_op_generator.py +276 -0
  512. mindspore/profiler/parser/ascend_steptrace_generator.py +94 -0
  513. mindspore/profiler/parser/ascend_timeline_generator.py +110 -54
  514. mindspore/profiler/parser/base_timeline_generator.py +11 -7
  515. mindspore/profiler/parser/cpu_gpu_timeline_generator.py +45 -46
  516. mindspore/profiler/parser/flops_parser.py +15 -11
  517. mindspore/profiler/parser/framework_parser.py +92 -73
  518. mindspore/profiler/parser/hccl_parser.py +16 -12
  519. mindspore/profiler/parser/integrator.py +22 -11
  520. mindspore/profiler/parser/memory_usage_parser.py +36 -11
  521. mindspore/profiler/parser/minddata_analyzer.py +12 -14
  522. mindspore/profiler/parser/minddata_pipeline_parser.py +1 -1
  523. mindspore/profiler/parser/msadvisor_parser.py +8 -4
  524. mindspore/profiler/parser/op_intermediate_parser.py +5 -2
  525. mindspore/profiler/parser/optime_parser.py +1 -1
  526. mindspore/profiler/parser/profiler_info.py +4 -5
  527. mindspore/profiler/parser/step_trace_parser.py +11 -14
  528. mindspore/profiler/profiling.py +678 -377
  529. mindspore/rewrite/api/node.py +211 -54
  530. mindspore/rewrite/api/node_type.py +5 -0
  531. mindspore/rewrite/api/pattern_engine.py +22 -23
  532. mindspore/rewrite/api/scoped_value.py +20 -17
  533. mindspore/rewrite/api/symbol_tree.py +252 -106
  534. mindspore/rewrite/api/tree_node_helper.py +3 -0
  535. mindspore/rewrite/ast_helpers/__init__.py +2 -1
  536. mindspore/rewrite/ast_helpers/ast_finder.py +129 -0
  537. mindspore/rewrite/ast_helpers/ast_modifier.py +116 -104
  538. mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +97 -46
  539. mindspore/rewrite/common/rewrite_elog.py +5 -1
  540. mindspore/rewrite/namer.py +51 -51
  541. mindspore/rewrite/namespace.py +14 -5
  542. mindspore/{ops/bprop_mindir → rewrite/node}/__init__.py +9 -4
  543. mindspore/rewrite/node/call_function.py +79 -0
  544. mindspore/rewrite/node/cell_container.py +135 -0
  545. mindspore/rewrite/node/control_flow.py +88 -0
  546. mindspore/rewrite/{node.py → node/node.py} +313 -247
  547. mindspore/rewrite/node/node_manager.py +254 -0
  548. mindspore/rewrite/node/node_topological_manager.py +243 -0
  549. mindspore/rewrite/parsers/arguments_parser.py +22 -21
  550. mindspore/rewrite/parsers/assign_parser.py +225 -239
  551. mindspore/rewrite/parsers/attribute_parser.py +9 -7
  552. mindspore/rewrite/parsers/class_def_parser.py +179 -218
  553. mindspore/rewrite/parsers/constant_parser.py +9 -6
  554. mindspore/rewrite/parsers/container_parser.py +9 -7
  555. mindspore/rewrite/parsers/for_parser.py +36 -15
  556. mindspore/rewrite/parsers/function_def_parser.py +23 -20
  557. mindspore/rewrite/parsers/if_parser.py +28 -24
  558. mindspore/rewrite/parsers/module_parser.py +202 -25
  559. mindspore/rewrite/{parser.py → parsers/parser.py} +4 -2
  560. mindspore/rewrite/{parser_register.py → parsers/parser_register.py} +1 -1
  561. mindspore/rewrite/parsers/return_parser.py +6 -6
  562. mindspore/rewrite/sparsify/sparse_transformer.py +12 -3
  563. mindspore/rewrite/sparsify/sparsify.py +4 -1
  564. mindspore/rewrite/sparsify/utils.py +11 -5
  565. mindspore/rewrite/symbol_tree.py +577 -732
  566. mindspore/rewrite/symbol_tree_builder.py +9 -175
  567. mindspore/rewrite/symbol_tree_dumper.py +2 -2
  568. mindspore/run_check/_check_version.py +46 -39
  569. mindspore/run_check/run_check.py +3 -2
  570. mindspore/{scipy/sparse → safeguard}/__init__.py +4 -5
  571. mindspore/safeguard/rewrite_obfuscation.py +517 -0
  572. mindspore/scipy/__init__.py +1 -1
  573. mindspore/scipy/linalg.py +67 -61
  574. mindspore/scipy/ops.py +5 -41
  575. mindspore/scipy/ops_grad.py +3 -2
  576. mindspore/scipy/ops_wrapper.py +5 -5
  577. mindspore/scipy/optimize/line_search.py +8 -8
  578. mindspore/scipy/optimize/linear_sum_assignment.py +4 -4
  579. mindspore/scipy/optimize/minimize.py +16 -12
  580. mindspore/scipy/utils.py +1 -52
  581. mindspore/scipy/utils_const.py +4 -4
  582. mindspore/train/__init__.py +4 -4
  583. mindspore/train/_utils.py +13 -5
  584. mindspore/train/amp.py +410 -148
  585. mindspore/train/anf_ir_pb2.py +16 -4
  586. mindspore/train/callback/_backup_and_restore.py +8 -11
  587. mindspore/train/callback/_callback.py +80 -3
  588. mindspore/train/callback/_checkpoint.py +82 -51
  589. mindspore/train/callback/_early_stop.py +12 -15
  590. mindspore/train/callback/_history.py +1 -1
  591. mindspore/train/callback/_lambda_callback.py +13 -13
  592. mindspore/train/callback/_landscape.py +21 -17
  593. mindspore/train/callback/_loss_monitor.py +9 -10
  594. mindspore/train/callback/_on_request_exit.py +16 -33
  595. mindspore/train/callback/_reduce_lr_on_plateau.py +21 -24
  596. mindspore/train/callback/_summary_collector.py +44 -30
  597. mindspore/train/callback/_time_monitor.py +62 -12
  598. mindspore/train/data_sink.py +10 -16
  599. mindspore/train/dataset_helper.py +154 -86
  600. mindspore/train/loss_scale_manager.py +14 -9
  601. mindspore/train/metrics/__init__.py +10 -2
  602. mindspore/train/metrics/accuracy.py +1 -1
  603. mindspore/train/metrics/auc.py +1 -1
  604. mindspore/train/metrics/bleu_score.py +2 -2
  605. mindspore/train/metrics/confusion_matrix.py +14 -14
  606. mindspore/train/metrics/cosine_similarity.py +3 -3
  607. mindspore/train/metrics/dice.py +1 -1
  608. mindspore/train/metrics/fbeta.py +1 -1
  609. mindspore/train/metrics/hausdorff_distance.py +8 -6
  610. mindspore/train/metrics/mean_surface_distance.py +5 -4
  611. mindspore/train/metrics/metric.py +49 -17
  612. mindspore/train/metrics/occlusion_sensitivity.py +4 -4
  613. mindspore/train/metrics/perplexity.py +1 -1
  614. mindspore/train/metrics/precision.py +2 -2
  615. mindspore/train/metrics/recall.py +2 -3
  616. mindspore/train/metrics/roc.py +7 -7
  617. mindspore/train/metrics/root_mean_square_surface_distance.py +5 -4
  618. mindspore/train/metrics/topk.py +7 -4
  619. mindspore/train/mind_ir_pb2.py +193 -48
  620. mindspore/train/model.py +377 -133
  621. mindspore/train/serialization.py +697 -245
  622. mindspore/train/summary/_summary_adapter.py +5 -2
  623. mindspore/train/summary/_writer_pool.py +4 -3
  624. mindspore/train/summary/summary_record.py +25 -23
  625. mindspore/train/train_thor/convert_utils.py +39 -23
  626. mindspore/train/train_thor/dataset_helper.py +4 -3
  627. mindspore/train/train_thor/model_thor.py +8 -8
  628. mindspore/version.py +1 -1
  629. {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/METADATA +7 -8
  630. {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/RECORD +633 -804
  631. {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/entry_points.txt +0 -1
  632. mindspore/_akg/akg/tvm/contrib/debugger/__init__.py +0 -16
  633. mindspore/_akg/akg/tvm/contrib/debugger/debug_result.py +0 -274
  634. mindspore/_akg/akg/tvm/contrib/debugger/debug_runtime.py +0 -259
  635. mindspore/_akg/akg/tvm/contrib/peak.py +0 -341
  636. mindspore/_akg/akg/tvm/contrib/rpc.py +0 -25
  637. mindspore/_akg/akg/tvm/contrib/xcode.py +0 -257
  638. mindspore/_akg/akg/tvm/exec/__init__.py +0 -17
  639. mindspore/_akg/akg/tvm/exec/autotvm_log_editor.py +0 -60
  640. mindspore/_akg/akg/tvm/exec/measure_peak.py +0 -48
  641. mindspore/_akg/akg/tvm/exec/query_rpc_tracker.py +0 -48
  642. mindspore/_akg/akg/tvm/exec/rpc_proxy.py +0 -98
  643. mindspore/_akg/akg/tvm/exec/rpc_server.py +0 -88
  644. mindspore/_akg/akg/tvm/exec/rpc_tracker.py +0 -62
  645. mindspore/_akg/akg/tvm/rpc/__init__.py +0 -29
  646. mindspore/_akg/akg/tvm/rpc/base.py +0 -182
  647. mindspore/_akg/akg/tvm/rpc/client.py +0 -436
  648. mindspore/_akg/akg/tvm/rpc/proxy.py +0 -595
  649. mindspore/_akg/akg/tvm/rpc/server.py +0 -413
  650. mindspore/_akg/akg/tvm/rpc/tornado_util.py +0 -121
  651. mindspore/_akg/akg/tvm/rpc/tracker.py +0 -431
  652. mindspore/_extends/graph_kernel/expander.py +0 -80
  653. mindspore/_extends/graph_kernel/expanders/__init__.py +0 -57
  654. mindspore/_extends/graph_kernel/expanders/_utils.py +0 -269
  655. mindspore/_extends/graph_kernel/expanders/addn.py +0 -33
  656. mindspore/_extends/graph_kernel/expanders/batchnorm.py +0 -152
  657. mindspore/_extends/graph_kernel/expanders/batchnorm_grad.py +0 -105
  658. mindspore/_extends/graph_kernel/expanders/bias_add_grad.py +0 -49
  659. mindspore/_extends/graph_kernel/expanders/clip_by_norm_no_div_sum.py +0 -33
  660. mindspore/_extends/graph_kernel/expanders/complex/abs.py +0 -30
  661. mindspore/_extends/graph_kernel/expanders/complex/add.py +0 -44
  662. mindspore/_extends/graph_kernel/expanders/complex/div.py +0 -62
  663. mindspore/_extends/graph_kernel/expanders/complex/mul.py +0 -52
  664. mindspore/_extends/graph_kernel/expanders/complex/real_div.py +0 -62
  665. mindspore/_extends/graph_kernel/expanders/complex/sub.py +0 -45
  666. mindspore/_extends/graph_kernel/expanders/conv2d.py +0 -200
  667. mindspore/_extends/graph_kernel/expanders/dropout_grad.py +0 -30
  668. mindspore/_extends/graph_kernel/expanders/equal_count.py +0 -50
  669. mindspore/_extends/graph_kernel/expanders/erfc.py +0 -35
  670. mindspore/_extends/graph_kernel/expanders/expand_dims.py +0 -50
  671. mindspore/_extends/graph_kernel/expanders/fused_adam.py +0 -44
  672. mindspore/_extends/graph_kernel/expanders/fused_adam_weight_decay.py +0 -47
  673. mindspore/_extends/graph_kernel/expanders/fused_mul_add.py +0 -28
  674. mindspore/_extends/graph_kernel/expanders/gather.py +0 -43
  675. mindspore/_extends/graph_kernel/expanders/gelu_grad.py +0 -70
  676. mindspore/_extends/graph_kernel/expanders/gkdropout.py +0 -40
  677. mindspore/_extends/graph_kernel/expanders/identity.py +0 -25
  678. mindspore/_extends/graph_kernel/expanders/layernorm.py +0 -93
  679. mindspore/_extends/graph_kernel/expanders/layernorm_grad.py +0 -113
  680. mindspore/_extends/graph_kernel/expanders/logsoftmax.py +0 -46
  681. mindspore/_extends/graph_kernel/expanders/logsoftmax_grad.py +0 -36
  682. mindspore/_extends/graph_kernel/expanders/matmul.py +0 -80
  683. mindspore/_extends/graph_kernel/expanders/maximum_grad.py +0 -59
  684. mindspore/_extends/graph_kernel/expanders/minimum_grad.py +0 -80
  685. mindspore/_extends/graph_kernel/expanders/oneslike.py +0 -26
  686. mindspore/_extends/graph_kernel/expanders/reduce_mean.py +0 -43
  687. mindspore/_extends/graph_kernel/expanders/relu_grad.py +0 -32
  688. mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits.py +0 -41
  689. mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits_grad.py +0 -35
  690. mindspore/_extends/graph_kernel/expanders/sigmoid_grad.py +0 -31
  691. mindspore/_extends/graph_kernel/expanders/slice.py +0 -35
  692. mindspore/_extends/graph_kernel/expanders/softmax_cross_entropy_with_logits.py +0 -42
  693. mindspore/_extends/graph_kernel/expanders/softmax_grad_ext.py +0 -41
  694. mindspore/_extends/graph_kernel/expanders/softsign.py +0 -28
  695. mindspore/_extends/graph_kernel/expanders/sqrt_grad.py +0 -29
  696. mindspore/_extends/graph_kernel/expanders/square_sum_all.py +0 -44
  697. mindspore/_extends/graph_kernel/expanders/square_sum_v1.py +0 -37
  698. mindspore/_extends/graph_kernel/expanders/squared_difference.py +0 -43
  699. mindspore/_extends/graph_kernel/expanders/tanh_grad.py +0 -31
  700. mindspore/_extends/graph_kernel/expanders/tile.py +0 -54
  701. mindspore/_extends/graph_kernel/model/op_infer.py +0 -506
  702. mindspore/_extends/parse/jit_fallback_modules.py +0 -51
  703. mindspore/dataset/datapreprocess/preprocess_imagenet_validate_dataset.py +0 -54
  704. mindspore/dataset/engine/graphdata.py +0 -1586
  705. mindspore/include/api/net.h +0 -142
  706. mindspore/ops/_grad/grad_array_ops.py +0 -1347
  707. mindspore/ops/_grad/grad_clip_ops.py +0 -84
  708. mindspore/ops/_grad/grad_debug_ops.py +0 -68
  709. mindspore/ops/_grad/grad_inner_ops.py +0 -235
  710. mindspore/ops/_grad/grad_math_ops.py +0 -1684
  711. mindspore/ops/_grad/grad_nn_ops.py +0 -1529
  712. mindspore/ops/_grad/grad_other_ops.py +0 -89
  713. mindspore/ops/_grad/grad_sequence_ops.py +0 -296
  714. mindspore/ops/_grad/grad_sparse.py +0 -323
  715. mindspore/ops/_grad_experimental/grad_image_ops.py +0 -249
  716. mindspore/ops/_grad_experimental/grad_linalg_ops.py +0 -195
  717. mindspore/ops/_grad_experimental/grad_scalar_ops.py +0 -112
  718. mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
  719. mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
  720. mindspore/ops/bprop_mindir/ApproximateEqual_bprop.mindir +0 -19
  721. mindspore/ops/bprop_mindir/Argmax_bprop.mindir +0 -15
  722. mindspore/ops/bprop_mindir/Argmin_bprop.mindir +0 -15
  723. mindspore/ops/bprop_mindir/AssignSub_bprop.mindir +0 -19
  724. mindspore/ops/bprop_mindir/Assign_bprop.mindir +0 -17
  725. mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +0 -150
  726. mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +0 -66
  727. mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
  728. mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +0 -15
  729. mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
  730. mindspore/ops/bprop_mindir/BatchToSpaceND_bprop.mindir +0 -28
  731. mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
  732. mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +0 -33
  733. mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +0 -306
  734. mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +0 -13
  735. mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
  736. mindspore/ops/bprop_mindir/Concat_bprop.mindir +0 -0
  737. mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +0 -240
  738. mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +0 -247
  739. mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +0 -247
  740. mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +0 -315
  741. mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +0 -278
  742. mindspore/ops/bprop_mindir/DType_bprop.mindir +0 -14
  743. mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +0 -58
  744. mindspore/ops/bprop_mindir/Depend_bprop.mindir +0 -13
  745. mindspore/ops/bprop_mindir/DepthToSpace_bprop.mindir +0 -23
  746. mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +0 -138
  747. mindspore/ops/bprop_mindir/DiagPart_bprop.mindir +0 -15
  748. mindspore/ops/bprop_mindir/Dropout2D_bprop.mindir +0 -0
  749. mindspore/ops/bprop_mindir/Dropout3D_bprop.mindir +0 -0
  750. mindspore/ops/bprop_mindir/DropoutDoMask_bprop.mindir +0 -25
  751. mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +0 -18
  752. mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +0 -27
  753. mindspore/ops/bprop_mindir/Dropout_bprop.mindir +0 -0
  754. mindspore/ops/bprop_mindir/DynamicGRUV2_bprop.mindir +0 -0
  755. mindspore/ops/bprop_mindir/DynamicRNN_bprop.mindir +0 -0
  756. mindspore/ops/bprop_mindir/DynamicShape_bprop.mindir +0 -14
  757. mindspore/ops/bprop_mindir/Elu_bprop.mindir +0 -16
  758. mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
  759. mindspore/ops/bprop_mindir/Equal_bprop.mindir +0 -19
  760. mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +0 -58
  761. mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +0 -16
  762. mindspore/ops/bprop_mindir/Flatten_bprop.mindir +0 -54
  763. mindspore/ops/bprop_mindir/FloorDiv_bprop.mindir +0 -19
  764. mindspore/ops/bprop_mindir/GatherD_bprop.mindir +0 -26
  765. mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +0 -57
  766. mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
  767. mindspore/ops/bprop_mindir/GreaterEqual_bprop.mindir +0 -19
  768. mindspore/ops/bprop_mindir/Greater_bprop.mindir +0 -19
  769. mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +0 -16
  770. mindspore/ops/bprop_mindir/HSwish_bprop.mindir +0 -16
  771. mindspore/ops/bprop_mindir/IOU_bprop.mindir +0 -19
  772. mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
  773. mindspore/ops/bprop_mindir/IsFinite_bprop.mindir +0 -15
  774. mindspore/ops/bprop_mindir/IsInf_bprop.mindir +0 -15
  775. mindspore/ops/bprop_mindir/IsNan_bprop.mindir +0 -15
  776. mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +0 -126
  777. mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +0 -15
  778. mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +0 -30
  779. mindspore/ops/bprop_mindir/LRN_bprop.mindir +0 -43
  780. mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
  781. mindspore/ops/bprop_mindir/LessEqual_bprop.mindir +0 -19
  782. mindspore/ops/bprop_mindir/Less_bprop.mindir +0 -19
  783. mindspore/ops/bprop_mindir/LinSpace_bprop.mindir +0 -23
  784. mindspore/ops/bprop_mindir/Load_bprop.mindir +0 -13
  785. mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +0 -23
  786. mindspore/ops/bprop_mindir/LogicalAnd_bprop.mindir +0 -19
  787. mindspore/ops/bprop_mindir/LogicalNot_bprop.mindir +0 -15
  788. mindspore/ops/bprop_mindir/MaskedSelect_bprop.mindir +0 -21
  789. mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +0 -74
  790. mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +0 -74
  791. mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +0 -75
  792. mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +0 -65
  793. mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
  794. mindspore/ops/bprop_mindir/Maximum_bprop.mindir +0 -0
  795. mindspore/ops/bprop_mindir/Minimum_bprop.mindir +0 -0
  796. mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +0 -27
  797. mindspore/ops/bprop_mindir/Mish_bprop.mindir +0 -35
  798. mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
  799. mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
  800. mindspore/ops/bprop_mindir/NonZero_bprop.mindir +0 -14
  801. mindspore/ops/bprop_mindir/NotEqual_bprop.mindir +0 -19
  802. mindspore/ops/bprop_mindir/OneHot_bprop.mindir +0 -26
  803. mindspore/ops/bprop_mindir/OnesLike_bprop.mindir +0 -14
  804. mindspore/ops/bprop_mindir/PReLU_bprop.mindir +0 -0
  805. mindspore/ops/bprop_mindir/Pad_bprop.mindir +0 -0
  806. mindspore/ops/bprop_mindir/Padding_bprop.mindir +0 -0
  807. mindspore/ops/bprop_mindir/RNNTLoss_bprop.mindir +0 -29
  808. mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +0 -82
  809. mindspore/ops/bprop_mindir/Range_bprop.mindir +0 -22
  810. mindspore/ops/bprop_mindir/Rank_bprop.mindir +0 -14
  811. mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +0 -16
  812. mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
  813. mindspore/ops/bprop_mindir/ReduceAll_bprop.mindir +0 -19
  814. mindspore/ops/bprop_mindir/ReduceAny_bprop.mindir +0 -19
  815. mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +0 -20
  816. mindspore/ops/bprop_mindir/Reshape_bprop.mindir +0 -60
  817. mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +0 -29
  818. mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +0 -89
  819. mindspore/ops/bprop_mindir/ReverseSequence_bprop.mindir +0 -52
  820. mindspore/ops/bprop_mindir/ReverseV2_bprop.mindir +0 -22
  821. mindspore/ops/bprop_mindir/Round_bprop.mindir +0 -15
  822. mindspore/ops/bprop_mindir/ScatterMax_bprop.mindir +0 -0
  823. mindspore/ops/bprop_mindir/ScatterMin_bprop.mindir +0 -0
  824. mindspore/ops/bprop_mindir/ScatterNdUpdate_bprop.mindir +0 -22
  825. mindspore/ops/bprop_mindir/ScatterNd_bprop.mindir +0 -24
  826. mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +0 -22
  827. mindspore/ops/bprop_mindir/ScatterUpdate_bprop.mindir +0 -0
  828. mindspore/ops/bprop_mindir/SeLU_bprop.mindir +0 -21
  829. mindspore/ops/bprop_mindir/Select_bprop.mindir +0 -31
  830. mindspore/ops/bprop_mindir/Shape_bprop.mindir +0 -14
  831. mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +0 -21
  832. mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
  833. mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +0 -16
  834. mindspore/ops/bprop_mindir/Sign_bprop.mindir +0 -15
  835. mindspore/ops/bprop_mindir/Slice_bprop.mindir +0 -26
  836. mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +0 -36
  837. mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
  838. mindspore/ops/bprop_mindir/Softplus_bprop.mindir +0 -16
  839. mindspore/ops/bprop_mindir/Softsign_bprop.mindir +0 -33
  840. mindspore/ops/bprop_mindir/Sort_bprop.mindir +0 -0
  841. mindspore/ops/bprop_mindir/SpaceToBatchND_bprop.mindir +0 -28
  842. mindspore/ops/bprop_mindir/SpaceToDepth_bprop.mindir +0 -23
  843. mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
  844. mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
  845. mindspore/ops/bprop_mindir/Split_bprop.mindir +0 -22
  846. mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +0 -54
  847. mindspore/ops/bprop_mindir/StridedSliceGrad_bprop.mindir +0 -95
  848. mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +0 -98
  849. mindspore/ops/bprop_mindir/Switch_bprop.mindir +0 -29
  850. mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
  851. mindspore/ops/bprop_mindir/Tanh_bprop.mindir +0 -66
  852. mindspore/ops/bprop_mindir/TensorScatterAdd_bprop.mindir +0 -22
  853. mindspore/ops/bprop_mindir/TensorScatterUpdate_bprop.mindir +0 -29
  854. mindspore/ops/bprop_mindir/TensorShape_bprop.mindir +0 -14
  855. mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
  856. mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
  857. mindspore/ops/bprop_mindir/TransShape_bprop.mindir +0 -23
  858. mindspore/ops/bprop_mindir/TruncateDiv_bprop.mindir +0 -19
  859. mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +0 -20
  860. mindspore/ops/bprop_mindir/Unique_bprop.mindir +0 -16
  861. mindspore/ops/bprop_mindir/Unstack_bprop.mindir +0 -22
  862. mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +0 -32
  863. mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +0 -38
  864. mindspore/ops/bprop_mindir/ZerosLike_bprop.mindir +0 -15
  865. mindspore/ops/bprop_mindir/generate_mindir.py +0 -114
  866. mindspore/rewrite/node_visitor.py +0 -44
  867. mindspore/rewrite/topological_manager.py +0 -203
  868. mindspore/scipy/sparse/linalg.py +0 -192
  869. {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/WHEEL +0 -0
  870. {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/top_level.txt +0 -0
@@ -1,150 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�)
3
- �+get_bprop_avg_pool_3d_grad.822:[CNode]823:1+get_bprop_avg_pool_3d_grad.822:[CNode]823:1"REF::bprop.824:Default/bprop.824-op627
4
- C[ValueNode]835[ValueNode]835"Constant*
5
- value* kernel_size�
6
- ?[ValueNode]836[ValueNode]836"Constant*
7
- value*strides�
8
- <[ValueNode]837[ValueNode]837"Constant*
9
- value*pads�
10
- A[ValueNode]838[ValueNode]838"Constant*
11
- value* ceil_mode�
12
- I[ValueNode]839[ValueNode]839"Constant*
13
- value*count_include_pad�
14
- H[ValueNode]840[ValueNode]840"Constant*
15
- value*divisor_override�
16
- C[ValueNode]841[ValueNode]841"Constant*
17
- value* data_format�
18
- @[ValueNode]842[ValueNode]842"Constant*
19
- value*pad_mode�
20
- �
21
- [ValueNode]835
22
- [ValueNode]836
23
- [ValueNode]837
24
- [ValueNode]838
25
- [ValueNode]839
26
- [ValueNode]840
27
- [ValueNode]841
28
- [ValueNode]842,get_bprop_avg_pool_3d_grad.822:[CNode]843:10,get_bprop_avg_pool_3d_grad.822:[CNode]843:10"REF::S-Prim-MakeTuple:11:Default/S-Prim-MakeTuple-op634
29
- C[ValueNode]844[ValueNode]844"Constant*
30
- value* kernel_size�
31
- �
32
- #get_bprop_avg_pool_3d_grad.822:self
33
- [ValueNode]844,get_bprop_avg_pool_3d_grad.822:[CNode]845:12,get_bprop_avg_pool_3d_grad.822:[CNode]845:12"REF::getattr:13:Default/getattr-op635
34
- ?[ValueNode]846[ValueNode]846"Constant*
35
- value*strides�
36
- �
37
- #get_bprop_avg_pool_3d_grad.822:self
38
- [ValueNode]846,get_bprop_avg_pool_3d_grad.822:[CNode]847:14,get_bprop_avg_pool_3d_grad.822:[CNode]847:14"REF::getattr:13:Default/getattr-op636
39
- E[ValueNode]848[ValueNode]848"Constant*
40
- value*
41
- �
42
- #get_bprop_avg_pool_3d_grad.822:self
43
- [ValueNode]848,get_bprop_avg_pool_3d_grad.822:[CNode]849:15,get_bprop_avg_pool_3d_grad.822:[CNode]849:15"REF::getattr:13:Default/getattr-op637
44
- �,get_bprop_avg_pool_3d_grad.822:[CNode]850:16,get_bprop_avg_pool_3d_grad.822:[CNode]850:16"1REF::get_bprop_avg_pool_3d_grad.822:[CNode]849:15:638
45
- @[ValueNode]851[ValueNode]851"Constant*
46
- value*pad_list�
47
- �
48
- ,get_bprop_avg_pool_3d_grad.822:[CNode]850:16
49
- [ValueNode]851*get_bprop_avg_pool_3d_grad.822:pad_list:17*get_bprop_avg_pool_3d_grad.822:pad_list:17"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op639
50
- A[ValueNode]852[ValueNode]852"Constant*
51
- value* ceil_mode�
52
- �
53
- #get_bprop_avg_pool_3d_grad.822:self
54
- [ValueNode]852,get_bprop_avg_pool_3d_grad.822:[CNode]853:18,get_bprop_avg_pool_3d_grad.822:[CNode]853:18"REF::getattr:13:Default/getattr-op640
55
- E[ValueNode]854[ValueNode]854"Constant*
56
- value*
57
- �
58
- #get_bprop_avg_pool_3d_grad.822:self
59
- [ValueNode]854,get_bprop_avg_pool_3d_grad.822:[CNode]855:19,get_bprop_avg_pool_3d_grad.822:[CNode]855:19"REF::getattr:13:Default/getattr-op641
60
- �,get_bprop_avg_pool_3d_grad.822:[CNode]856:20,get_bprop_avg_pool_3d_grad.822:[CNode]856:20"1REF::get_bprop_avg_pool_3d_grad.822:[CNode]855:19:642
61
- I[ValueNode]857[ValueNode]857"Constant*
62
- value*count_include_pad�
63
- �
64
- ,get_bprop_avg_pool_3d_grad.822:[CNode]856:20
65
- [ValueNode]8573get_bprop_avg_pool_3d_grad.822:count_include_pad:213get_bprop_avg_pool_3d_grad.822:count_include_pad:21"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op643
66
- H[ValueNode]858[ValueNode]858"Constant*
67
- value*divisor_override�
68
- �
69
- #get_bprop_avg_pool_3d_grad.822:self
70
- [ValueNode]858,get_bprop_avg_pool_3d_grad.822:[CNode]859:22,get_bprop_avg_pool_3d_grad.822:[CNode]859:22"REF::getattr:13:Default/getattr-op644
71
- C[ValueNode]860[ValueNode]860"Constant*
72
- value* data_format�
73
- �
74
- #get_bprop_avg_pool_3d_grad.822:self
75
- [ValueNode]860,get_bprop_avg_pool_3d_grad.822:[CNode]861:23,get_bprop_avg_pool_3d_grad.822:[CNode]861:23"REF::getattr:13:Default/getattr-op645
76
- @[ValueNode]862[ValueNode]862"Constant*
77
- value*pad_mode�
78
- �
79
- #get_bprop_avg_pool_3d_grad.822:self
80
- [ValueNode]862,get_bprop_avg_pool_3d_grad.822:[CNode]863:24,get_bprop_avg_pool_3d_grad.822:[CNode]863:24"REF::getattr:13:Default/getattr-op646
81
- �
82
- ,get_bprop_avg_pool_3d_grad.822:[CNode]845:12
83
- ,get_bprop_avg_pool_3d_grad.822:[CNode]847:14
84
- *get_bprop_avg_pool_3d_grad.822:pad_list:17
85
- ,get_bprop_avg_pool_3d_grad.822:[CNode]853:18
86
- 3get_bprop_avg_pool_3d_grad.822:count_include_pad:21
87
- ,get_bprop_avg_pool_3d_grad.822:[CNode]859:22
88
- ,get_bprop_avg_pool_3d_grad.822:[CNode]861:23
89
- ,get_bprop_avg_pool_3d_grad.822:[CNode]863:24,get_bprop_avg_pool_3d_grad.822:[CNode]864:25,get_bprop_avg_pool_3d_grad.822:[CNode]864:25"REF::S-Prim-MakeTuple:26:Default/S-Prim-MakeTuple-op647
90
- �
91
- ,get_bprop_avg_pool_3d_grad.822:[CNode]843:10
92
- ,get_bprop_avg_pool_3d_grad.822:[CNode]864:25,get_bprop_avg_pool_3d_grad.822:[CNode]865:27,get_bprop_avg_pool_3d_grad.822:[CNode]865:27"REF::S-Prim-make_dict:28:Default/S-Prim-make_dict-op648
93
- R866866"Constant*<
94
- value*0mindspore.ops.operations._grad_ops.AvgPool3DGrad�!
95
- �
96
- 866
97
- ,get_bprop_avg_pool_3d_grad.822:[CNode]865:272get_bprop_avg_pool_3d_grad.822:фavgpool3d_grad:292get_bprop_avg_pool_3d_grad.822:фavgpool3d_grad:29"REF::MetaFuncGraph::unpack_call:Default/unpack_call.867-op649get_bprop_avg_pool_3d_grad.822*%
98
- #get_bprop_avg_pool_3d_grad.822:self*"
99
- get_bprop_avg_pool_3d_grad.822:x*$
100
- "get_bprop_avg_pool_3d_grad.822:out*'
101
- %get_bprop_avg_pool_3d_grad.822:фdout2-
102
- +get_bprop_avg_pool_3d_grad.822:[CNode]823:1:@bf8b8248fd9c9db80b0d31d687666512880861acf459f9165036db6d5b9bb2a0J/grad_nn_ops.pyB�
103
- u
104
- get_bprop_avg_pool_3d_grad.822:xbprop.824:x_shape:2bprop.824:x_shape:2"REF::shape.825:Default/shape.825-op628
105
- �
106
- bprop.824:x_shape:2bprop.824:[CNode]826:3bprop.824:[CNode]826:3"REF::S-Prim-IsShapeUnKnown:4:#Default/S-Prim-IsShapeUnKnown-op629
107
- q
108
- bprop.824:[CNode]826:3bprop.824:[CNode]827:5bprop.824:[CNode]827:5"REF::bool_.828:Default/bool_.828-op630
109
- �
110
- bprop.824:[CNode]827:5
111
- ✓bprop.829
112
- ✗bprop.830bprop.824:[CNode]831:6bprop.824:[CNode]831:6"
113
- Rbprop.824:[CNode]832:8bprop.824:[CNode]832:8"REF::bprop.824:[CNode]831:6:632
114
- w
115
- bprop.824:[CNode]832:8bprop.824:[CNode]833:9bprop.824:[CNode]833:9"REF::↓bprop.834:Default/↓bprop.834-op633 bprop.8242
116
- bprop.824:[CNode]833:9B�
117
- �
118
- ↓bprop.834:фx_shape
119
- %get_bprop_avg_pool_3d_grad.822:фdout↓bprop.834:dx:30↓bprop.834:dx:30"7REF::get_bprop_avg_pool_3d_grad.822:фavgpool3d_grad:29:650
120
- �
121
- ↓bprop.834:dx:30↓bprop.834:[CNode]868:31↓bprop.834:[CNode]868:31"REF::S-Prim-MakeTuple:32:Default/S-Prim-MakeTuple-op651 ↓bprop.834*
122
- ↓bprop.834:фx_shape2
123
- ↓bprop.834:[CNode]868:31B
124
- after_block� B% ✗bprop.8302
125
- bprop.824:x_shape:2B�
126
- �✓bprop.829:[CNode]869:33✓bprop.829:[CNode]869:33">REF::ClassType::mindspore.ops.operations.array_ops.TensorShape:TDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.TensorShape'-op652
127
- z
128
- get_bprop_avg_pool_3d_grad.822:x✓bprop.829:x_shape:34✓bprop.829:x_shape:34"REF::✓bprop.829:[CNode]869:33:653 ✓bprop.8292
129
- ✓bprop.829:x_shape:34B�
130
- @[ValueNode]870[ValueNode]870"Constant*
131
- value*__bool__�
132
- w
133
- bool_.828:x
134
- [ValueNode]870bool_.828:[CNode]871:35bool_.828:[CNode]871:35"REF::getattr:13:Default/getattr-op654
135
- Ubool_.828:[CNode]872:36bool_.828:[CNode]872:36"REF::bool_.828:[CNode]871:35:655 bool_.828*
136
- bool_.828:x2
137
- bool_.828:[CNode]872:36B�
138
- w
139
- shape.825:input_xshape.825:[CNode]873:37shape.825:[CNode]873:37"REF::S-Prim-Shape:38:Default/S-Prim-Shape-op656 shape.825*
140
- shape.825:input_x2
141
- shape.825:[CNode]873:37Pb
142
- Switch:7Switchb
143
-
144
- getattr:13getattrb
145
- S-Prim-Shape:38 S-Prim-Shapeb'
146
- S-Prim-MakeTuple:26S-Prim-MakeTupleb'
147
- S-Prim-MakeTuple:32S-Prim-MakeTupleb0
148
- S-Prim-IsShapeUnKnown:4S-Prim-IsShapeUnKnownb'
149
- S-Prim-MakeTuple:11S-Prim-MakeTupleb'
150
- S-Prim-make_dict:28S-Prim-make_dicth
@@ -1,66 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�
3
- s%get_bprop_avg_pool_grad.9:[CNode]16:1%get_bprop_avg_pool_grad.9:[CNode]16:1"
4
- A
5
- value* kernel_size�
6
- =
7
- value*strides�
8
- >
9
- value*pad_mode�
10
- A
11
- value* data_format�
12
- �
13
-
14
-
15
-
16
-
17
- A
18
- value* kernel_size�
19
- �
20
- get_bprop_avg_pool_grad.9:self
21
-
22
- =
23
- value*strides�
24
- �
25
- get_bprop_avg_pool_grad.9:self
26
-
27
- >
28
- value*pad_mode�
29
- �
30
- get_bprop_avg_pool_grad.9:self
31
-
32
- <
33
- value*format�
34
- �
35
- get_bprop_avg_pool_grad.9:self
36
-
37
- �
38
- $get_bprop_avg_pool_grad.9:[CNode]2:4
39
- $get_bprop_avg_pool_grad.9:[CNode]3:6
40
- $get_bprop_avg_pool_grad.9:[CNode]4:7
41
- $get_bprop_avg_pool_grad.9:[CNode]5:8$get_bprop_avg_pool_grad.9:[CNode]6:9$get_bprop_avg_pool_grad.9:[CNode]6:9"REF::S-Prim-MakeTuple:10:Default/S-Prim-MakeTuple-op6
42
- �
43
- $get_bprop_avg_pool_grad.9:[CNode]1:2
44
- $get_bprop_avg_pool_grad.9:[CNode]6:9%get_bprop_avg_pool_grad.9:[CNode]7:11%get_bprop_avg_pool_grad.9:[CNode]7:11"REF::S-Prim-make_dict:12:Default/S-Prim-make_dict-op7
45
- N2525"Constant*:
46
- value*.mindspore.ops.operations._grad_ops.AvgPoolGrad�!
47
- �
48
- 25
49
- %get_bprop_avg_pool_grad.9:[CNode]7:11)get_bprop_avg_pool_grad.9:avgpool_grad:13)get_bprop_avg_pool_grad.9:avgpool_grad:13"REF::MetaFuncGraph::unpack_call:Default/unpack_call.8-op8get_bprop_avg_pool_grad.9*
50
- get_bprop_avg_pool_grad.9:self*
51
- get_bprop_avg_pool_grad.9:x*
52
- get_bprop_avg_pool_grad.9:out*
53
- get_bprop_avg_pool_grad.9:dout2'
54
- %get_bprop_avg_pool_grad.9:[CNode]16:1:@4a74a0ba73d5078e776e116148e7145421a39dc8a84d9ba3d7c495f451ee6680J/grad_nn_ops.pyB�
55
- �
56
- get_bprop_avg_pool_grad.9:x
57
- get_bprop_avg_pool_grad.9:out
58
- get_bprop_avg_pool_grad.9:doutbprop.11:dx:14bprop.11:dx:14".REF::get_bprop_avg_pool_grad.9:avgpool_grad:13:9
59
- w
60
- bprop.11:dx:14bprop.11:[CNode]10:15bprop.11:[CNode]10:15"REF::S-Prim-MakeTuple:16:Default/S-Prim-MakeTuple-op10bprop.112
61
- bprop.11:[CNode]10:15Pb
62
- getattr:5getattrb'
63
- S-Prim-make_dict:12S-Prim-make_dictb'
64
- S-Prim-MakeTuple:16S-Prim-MakeTupleb&
65
- S-Prim-MakeTuple:3S-Prim-MakeTupleb'
66
- S-Prim-MakeTuple:10S-Prim-MakeTupleh
@@ -1,15 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�
3
- �/get_bprop_bn_training_reduce.1308:[CNode]1309:1/get_bprop_bn_training_reduce.1308:[CNode]1309:1"REF::bprop.1310:Default/bprop.1310-op1015!get_bprop_bn_training_reduce.1308*(
4
- &get_bprop_bn_training_reduce.1308:self*%
5
- #get_bprop_bn_training_reduce.1308:x*'
6
- %get_bprop_bn_training_reduce.1308:out*(
7
- &get_bprop_bn_training_reduce.1308:dout21
8
- /get_bprop_bn_training_reduce.1308:[CNode]1309:1:@398f9bcbcf29e1e9801c606ad54046cd56ad62bc367f76d4ecb6fc930583d567J/grad_quant_ops.pyB�
9
- �
10
- #get_bprop_bn_training_reduce.1308:xbprop.1310:[CNode]1311:2bprop.1310:[CNode]1311:2".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:0Default/S-Prim-hyper_map[zeros_like_leaf]-op1016
11
- �
12
- bprop.1310:[CNode]1311:2bprop.1310:[CNode]1312:3bprop.1310:[CNode]1312:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op1017
13
- bprop.13102
14
- bprop.1310:[CNode]1312:3Pb&
15
- S-Prim-MakeTuple:4S-Prim-MakeTupleh
@@ -1,28 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�
3
- w(get_bprop_batch_to_space_nd.1:[CNode]2:1(get_bprop_batch_to_space_nd.1:[CNode]2:1" REF::bprop.3:Default/bprop.3-op0
4
- ? [ValueNode]4 [ValueNode]4"Constant*
5
- value* block_shape�
6
- �
7
- "get_bprop_batch_to_space_nd.1:self
8
- [ValueNode]4(get_bprop_batch_to_space_nd.1:[CNode]5:2(get_bprop_batch_to_space_nd.1:[CNode]5:2"REF::getattr:3:Default/getattr-op1
9
- 9 [ValueNode]6 [ValueNode]6"Constant*
10
- value*crops�
11
- �
12
- "get_bprop_batch_to_space_nd.1:self
13
- [ValueNode]6(get_bprop_batch_to_space_nd.1:[CNode]7:4(get_bprop_batch_to_space_nd.1:[CNode]7:4"REF::getattr:3:Default/getattr-op2
14
- �
15
- (get_bprop_batch_to_space_nd.1:[CNode]5:2
16
- (get_bprop_batch_to_space_nd.1:[CNode]7:46get_bprop_batch_to_space_nd.1:batch_to_space_nd_grad:56get_bprop_batch_to_space_nd.1:batch_to_space_nd_grad:5"AREF::ClassType::mindspore.ops.operations.array_ops.SpaceToBatchND:EDefault/class 'mindspore.ops.operations.array_ops.SpaceToBatchND'-op3get_bprop_batch_to_space_nd.1*$
17
- "get_bprop_batch_to_space_nd.1:self*!
18
- get_bprop_batch_to_space_nd.1:x*#
19
- !get_bprop_batch_to_space_nd.1:out*$
20
- "get_bprop_batch_to_space_nd.1:dout2*
21
- (get_bprop_batch_to_space_nd.1:[CNode]2:1:@bae659ee9c4ba628e8eb4986b38e636fc494d3625431852b881adfb243c2e261J/grad_array_ops.pyB�
22
- �
23
- "get_bprop_batch_to_space_nd.1:dout bprop.3:dx:6 bprop.3:dx:6";REF::get_bprop_batch_to_space_nd.1:batch_to_space_nd_grad:5:4
24
- m
25
- bprop.3:dx:6bprop.3:[CNode]8:7bprop.3:[CNode]8:7"REF::S-Prim-MakeTuple:8:Default/S-Prim-MakeTuple-op5bprop.32
26
- bprop.3:[CNode]8:7Pb
27
- getattr:3getattrb&
28
- S-Prim-MakeTuple:8S-Prim-MakeTupleh
@@ -1,33 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�
3
- ,get_bprop_binary_cross_entropy.2:[CNode]17:1,get_bprop_binary_cross_entropy.2:[CNode]17:1" REF::bprop.6:Default/bprop.6-op0
4
- ?
5
- value* reduction�
6
- �
7
- %get_bprop_binary_cross_entropy.2:self
8
-
9
- �
10
- +get_bprop_binary_cross_entropy.2:[CNode]1:2'get_bprop_binary_cross_entropy.2:grad:4'get_bprop_binary_cross_entropy.2:grad:4"IREF::ClassType::mindspore.ops.operations._grad_ops.BinaryCrossEntropyGrad:]Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.BinaryCrossEntropyGrad'-op2 get_bprop_binary_cross_entropy.2*'
11
- %get_bprop_binary_cross_entropy.2:self*$
12
- "get_bprop_binary_cross_entropy.2:x*$
13
- "get_bprop_binary_cross_entropy.2:y*)
14
- 'get_bprop_binary_cross_entropy.2:weight*&
15
- $get_bprop_binary_cross_entropy.2:out*'
16
- %get_bprop_binary_cross_entropy.2:dout2.
17
- ,get_bprop_binary_cross_entropy.2:[CNode]17:1:@ae043af2d528e56aa13b09b1bb8b1799c25e61a31aed38bbb97885ccd3c4ccf4J/grad_nn_ops.pyB�
18
- �
19
- "get_bprop_binary_cross_entropy.2:x
20
- "get_bprop_binary_cross_entropy.2:y
21
- %get_bprop_binary_cross_entropy.2:dout
22
- 'get_bprop_binary_cross_entropy.2:weight bprop.6:dx:5 bprop.6:dx:5",REF::get_bprop_binary_cross_entropy.2:grad:4:3
23
- �
24
- "get_bprop_binary_cross_entropy.2:ybprop.6:[CNode]3:6bprop.6:[CNode]3:6".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op4
25
- �
26
- 'get_bprop_binary_cross_entropy.2:weightbprop.6:[CNode]4:7bprop.6:[CNode]4:7".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op5
27
- �
28
- bprop.6:dx:5
29
- bprop.6:[CNode]3:6
30
- bprop.6:[CNode]4:7bprop.6:[CNode]5:8bprop.6:[CNode]5:8"REF::S-Prim-MakeTuple:9:Default/S-Prim-MakeTuple-op6bprop.62
31
- bprop.6:[CNode]5:8Pb
32
- getattr:3getattrb&
33
- S-Prim-MakeTuple:9S-Prim-MakeTupleh
@@ -1,306 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�
3
- m#get_bprop_broadcast_to.1:[CNode]2:1#get_bprop_broadcast_to.1:[CNode]2:1" REF::bprop.3:Default/bprop.3-op0
4
- =4949"Constant*)
5
- value*mindspore.ops._grad.grad_base�"
6
- :49_4149_41"Constant*
7
- value*sum_grad_reduce_axis�#
8
- r
9
- 49
10
- 49_41get_bprop_broadcast_to.1:49:42get_bprop_broadcast_to.1:49:42"REF::resolve:43:Default/resolve-op33
11
- ?
12
- value* keep_dims�
13
- �
14
-
15
- 6
16
- value�
17
- �
18
-
19
- �
20
- %get_bprop_broadcast_to.1:[CNode]86:81
21
- %get_bprop_broadcast_to.1:[CNode]88:83%get_bprop_broadcast_to.1:[CNode]89:85%get_bprop_broadcast_to.1:[CNode]89:85"REF::S-Prim-make_dict:86:Default/S-Prim-make_dict-op64
22
- K9090"Constant*7
23
- value*+mindspore.ops.operations.math_ops.ReduceSum�!
24
- �
25
- 90
26
- %get_bprop_broadcast_to.1:[CNode]89:85+get_bprop_broadcast_to.1:reduce_keep_dim:87+get_bprop_broadcast_to.1:reduce_keep_dim:87"REF::MetaFuncGraph::unpack_call:Default/unpack_call.91-op65get_bprop_broadcast_to.1*
27
- get_bprop_broadcast_to.1:self*
28
- get_bprop_broadcast_to.1:x*
29
- get_bprop_broadcast_to.1:out*
30
- get_bprop_broadcast_to.1:dout2%
31
- #get_bprop_broadcast_to.1:[CNode]2:1:@a4cfb55f3df63a6c0d4df2948c19dc8ce349f1dbfc89aff5d56ae2941f16945fJ/grad_array_ops.pyB�
32
- u
33
- get_bprop_broadcast_to.1:xbprop.3:фx_shape:2bprop.3:фx_shape:2"REF::S-Prim-Shape:3:Default/S-Prim-Shape-op1
34
- ~
35
- bprop.3:фx_shape:2bprop.3:[CNode]4:4bprop.3:[CNode]4:4"REF::S-Prim-IsShapeUnKnown:5:!Default/S-Prim-IsShapeUnKnown-op2
36
- _
37
- bprop.3:[CNode]4:4bprop.3:[CNode]5:6bprop.3:[CNode]5:6" REF::bool_.6:Default/bool_.6-op3
38
- w
39
- bprop.3:[CNode]5:6
40
-
41
- ↰bprop.7
42
-
43
- ↱bprop.8bprop.3:[CNode]9:7bprop.3:[CNode]9:7"
44
- Bbprop.3:dynamic:9bprop.3:dynamic:9"REF::bprop.3:[CNode]9:7:5
45
- �
46
- bprop.3:dynamic:9bprop.3:[CNode]10:10bprop.3:[CNode]10:10"REF::MetaFuncGraph::logical_not:Default/S-Prim-logical_not-op6
47
- e
48
- bprop.3:[CNode]10:10bprop.3:[CNode]11:11bprop.3:[CNode]11:11" REF::bool_.6:Default/bool_.6-op7
49
- 
50
- bprop.3:[CNode]11:11
51
- ↰bprop.12
52
- ↱bprop.13bprop.3:[CNode]14:12bprop.3:[CNode]14:12"
53
- Jbprop.3:[CNode]15:13bprop.3:[CNode]15:13"REF::bprop.3:[CNode]14:12:9
54
- f
55
- bprop.3:[CNode]15:13bprop.3:[CNode]16:14bprop.3:[CNode]16:14" REF::bool_.6:Default/bool_.6-op10
56
- �
57
- bprop.3:[CNode]16:14
58
- ✓bprop.17
59
- ✗bprop.18bprop.3:[CNode]19:15bprop.3:[CNode]19:15"
60
- Kbprop.3:[CNode]20:16bprop.3:[CNode]20:16"REF::bprop.3:[CNode]19:15:12
61
- �
62
- get_bprop_broadcast_to.1:outbprop.3:broadcast_shape:88bprop.3:broadcast_shape:88"REF::S-Prim-Shape:3:Default/S-Prim-Shape-op66
63
- 
64
- get_bprop_broadcast_to.1:doutbprop.3:dout_shape:100bprop.3:dout_shape:100"REF::S-Prim-Shape:3:Default/S-Prim-Shape-op76bprop.32
65
- bprop.3:[CNode]20:16B�
66
- `✗bprop.18:[CNode]21:17✗bprop.18:[CNode]21:17"REF::↓bprop.22:Default/↓bprop.22-op13 ✗bprop.182
67
- ✗bprop.18:[CNode]21:17B�
68
- k
69
- bprop.3:dynamic:9↓bprop.22:[CNode]23:18↓bprop.22:[CNode]23:18" REF::bool_.6:Default/bool_.6-op14
70
- �
71
- ↓bprop.22:[CNode]23:18
72
- ↰↓bprop.24
73
- ↱↓bprop.25↓bprop.22:[CNode]26:19↓bprop.22:[CNode]26:19"
74
- S↓bprop.22:dynamic:20↓bprop.22:dynamic:20"REF::↓bprop.22:[CNode]26:19:16
75
- �
76
- ↓bprop.22:dynamic:20↓bprop.22:[CNode]27:21↓bprop.22:[CNode]27:21"REF::MetaFuncGraph::logical_not:Default/S-Prim-logical_not-op17
77
- r
78
- ↓bprop.22:[CNode]27:21↓bprop.22:[CNode]28:22↓bprop.22:[CNode]28:22" REF::bool_.6:Default/bool_.6-op18
79
- �
80
- ↓bprop.22:[CNode]28:22
81
- ✓↓bprop.29
82
- ✗↓bprop.30↓bprop.22:[CNode]31:23↓bprop.22:[CNode]31:23"
83
- W↓bprop.22:[CNode]32:24↓bprop.22:[CNode]32:24"REF::↓bprop.22:[CNode]31:23:20
84
- �
85
- ↓bprop.22:[CNode]32:24↓bprop.22:[CNode]33:25↓bprop.22:[CNode]33:25"REF::↓↓bprop.34:Default/↓↓bprop.34-op21
86
- ;
87
- value*dtype�
88
- �
89
- get_bprop_broadcast_to.1:dout
90
-
91
- ↓bprop.22:[CNode]33:25B
92
- after_block� B�
93
- �
94
- ↓↓bprop.34:фdx↓↓bprop.34:[CNode]35:26↓↓bprop.34:[CNode]35:26"REF::S-Prim-MakeTuple:27:Default/S-Prim-MakeTuple-op22↓↓bprop.34*
95
- ↓↓bprop.34:фdx2
96
- ↓↓bprop.34:[CNode]35:26B
97
- after_block� B�
98
- @
99
- valueb
100
- :value0�
101
- @
102
- valueb
103
- :value0�
104
- @
105
- valueb
106
- :value0�
107
- �
108
-
109
-
110
-
111
- �
112
- ↓bprop.22:out_type:28
113
- ✗↓bprop.30:[CNode]40:30✗↓bprop.30:[CNode]41:32✗↓bprop.30:[CNode]41:32"REF::MetaFuncGraph::in:Default/S-Prim-in-op25
114
- {
115
- ✗↓bprop.30:[CNode]41:32✗↓bprop.30:[CNode]42:33✗↓bprop.30:[CNode]42:33" REF::bool_.6:Default/bool_.6-op26
116
- �
117
- ✗↓bprop.30:[CNode]42:33
118
- ✓✗↓bprop.43
119
- ✗✗↓bprop.44✗↓bprop.30:[CNode]45:34✗↓bprop.30:[CNode]45:34"
120
- `✗↓bprop.30:[CNode]46:35✗↓bprop.30:[CNode]46:35" REF::✗↓bprop.30:[CNode]45:34:28
121
- �
122
- ✗↓bprop.30:[CNode]46:35✗↓bprop.30:[CNode]47:36✗↓bprop.30:[CNode]47:36"REF::↓✗↓bprop.48:Default/↓✗↓bprop.48-op29
123
- �
124
- get_bprop_broadcast_to.1:x✗↓bprop.30:фx_shape:38✗↓bprop.30:фx_shape:38"REF::S-Prim-TensorShape:39:Default/S-Prim-TensorShape-op31
125
- �✗↓bprop.30:[CNode]50:44✗↓bprop.30:[CNode]50:44"PREF::ClassType::mindspore.ops.operations._inner_ops.DynamicBroadcastGradientArgs:eDefault/MindIRClassType:class 'mindspore.ops.operations._inner_ops.DynamicBroadcastGradientArgs'-op34
126
- �
127
- get_bprop_broadcast_to.1:out!✗↓bprop.30:broadcast_shape:45!✗↓bprop.30:broadcast_shape:45"REF::S-Prim-TensorShape:39:Default/S-Prim-TensorShape-op35
128
- �
129
- !✗↓bprop.30:broadcast_shape:45
130
- ✗↓bprop.30:фx_shape:38✗↓bprop.30:[CNode]51:46✗↓bprop.30:[CNode]51:46" REF::✗↓bprop.30:[CNode]50:44:36
131
- 6
132
- value�
133
- �
134
- ✗↓bprop.30:[CNode]51:46
135
-
136
- ✗↓bprop.30:[CNode]47:36B�
137
- �↓✗↓bprop.48:dx:37↓✗↓bprop.48:dx:37":REF::ClassType::mindspore.ops.operations.array_ops.Reshape:ODefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Reshape'-op30
138
- �
139
- ↓✗↓bprop.48:фreduced_grad
140
- ✗↓bprop.30:фx_shape:38↓✗↓bprop.48:dx:40↓✗↓bprop.48:dx:40"REF::↓✗↓bprop.48:dx:37:32↓✗↓bprop.48*"
141
- ↓✗↓bprop.48:фreduced_grad2
142
- ↓✗↓bprop.48:dx:40B
143
- after_block� B�
144
- �
145
- get_bprop_broadcast_to.1:dout
146
- ✗↓bprop.30:reduction_axes:47✗✗↓bprop.44:[CNode]53:48✗✗↓bprop.44:[CNode]53:48"REF::S-Prim-MakeTuple:49:Default/S-Prim-MakeTuple-op38
147
- ?
148
- value* keep_dims�
149
- �
150
-
151
- 6
152
- value�
153
- �
154
-
155
- �
156
- ✗✗↓bprop.44:[CNode]55:50
157
- ✗✗↓bprop.44:[CNode]57:52✗✗↓bprop.44:[CNode]58:54✗✗↓bprop.44:[CNode]58:54"REF::S-Prim-make_dict:55:Default/S-Prim-make_dict-op41
158
- �
159
- get_bprop_broadcast_to.1:49:42
160
- ✗✗↓bprop.44:[CNode]53:48
161
- ✗✗↓bprop.44:[CNode]58:54!✗✗↓bprop.44:reduced_grad:56!✗✗↓bprop.44:reduced_grad:56"REF::MetaFuncGraph::unpack_call:Default/unpack_call.59-op42✗✗↓bprop.442#
162
- !✗✗↓bprop.44:reduced_grad:56B�
163
- �✓✗↓bprop.43:[CNode]60:57✓✗↓bprop.43:[CNode]60:57"7REF::ClassType::mindspore.ops.operations.array_ops.Cast:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Cast'-op43
164
- �✓✗↓bprop.43:[CNode]61:58✓✗↓bprop.43:[CNode]61:58"7REF::ClassType::mindspore.ops.operations.array_ops.Cast:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Cast'-op44
165
- @
166
- valueb
167
- :value0�
168
- �
169
- get_bprop_broadcast_to.1:dout
170
-
171
- �
172
- ✓✗↓bprop.43:dout:59
173
- ✗↓bprop.30:reduction_axes:47✓✗↓bprop.43:[CNode]63:60✓✗↓bprop.43:[CNode]63:60"REF::S-Prim-MakeTuple:61:Default/S-Prim-MakeTuple-op46
174
- ?
175
- value* keep_dims�
176
- �
177
-
178
- 6
179
- value�
180
- �
181
-
182
- �
183
- ✓✗↓bprop.43:[CNode]65:62
184
- ✓✗↓bprop.43:[CNode]67:64✓✗↓bprop.43:[CNode]68:66✓✗↓bprop.43:[CNode]68:66"REF::S-Prim-make_dict:67:Default/S-Prim-make_dict-op49
185
- �
186
- get_bprop_broadcast_to.1:49:42
187
- ✓✗↓bprop.43:[CNode]63:60
188
- ✓✗↓bprop.43:[CNode]68:66!✓✗↓bprop.43:reduced_grad:68!✓✗↓bprop.43:reduced_grad:68"REF::MetaFuncGraph::unpack_call:Default/unpack_call.69-op50
189
- �
190
- !✓✗↓bprop.43:reduced_grad:68
191
- ↓bprop.22:out_type:28!✓✗↓bprop.43:reduced_grad:69!✓✗↓bprop.43:reduced_grad:69"#REF::✓✗↓bprop.43:[CNode]60:57:51✓✗↓bprop.432#
192
- !✓✗↓bprop.43:reduced_grad:69B�
193
- >
194
- value*__bool__�
195
- m
196
- bool_.6:x
197
-
198
- Kbool_.6:[CNode]72:71bool_.6:[CNode]72:71"REF::bool_.6:[CNode]71:70:53bool_.6*
199
- bool_.6:x2
200
- bool_.6:[CNode]72:71B�
201
- @
202
- valueb
203
- :value0�
204
- @
205
- valueb
206
- :value0�
207
- @
208
- valueb
209
- :value0�
210
- �
211
-
212
-
213
-
214
- �
215
- ↓bprop.22:out_type:28
216
- ✓↓bprop.29:[CNode]76:72✓↓bprop.29:[CNode]77:74✓↓bprop.29:[CNode]77:74"REF::MetaFuncGraph::in:Default/S-Prim-in-op55
217
- {
218
- ✓↓bprop.29:[CNode]77:74✓↓bprop.29:[CNode]78:75✓↓bprop.29:[CNode]78:75" REF::bool_.6:Default/bool_.6-op56
219
- �
220
- ✓↓bprop.29:[CNode]78:75
221
- ✓✓↓bprop.79
222
- ✗✓↓bprop.80✓↓bprop.29:[CNode]81:76✓↓bprop.29:[CNode]81:76"
223
- `✓↓bprop.29:[CNode]82:77✓↓bprop.29:[CNode]82:77" REF::✓↓bprop.29:[CNode]81:76:58
224
- �
225
- ✓↓bprop.29:[CNode]82:77✓↓bprop.29:[CNode]83:78✓↓bprop.29:[CNode]83:78"REF::↓✓↓bprop.84:Default/↓✓↓bprop.84-op59
226
- �
227
- bprop.3:broadcast_shape:88
228
- bprop.3:фx_shape:2✓↓bprop.29:[CNode]92:89✓↓bprop.29:[CNode]92:89"$REF::S-Prim-BroadcastGradientArgs:90:)Default/S-Prim-BroadcastGradientArgs-op67
229
- 6
230
- value�
231
- �
232
- ✓↓bprop.29:[CNode]92:89
233
-
234
- ✓↓bprop.29:[CNode]83:78B�
235
- �↓✓↓bprop.84:dx:79↓✓↓bprop.84:dx:79":REF::ClassType::mindspore.ops.operations.array_ops.Reshape:ODefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Reshape'-op60
236
- �
237
- ↓✓↓bprop.84:фreduced_grad
238
- bprop.3:фx_shape:2↓✓↓bprop.84:dx:80↓✓↓bprop.84:dx:80"REF::↓✓↓bprop.84:dx:79:61↓✓↓bprop.84*"
239
- ↓✓↓bprop.84:фreduced_grad2
240
- ↓✓↓bprop.84:dx:80B
241
- after_block� B�
242
- �
243
- get_bprop_broadcast_to.1:dout
244
- ✓↓bprop.29:reduction_axes:91!✗✓↓bprop.80:reduced_grad:92!✗✓↓bprop.80:reduced_grad:92"0REF::get_bprop_broadcast_to.1:reduce_keep_dim:87:69✗✓↓bprop.802#
245
- !✗✓↓bprop.80:reduced_grad:92B�
246
- �✓✓↓bprop.79:[CNode]94:93✓✓↓bprop.79:[CNode]94:93"7REF::ClassType::mindspore.ops.operations.array_ops.Cast:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Cast'-op70
247
- �✓✓↓bprop.79:[CNode]95:94✓✓↓bprop.79:[CNode]95:94"7REF::ClassType::mindspore.ops.operations.array_ops.Cast:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Cast'-op71
248
- @
249
- valueb
250
- :value0�
251
- �
252
- get_bprop_broadcast_to.1:dout
253
-
254
- �
255
- ✓✓↓bprop.79:dout:95
256
- ✓↓bprop.29:reduction_axes:91!✓✓↓bprop.79:reduced_grad:96!✓✓↓bprop.79:reduced_grad:96"0REF::get_bprop_broadcast_to.1:reduce_keep_dim:87:73
257
- �
258
- !✓✓↓bprop.79:reduced_grad:96
259
- ↓bprop.22:out_type:28!✓✓↓bprop.79:reduced_grad:97!✓✓↓bprop.79:reduced_grad:97"#REF::✓✓↓bprop.79:[CNode]94:93:74✓✓↓bprop.792#
260
- !✓✓↓bprop.79:reduced_grad:97B�
261
- �
262
- bprop.3:broadcast_shape:88↱↓bprop.25:[CNode]97:98↱↓bprop.25:[CNode]97:98"REF::S-Prim-IsShapeUnKnown:99:"Default/S-Prim-IsShapeUnKnown-op75↱↓bprop.252
263
- ↱↓bprop.25:[CNode]97:98B%↰↓bprop.242
264
- bprop.3:dynamic:9B�
265
- �
266
- bprop.3:dout_shape:100↱bprop.8:[CNode]98:101↱bprop.8:[CNode]98:101"REF::S-Prim-IsShapeUnKnown:102:"Default/S-Prim-IsShapeUnKnown-op77
267
- ↱bprop.82
268
- ↱bprop.8:[CNode]98:101B"
269
- ↰bprop.72
270
- bprop.3:[CNode]4:4B�
271
- �
272
- get_bprop_broadcast_to.1:dout✓bprop.17:[CNode]99:103✓bprop.17:[CNode]99:103"REF::S-Prim-MakeTuple:104:Default/S-Prim-MakeTuple-op78 ✓bprop.172
273
- ✓bprop.17:[CNode]99:103B% ↱bprop.132
274
- bprop.3:[CNode]10:10B�
275
- �
276
- bprop.3:фx_shape:2
277
- bprop.3:dout_shape:100↰bprop.12:[CNode]100:105↰bprop.12:[CNode]100:105"REF::MetaFuncGraph::equal:Default/S-Prim-equal-op79 ↰bprop.122
278
- ↰bprop.12:[CNode]100:105Pb
279
- Switch:8Switchb
280
-
281
- getattr:29getattrb
282
-
283
- resolve:43resolveb(
284
- S-Prim-MakeTuple:104S-Prim-MakeTupleb'
285
- S-Prim-MakeTuple:27S-Prim-MakeTupleb
286
- S-Prim-Shape:3 S-Prim-Shapebm
287
- S-Prim-TensorShape:39S-Prim-TensorShape
288
- output_names�� Zoutput�
289
- input_names�� Zinput_x�b'
290
- S-Prim-MakeTuple:61S-Prim-MakeTupleb'
291
- S-Prim-make_dict:67S-Prim-make_dictb'
292
- S-Prim-MakeTuple:63S-Prim-MakeTupleb'
293
- S-Prim-MakeTuple:65S-Prim-MakeTupleb'
294
- S-Prim-MakeTuple:51S-Prim-MakeTupleb'
295
- S-Prim-make_dict:55S-Prim-make_dictb'
296
- S-Prim-MakeTuple:82S-Prim-MakeTupleb?
297
- S-Prim-BroadcastGradientArgs:90S-Prim-BroadcastGradientArgsb'
298
- S-Prim-MakeTuple:73S-Prim-MakeTupleb'
299
- S-Prim-MakeTuple:31S-Prim-MakeTupleb'
300
- S-Prim-MakeTuple:49S-Prim-MakeTupleb'
301
- S-Prim-make_dict:86S-Prim-make_dictb'
302
- S-Prim-MakeTuple:53S-Prim-MakeTupleb'
303
- S-Prim-MakeTuple:84S-Prim-MakeTupleb0
304
- S-Prim-IsShapeUnKnown:5S-Prim-IsShapeUnKnownb2
305
- S-Prim-IsShapeUnKnown:102S-Prim-IsShapeUnKnownb1
306
- S-Prim-IsShapeUnKnown:99S-Prim-IsShapeUnKnownh
@@ -1,13 +0,0 @@
1
-
2
- 0.1.1 MindSpore*2.0.0:�
3
- }'get_bprop_broad_cast.1037:[CNode]1038:1'get_bprop_broad_cast.1037:[CNode]1038:1"REF::bprop.1039:Default/bprop.1039-op826get_bprop_broad_cast.1037*
4
- get_bprop_broad_cast.1037:self*
5
- get_bprop_broad_cast.1037:x*
6
- get_bprop_broad_cast.1037:out*
7
- get_bprop_broad_cast.1037:dout2)
8
- 'get_bprop_broad_cast.1037:[CNode]1038:1:@403d64b65284847b0b5f01f2121501208c9d8fb5c1da4b332a158de570991518J/grad_comm_ops.pyB�
9
- �
10
- get_bprop_broad_cast.1037:doutbprop.1039:[CNode]1040:2bprop.1039:[CNode]1040:2"REF::S-Prim-MakeTuple:3:Default/S-Prim-MakeTuple-op827
11
- bprop.10392
12
- bprop.1039:[CNode]1040:2Pb&
13
- S-Prim-MakeTuple:3S-Prim-MakeTupleh