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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of mindspore might be problematic. Click here for more details.

Files changed (944) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/Third_Party_Open_Source_Software_Notice +9064 -0
  3. mindspore/__init__.py +9 -4
  4. mindspore/_akg/akg/composite/build_module.py +11 -0
  5. mindspore/_akg/akg/config/repository_cuda.json +11 -0
  6. mindspore/_akg/akg/tvm/contrib/nvcc.py +4 -3
  7. mindspore/_c_dataengine.cpython-37m-aarch64-linux-gnu.so +0 -0
  8. mindspore/_c_expression.cpython-37m-aarch64-linux-gnu.so +0 -0
  9. mindspore/_c_mindrecord.cpython-37m-aarch64-linux-gnu.so +0 -0
  10. mindspore/_check_jit_forbidden_api.py +102 -0
  11. mindspore/_checkparam.py +1066 -1001
  12. mindspore/_extends/builtin_operations.py +32 -4
  13. mindspore/_extends/graph_kernel/model/graph_split.py +66 -222
  14. mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +12 -9
  15. mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +119 -26
  16. mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +50 -50
  17. mindspore/_extends/parallel_compile/akg_compiler/util.py +9 -6
  18. mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +4 -25
  19. mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +9 -4
  20. mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -27
  21. mindspore/_extends/parse/__init__.py +5 -3
  22. mindspore/_extends/parse/namespace.py +17 -2
  23. mindspore/_extends/parse/parser.py +193 -34
  24. mindspore/_extends/parse/resources.py +7 -8
  25. mindspore/_extends/parse/standard_method.py +1780 -435
  26. mindspore/_extends/parse/trope.py +3 -1
  27. mindspore/_mindspore_offline_debug.cpython-37m-aarch64-linux-gnu.so +0 -0
  28. mindspore/amp.py +53 -58
  29. mindspore/bin/cache_admin +0 -0
  30. mindspore/bin/cache_server +0 -0
  31. mindspore/boost/adasum.py +3 -2
  32. mindspore/boost/boost.py +2 -2
  33. mindspore/boost/boost_cell_wrapper.py +46 -26
  34. mindspore/boost/dim_reduce.py +6 -5
  35. mindspore/boost/grad_accumulation.py +2 -1
  36. mindspore/boost/group_loss_scale_manager.py +1 -1
  37. mindspore/common/__init__.py +11 -10
  38. mindspore/common/_decorator.py +2 -0
  39. mindspore/common/_register_for_adapter.py +55 -0
  40. mindspore/common/_stub_tensor.py +201 -0
  41. mindspore/common/_utils.py +57 -0
  42. mindspore/common/api.py +582 -297
  43. mindspore/common/dtype.py +66 -18
  44. mindspore/common/dump.py +2 -2
  45. mindspore/common/initializer.py +38 -1
  46. mindspore/common/jit_config.py +25 -13
  47. mindspore/common/mutable.py +53 -24
  48. mindspore/common/parameter.py +60 -37
  49. mindspore/common/seed.py +8 -24
  50. mindspore/common/sparse_tensor.py +927 -0
  51. mindspore/common/tensor.py +1627 -3900
  52. mindspore/communication/__init__.py +10 -5
  53. mindspore/communication/_comm_helper.py +78 -214
  54. mindspore/communication/_hccl_management.py +2 -1
  55. mindspore/communication/management.py +136 -47
  56. mindspore/config/op_info.config +501 -1008
  57. mindspore/config/super_bar_config.json +512 -0
  58. mindspore/context.py +291 -56
  59. mindspore/dataset/__init__.py +12 -8
  60. mindspore/dataset/audio/__init__.py +9 -9
  61. mindspore/dataset/audio/transforms.py +1090 -228
  62. mindspore/dataset/audio/utils.py +87 -39
  63. mindspore/dataset/audio/validators.py +223 -1
  64. mindspore/dataset/callback/ds_callback.py +17 -15
  65. mindspore/dataset/core/config.py +246 -17
  66. mindspore/dataset/core/py_util_helpers.py +4 -3
  67. mindspore/dataset/core/validator_helpers.py +10 -10
  68. mindspore/{parallel/nn/layers.py → dataset/debug/__init__.py} +7 -8
  69. mindspore/dataset/debug/debug_hook.py +65 -0
  70. mindspore/dataset/debug/pre_defined_hook.py +67 -0
  71. mindspore/dataset/engine/__init__.py +7 -3
  72. mindspore/dataset/engine/cache_client.py +9 -9
  73. mindspore/dataset/engine/datasets.py +648 -477
  74. mindspore/dataset/engine/datasets_audio.py +165 -167
  75. mindspore/dataset/engine/datasets_standard_format.py +93 -67
  76. mindspore/dataset/engine/datasets_text.py +492 -342
  77. mindspore/dataset/engine/datasets_user_defined.py +85 -50
  78. mindspore/dataset/engine/datasets_vision.py +1224 -699
  79. mindspore/dataset/engine/graphdata.py +134 -69
  80. mindspore/dataset/engine/iterators.py +50 -9
  81. mindspore/dataset/engine/offload.py +52 -31
  82. mindspore/dataset/engine/samplers.py +27 -24
  83. mindspore/dataset/engine/serializer_deserializer.py +14 -15
  84. mindspore/dataset/engine/validators.py +213 -52
  85. mindspore/dataset/text/__init__.py +10 -8
  86. mindspore/dataset/text/transforms.py +152 -57
  87. mindspore/dataset/text/utils.py +98 -49
  88. mindspore/dataset/text/validators.py +25 -0
  89. mindspore/dataset/transforms/__init__.py +4 -2
  90. mindspore/dataset/transforms/c_transforms.py +11 -13
  91. mindspore/dataset/transforms/py_transforms.py +2 -2
  92. mindspore/dataset/transforms/py_transforms_util.py +10 -0
  93. mindspore/dataset/transforms/transforms.py +13 -15
  94. mindspore/dataset/transforms/validators.py +7 -7
  95. mindspore/dataset/utils/__init__.py +2 -1
  96. mindspore/dataset/utils/browse_dataset.py +13 -13
  97. mindspore/dataset/utils/line_reader.py +121 -0
  98. mindspore/dataset/vision/__init__.py +8 -7
  99. mindspore/dataset/vision/c_transforms.py +125 -126
  100. mindspore/dataset/vision/py_transforms.py +37 -37
  101. mindspore/dataset/vision/py_transforms_util.py +23 -20
  102. mindspore/dataset/vision/transforms.py +316 -315
  103. mindspore/dataset/vision/utils.py +313 -17
  104. mindspore/dataset/vision/validators.py +6 -6
  105. mindspore/default_config.py +0 -1
  106. mindspore/{compression → experimental}/__init__.py +6 -5
  107. mindspore/experimental/map_parameter.py +275 -0
  108. mindspore/include/OWNERS +0 -1
  109. mindspore/include/api/callback/callback.h +9 -13
  110. mindspore/include/api/callback/ckpt_saver.h +2 -2
  111. mindspore/include/api/callback/loss_monitor.h +2 -2
  112. mindspore/include/api/callback/lr_scheduler.h +5 -5
  113. mindspore/include/api/callback/time_monitor.h +2 -2
  114. mindspore/include/api/callback/train_accuracy.h +4 -6
  115. mindspore/include/api/cfg.h +19 -6
  116. mindspore/include/api/context.h +70 -9
  117. mindspore/include/api/delegate.h +8 -1
  118. mindspore/include/api/dual_abi_helper.h +8 -24
  119. mindspore/include/api/metrics/accuracy.h +2 -2
  120. mindspore/include/api/metrics/metrics.h +4 -3
  121. mindspore/include/api/model.h +9 -4
  122. mindspore/include/api/model_group.h +68 -0
  123. mindspore/include/api/model_parallel_runner.h +17 -17
  124. mindspore/include/api/net.h +12 -11
  125. mindspore/include/api/serialization.h +20 -4
  126. mindspore/include/api/status.h +7 -1
  127. mindspore/include/api/types.h +25 -21
  128. mindspore/include/api/visible.h +4 -0
  129. mindspore/include/c_api/model_c.h +5 -0
  130. mindspore/include/c_api/status_c.h +1 -1
  131. mindspore/include/dataset/config.h +1 -1
  132. mindspore/include/dataset/constants.h +14 -0
  133. mindspore/include/dataset/text.h +59 -0
  134. mindspore/include/dataset/vision.h +56 -117
  135. mindspore/include/dataset/vision_lite.h +102 -0
  136. mindspore/include/mindapi/base/type_id.h +42 -3
  137. mindspore/lib/libdnnl.so.2 +0 -0
  138. mindspore/lib/libicudata.so.69 +0 -0
  139. mindspore/lib/libicui18n.so.69 +0 -0
  140. mindspore/lib/libicuuc.so.69 +0 -0
  141. mindspore/lib/libmindspore.so +0 -0
  142. mindspore/lib/libmindspore_backend.so +0 -0
  143. mindspore/lib/libmindspore_common.so +0 -0
  144. mindspore/lib/libmindspore_core.so +0 -0
  145. mindspore/lib/libmindspore_glog.so.0 +0 -0
  146. mindspore/lib/libmindspore_gpr.so.15 +0 -0
  147. mindspore/lib/libmindspore_grpc++.so.1 +0 -0
  148. mindspore/lib/libmindspore_grpc.so.15 +0 -0
  149. mindspore/lib/libmindspore_shared_lib.so +0 -0
  150. mindspore/lib/libmpi_adapter.so +0 -0
  151. mindspore/lib/libmpi_collective.so +0 -0
  152. mindspore/lib/libnnacl.so +0 -0
  153. mindspore/lib/libopencv_core.so.4.5 +0 -0
  154. mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
  155. mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
  156. mindspore/lib/libps_cache.so +0 -0
  157. mindspore/lib/plugin/ascend/libakg.so +0 -0
  158. mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
  159. mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
  160. mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
  161. mindspore/lib/plugin/ascend/libmindspore_aicpu_kernels.so +0 -0
  162. mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
  163. mindspore/lib/{libakg.so → plugin/cpu/libakg.so} +0 -0
  164. mindspore/lib/plugin/libmindspore_ascend.so.1 +0 -0
  165. mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
  166. mindspore/log.py +28 -28
  167. mindspore/mindrecord/common/exceptions.py +2 -4
  168. mindspore/mindrecord/filereader.py +19 -1
  169. mindspore/mindrecord/filewriter.py +250 -88
  170. mindspore/mindrecord/mindpage.py +13 -13
  171. mindspore/mindrecord/shardheader.py +15 -15
  172. mindspore/mindrecord/shardreader.py +9 -0
  173. mindspore/mindrecord/shardwriter.py +29 -29
  174. mindspore/mindrecord/tools/cifar100_to_mr.py +9 -9
  175. mindspore/mindrecord/tools/cifar10_to_mr.py +9 -9
  176. mindspore/mindrecord/tools/csv_to_mr.py +4 -4
  177. mindspore/mindrecord/tools/imagenet_to_mr.py +70 -65
  178. mindspore/mindrecord/tools/mnist_to_mr.py +41 -41
  179. mindspore/mindrecord/tools/tfrecord_to_mr.py +6 -6
  180. mindspore/nn/__init__.py +1 -5
  181. mindspore/nn/cell.py +297 -234
  182. mindspore/nn/dynamic_lr.py +1 -1
  183. mindspore/nn/grad/cell_grad.py +17 -42
  184. mindspore/nn/layer/__init__.py +7 -4
  185. mindspore/nn/layer/activation.py +131 -88
  186. mindspore/nn/layer/basic.py +313 -613
  187. mindspore/nn/layer/channel_shuffle.py +103 -0
  188. mindspore/nn/layer/combined.py +1 -1
  189. mindspore/nn/layer/container.py +52 -6
  190. mindspore/nn/layer/conv.py +112 -43
  191. mindspore/nn/layer/dense.py +10 -9
  192. mindspore/nn/layer/embedding.py +36 -34
  193. mindspore/nn/layer/image.py +123 -27
  194. mindspore/nn/layer/math.py +108 -107
  195. mindspore/nn/layer/normalization.py +212 -366
  196. mindspore/nn/layer/padding.py +370 -42
  197. mindspore/nn/layer/pooling.py +1443 -219
  198. mindspore/nn/layer/rnn_cells.py +11 -16
  199. mindspore/nn/layer/rnns.py +38 -39
  200. mindspore/nn/layer/thor_layer.py +24 -25
  201. mindspore/nn/layer/timedistributed.py +5 -5
  202. mindspore/nn/layer/transformer.py +701 -0
  203. mindspore/nn/learning_rate_schedule.py +8 -8
  204. mindspore/nn/loss/__init__.py +9 -6
  205. mindspore/nn/loss/loss.py +678 -142
  206. mindspore/nn/metrics.py +53 -0
  207. mindspore/nn/optim/_dist_optimizer_registry.py +2 -2
  208. mindspore/nn/optim/ada_grad.py +8 -8
  209. mindspore/nn/optim/adadelta.py +2 -3
  210. mindspore/nn/optim/adafactor.py +18 -14
  211. mindspore/nn/optim/adam.py +429 -87
  212. mindspore/nn/optim/adamax.py +5 -6
  213. mindspore/nn/optim/adasum.py +10 -8
  214. mindspore/nn/optim/asgd.py +7 -7
  215. mindspore/nn/optim/ftrl.py +81 -11
  216. mindspore/nn/optim/lamb.py +7 -8
  217. mindspore/nn/optim/lars.py +4 -4
  218. mindspore/nn/optim/lazyadam.py +82 -7
  219. mindspore/nn/optim/momentum.py +8 -7
  220. mindspore/nn/optim/optimizer.py +19 -10
  221. mindspore/nn/optim/proximal_ada_grad.py +6 -5
  222. mindspore/nn/optim/rmsprop.py +3 -3
  223. mindspore/nn/optim/rprop.py +20 -16
  224. mindspore/nn/optim/sgd.py +21 -15
  225. mindspore/nn/optim/thor.py +23 -21
  226. mindspore/nn/probability/__init__.py +0 -2
  227. mindspore/nn/probability/bijector/bijector.py +7 -6
  228. mindspore/nn/probability/bijector/invert.py +4 -2
  229. mindspore/nn/probability/bijector/softplus.py +2 -2
  230. mindspore/nn/probability/bnn_layers/dense_variational.py +1 -1
  231. mindspore/nn/probability/bnn_layers/layer_distribution.py +2 -2
  232. mindspore/nn/probability/distribution/__init__.py +6 -0
  233. mindspore/nn/probability/distribution/_utils/custom_ops.py +3 -2
  234. mindspore/nn/probability/distribution/_utils/utils.py +11 -17
  235. mindspore/nn/probability/distribution/bernoulli.py +6 -6
  236. mindspore/nn/probability/distribution/beta.py +1 -1
  237. mindspore/nn/probability/distribution/categorical.py +9 -9
  238. mindspore/nn/probability/distribution/cauchy.py +8 -8
  239. mindspore/nn/probability/distribution/distribution.py +12 -6
  240. mindspore/nn/probability/distribution/exponential.py +5 -5
  241. mindspore/nn/probability/distribution/gamma.py +3 -3
  242. mindspore/nn/probability/distribution/geometric.py +6 -5
  243. mindspore/nn/probability/distribution/gumbel.py +5 -5
  244. mindspore/nn/probability/distribution/half_normal.py +133 -0
  245. mindspore/nn/probability/distribution/laplace.py +128 -0
  246. mindspore/nn/probability/distribution/log_normal.py +0 -1
  247. mindspore/nn/probability/distribution/logistic.py +4 -5
  248. mindspore/nn/probability/distribution/normal.py +11 -15
  249. mindspore/nn/probability/distribution/poisson.py +6 -2
  250. mindspore/nn/probability/distribution/student_t.py +150 -0
  251. mindspore/nn/probability/distribution/transformed_distribution.py +4 -4
  252. mindspore/nn/probability/distribution/uniform.py +5 -5
  253. mindspore/nn/reinforcement/_tensors_queue.py +3 -3
  254. mindspore/nn/reinforcement/tensor_array.py +2 -2
  255. mindspore/nn/sparse/sparse.py +8 -1
  256. mindspore/nn/wrap/cell_wrapper.py +55 -27
  257. mindspore/nn/wrap/grad_reducer.py +20 -11
  258. mindspore/nn/wrap/loss_scale.py +47 -30
  259. mindspore/numpy/array_creations.py +33 -22
  260. mindspore/numpy/array_ops.py +46 -42
  261. mindspore/numpy/logic_ops.py +6 -27
  262. mindspore/numpy/math_ops.py +26 -19
  263. mindspore/numpy/utils.py +1 -8
  264. mindspore/numpy/utils_const.py +112 -62
  265. mindspore/ops/__init__.py +6 -3
  266. mindspore/ops/_constants.py +0 -6
  267. mindspore/ops/_grad/__init__.py +2 -1
  268. mindspore/ops/_grad/grad_array_ops.py +209 -152
  269. mindspore/ops/_grad/grad_base.py +55 -17
  270. mindspore/ops/_grad/grad_clip_ops.py +11 -3
  271. mindspore/ops/_grad/grad_comm_ops.py +58 -47
  272. mindspore/ops/_grad/grad_implementations.py +21 -61
  273. mindspore/ops/_grad/grad_inner_ops.py +48 -6
  274. mindspore/ops/_grad/grad_math_ops.py +306 -161
  275. mindspore/ops/_grad/grad_nn_ops.py +192 -181
  276. mindspore/ops/_grad/grad_other_ops.py +1 -1
  277. mindspore/ops/_grad/grad_quant_ops.py +5 -5
  278. mindspore/ops/_grad/grad_sequence_ops.py +296 -0
  279. mindspore/ops/_grad/grad_sparse.py +15 -9
  280. mindspore/ops/_grad_experimental/__init__.py +1 -0
  281. mindspore/ops/_grad_experimental/grad_array_ops.py +441 -55
  282. mindspore/ops/_grad_experimental/grad_image_ops.py +25 -7
  283. mindspore/ops/_grad_experimental/grad_inner_ops.py +3 -44
  284. mindspore/ops/_grad_experimental/grad_linalg_ops.py +16 -21
  285. mindspore/ops/_grad_experimental/grad_math_ops.py +979 -49
  286. mindspore/ops/_grad_experimental/grad_nn_ops.py +78 -8
  287. mindspore/ops/_grad_experimental/grad_scalar_ops.py +112 -0
  288. mindspore/ops/_grad_experimental/grad_sparse_ops.py +197 -13
  289. mindspore/ops/_op_impl/__init__.py +3 -3
  290. mindspore/ops/_op_impl/_custom_op/__init__.py +0 -1
  291. mindspore/ops/_op_impl/_custom_op/_basic.py +0 -1
  292. mindspore/ops/_op_impl/_custom_op/batch_matmul_impl.py +1 -1
  293. mindspore/ops/_op_impl/_custom_op/batchnorm_fold.py +4 -2
  294. mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py +2 -2
  295. mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py +2 -2
  296. mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py +5 -5
  297. mindspore/ops/_op_impl/_custom_op/batchnorm_fold_grad.py +3 -3
  298. mindspore/ops/_op_impl/_custom_op/cholesky_trsm_impl.py +1 -1
  299. mindspore/ops/_op_impl/_custom_op/correction_mul.py +3 -3
  300. mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py +2 -2
  301. mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +4 -8
  302. mindspore/ops/_op_impl/_custom_op/dsd_impl.py +1 -1
  303. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py +2 -2
  304. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py +2 -2
  305. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py +2 -2
  306. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py +2 -2
  307. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py +2 -2
  308. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py +2 -2
  309. mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py +2 -2
  310. mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py +2 -2
  311. mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py +2 -2
  312. mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py +2 -2
  313. mindspore/ops/_op_impl/_custom_op/fused_abs_max1_impl.py +1 -1
  314. mindspore/ops/_op_impl/_custom_op/img2col_impl.py +1 -1
  315. mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
  316. mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_right_impl.py +1 -1
  317. mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py +1 -1
  318. mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py +1 -1
  319. mindspore/ops/_op_impl/_custom_op/matmul_cube_impl.py +2 -2
  320. mindspore/ops/_op_impl/_custom_op/matmul_dds_grad_impl.py +0 -1
  321. mindspore/ops/_op_impl/_custom_op/matmul_dds_impl.py +0 -1
  322. mindspore/ops/_op_impl/_custom_op/matrix_combine_impl.py +1 -1
  323. mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py +2 -2
  324. mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py +2 -2
  325. mindspore/ops/_op_impl/_custom_op/transpose02314_impl.py +1 -1
  326. mindspore/ops/_op_impl/aicpu/__init__.py +238 -3
  327. mindspore/ops/_op_impl/aicpu/abs.py +36 -0
  328. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d.py +34 -0
  329. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d_grad.py +34 -0
  330. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d.py +39 -0
  331. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d_grad.py +39 -0
  332. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_2d_grad.py +37 -0
  333. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d.py +42 -0
  334. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d_grad.py +152 -0
  335. mindspore/ops/_op_impl/aicpu/add.py +43 -0
  336. mindspore/ops/_op_impl/aicpu/addcdiv.py +0 -32
  337. mindspore/ops/_op_impl/aicpu/addcmul.py +0 -84
  338. mindspore/ops/_op_impl/aicpu/affine_grid_grad.py +35 -0
  339. mindspore/ops/_op_impl/aicpu/arg_max.py +75 -0
  340. mindspore/ops/_op_impl/aicpu/arg_min.py +75 -0
  341. mindspore/ops/_op_impl/aicpu/argmin_with_value.py +43 -0
  342. mindspore/ops/_op_impl/aicpu/batch_matmul.py +43 -0
  343. mindspore/ops/_op_impl/aicpu/batch_norm_grad_grad.py +49 -0
  344. mindspore/ops/_op_impl/aicpu/bernoulli.py +48 -0
  345. mindspore/ops/_op_impl/aicpu/bessel_i0.py +31 -0
  346. mindspore/ops/_op_impl/aicpu/bias_add.py +44 -0
  347. mindspore/ops/_op_impl/aicpu/bias_add_grad.py +43 -0
  348. mindspore/ops/_op_impl/aicpu/bincount.py +33 -0
  349. mindspore/{nn/probability/infer/variational/__init__.py → ops/_op_impl/aicpu/cauchy.py} +17 -10
  350. mindspore/ops/_op_impl/aicpu/channel_shuffle.py +40 -0
  351. mindspore/ops/_op_impl/aicpu/cholesky.py +1 -1
  352. mindspore/ops/_op_impl/{cpu/bias_add.py → aicpu/choleskygrad.py} +9 -7
  353. mindspore/ops/_op_impl/aicpu/combined_non_max_suppression.py +42 -0
  354. mindspore/ops/_op_impl/aicpu/concat_offset.py +42 -0
  355. mindspore/ops/_op_impl/aicpu/concat_offset_v1.py +31 -0
  356. mindspore/ops/_op_impl/aicpu/conj.py +11 -0
  357. mindspore/ops/_op_impl/aicpu/crop_and_resize_grad_image.py +38 -0
  358. mindspore/ops/_op_impl/aicpu/cumulative_logsumexp.py +36 -0
  359. mindspore/ops/_op_impl/aicpu/deformable_offsets.py +38 -0
  360. mindspore/ops/_op_impl/aicpu/deformable_offsets_grad.py +2 -2
  361. mindspore/ops/_op_impl/aicpu/dense_to_sparse_set_operation.py +48 -0
  362. mindspore/ops/_op_impl/aicpu/diag.py +36 -0
  363. mindspore/ops/_op_impl/aicpu/diag_part.py +36 -0
  364. mindspore/ops/_op_impl/aicpu/diagonal.py +35 -0
  365. mindspore/ops/_op_impl/{cpu/bias_add_grad.py → aicpu/digamma.py} +9 -7
  366. mindspore/ops/_op_impl/aicpu/eig.py +35 -0
  367. mindspore/ops/_op_impl/aicpu/fft_with_size.py +41 -0
  368. mindspore/ops/_op_impl/aicpu/flatten.py +1 -0
  369. mindspore/ops/_op_impl/aicpu/fmax.py +36 -0
  370. mindspore/ops/_op_impl/aicpu/fmin.py +37 -0
  371. mindspore/ops/_op_impl/aicpu/fractional_max_pool3d_with_fixed_ksize.py +1 -1
  372. mindspore/ops/_op_impl/aicpu/fse_decode.py +43 -0
  373. mindspore/ops/_op_impl/aicpu/glu.py +33 -0
  374. mindspore/ops/_op_impl/aicpu/glu_grad.py +34 -0
  375. mindspore/ops/_op_impl/aicpu/greater.py +41 -0
  376. mindspore/ops/_op_impl/aicpu/greater_equal.py +41 -0
  377. mindspore/ops/_op_impl/aicpu/index_put.py +50 -0
  378. mindspore/ops/_op_impl/{tbe/scatter_add_ds.py → aicpu/inplace_index_add.py} +17 -21
  379. mindspore/ops/_op_impl/aicpu/instance_norm_v2.py +41 -0
  380. mindspore/ops/_op_impl/aicpu/instance_norm_v2_grad.py +44 -0
  381. mindspore/ops/_op_impl/aicpu/layer_norm_grad_grad.py +47 -0
  382. mindspore/ops/_op_impl/aicpu/less.py +41 -0
  383. mindspore/ops/_op_impl/aicpu/less_equal.py +41 -0
  384. mindspore/ops/_op_impl/aicpu/lgamma.py +32 -0
  385. mindspore/ops/_op_impl/aicpu/log_normal_reverse.py +33 -0
  386. mindspore/ops/_op_impl/aicpu/logit.py +33 -0
  387. mindspore/ops/_op_impl/aicpu/logit_grad.py +34 -0
  388. mindspore/ops/_op_impl/aicpu/masked_fill.py +42 -0
  389. mindspore/ops/_op_impl/aicpu/masked_scatter.py +39 -0
  390. mindspore/ops/_op_impl/aicpu/matmul.py +39 -0
  391. mindspore/ops/_op_impl/aicpu/matrix_logarithm.py +31 -0
  392. mindspore/ops/_op_impl/aicpu/matrix_power.py +32 -0
  393. mindspore/ops/_op_impl/aicpu/matrix_solve_ls.py +36 -0
  394. mindspore/ops/_op_impl/aicpu/matrix_triangular_solve.py +36 -0
  395. mindspore/ops/_op_impl/aicpu/mirror_pad.py +2 -0
  396. mindspore/ops/_op_impl/aicpu/mirror_pad_grad.py +0 -4
  397. mindspore/ops/_op_impl/aicpu/mul.py +3 -1
  398. mindspore/ops/_op_impl/aicpu/multinomial.py +14 -6
  399. mindspore/ops/_op_impl/aicpu/multinomial_with_replacement.py +35 -0
  400. mindspore/ops/_op_impl/aicpu/nan_to_num.py +34 -0
  401. mindspore/ops/_op_impl/aicpu/nllloss.py +38 -0
  402. mindspore/ops/_op_impl/aicpu/nllloss_grad.py +39 -0
  403. mindspore/ops/_op_impl/aicpu/ones_like.py +0 -2
  404. mindspore/ops/_op_impl/aicpu/polar.py +32 -0
  405. mindspore/ops/_op_impl/aicpu/polygamma.py +34 -0
  406. mindspore/ops/_op_impl/aicpu/qr.py +36 -0
  407. mindspore/ops/_op_impl/aicpu/quant_dtype_cast.py +40 -0
  408. mindspore/ops/_op_impl/aicpu/quantile.py +35 -0
  409. mindspore/ops/_op_impl/aicpu/ragged_tensor_to_sparse.py +73 -0
  410. mindspore/ops/_op_impl/aicpu/ragged_tensor_to_tensor.py +74 -0
  411. mindspore/ops/_op_impl/aicpu/random_shuffle.py +3 -0
  412. mindspore/ops/_op_impl/aicpu/randperm_v2.py +41 -0
  413. mindspore/ops/_op_impl/aicpu/range.py +36 -0
  414. mindspore/ops/_op_impl/aicpu/reciprocal.py +34 -0
  415. mindspore/ops/_op_impl/aicpu/reciprocal_grad.py +35 -0
  416. mindspore/ops/_op_impl/aicpu/reduce_sum.py +57 -0
  417. mindspore/ops/_op_impl/aicpu/resize_bicubic.py +2 -8
  418. mindspore/ops/_op_impl/aicpu/resize_bicubic_grad.py +1 -1
  419. mindspore/ops/_op_impl/aicpu/resize_v2.py +68 -0
  420. mindspore/ops/_op_impl/aicpu/resize_v2_grad.py +68 -0
  421. mindspore/ops/_op_impl/aicpu/scatter_elements.py +4 -0
  422. mindspore/ops/_op_impl/aicpu/scatter_nd_update.py +2 -0
  423. mindspore/ops/_op_impl/aicpu/search_sorted.py +12 -6
  424. mindspore/ops/_op_impl/aicpu/self_adjoint_eig.py +34 -0
  425. mindspore/ops/_op_impl/aicpu/sequence_add.py +34 -0
  426. mindspore/ops/_op_impl/aicpu/sequence_add_offset.py +34 -0
  427. mindspore/ops/_op_impl/aicpu/sequence_addn.py +38 -0
  428. mindspore/ops/_op_impl/aicpu/slice_grad.py +76 -0
  429. mindspore/ops/_op_impl/aicpu/smooth_l1_loss.py +35 -0
  430. mindspore/ops/_op_impl/aicpu/smooth_l1_loss_grad.py +37 -0
  431. mindspore/ops/_op_impl/aicpu/sort.py +39 -0
  432. mindspore/ops/_op_impl/aicpu/sparse_apply_adagrad_da.py +0 -24
  433. mindspore/ops/_op_impl/aicpu/sparse_cross.py +42 -0
  434. mindspore/ops/_op_impl/aicpu/sparse_fill_empty_rows.py +63 -0
  435. mindspore/ops/_op_impl/aicpu/sparse_fill_empty_rows_grad.py +45 -0
  436. mindspore/ops/_op_impl/aicpu/sparse_matrix_mat_mul.py +56 -0
  437. mindspore/ops/_op_impl/{tbe/slice_ds.py → aicpu/sparse_segment_sum.py} +16 -24
  438. mindspore/ops/_op_impl/aicpu/sparse_segment_sum_with_num_segments.py +68 -0
  439. mindspore/ops/_op_impl/aicpu/sparse_slice.py +63 -0
  440. mindspore/ops/_op_impl/aicpu/sparse_slice_grad.py +61 -0
  441. mindspore/ops/_op_impl/aicpu/squared_difference.py +2 -0
  442. mindspore/ops/_op_impl/aicpu/strided_slice_v2.py +93 -0
  443. mindspore/ops/_op_impl/aicpu/strided_slice_v2_grad.py +66 -0
  444. mindspore/ops/_op_impl/aicpu/tensor_scatter_update.py +59 -0
  445. mindspore/ops/_op_impl/{tbe/gather_v2.py → aicpu/tile.py} +24 -24
  446. mindspore/ops/_op_impl/aicpu/tridiagonal_solve.py +35 -0
  447. mindspore/ops/_op_impl/aicpu/tril_indices.py +34 -0
  448. mindspore/ops/_op_impl/aicpu/triu_indices.py +34 -0
  449. mindspore/ops/_op_impl/aicpu/uniform.py +34 -0
  450. mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +1 -0
  451. mindspore/ops/_op_impl/aicpu/unique_consecutive.py +10 -2
  452. mindspore/ops/_op_impl/cpu/__init__.py +1 -2
  453. mindspore/ops/_op_impl/cpu/dynamic_shape.py +5 -1
  454. mindspore/ops/_op_impl/cpu/maximum_grad.py +2 -0
  455. mindspore/{compression/common/__init__.py → ops/_op_impl/cpu/pyexecute.py} +13 -8
  456. mindspore/ops/_op_impl/cpu/reduce_sum.py +8 -0
  457. mindspore/ops/_op_impl/cpu/sparse_slice.py +62 -0
  458. mindspore/ops/_op_impl/cpu/sparse_slice_grad.py +60 -0
  459. mindspore/ops/_op_impl/cpu/tensor_shape.py +5 -1
  460. mindspore/ops/_op_impl/tbe/__init__.py +27 -608
  461. mindspore/ops/_op_impl/tbe/addcdiv_ds.py +42 -0
  462. mindspore/ops/_op_impl/tbe/addcmul_ds.py +44 -0
  463. mindspore/ops/_op_impl/tbe/assign_add_ds.py +1 -0
  464. mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
  465. mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +1 -1
  466. mindspore/ops/_op_impl/tbe/basic_lstm_cell_c_state_grad_v2.py +0 -1
  467. mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
  468. mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +1 -1
  469. mindspore/ops/_op_impl/tbe/batch_to_space_nd_v2.py +41 -0
  470. mindspore/ops/_op_impl/tbe/bce_with_logits_loss.py +1 -0
  471. mindspore/ops/_op_impl/tbe/bias_add_grad.py +2 -0
  472. mindspore/ops/_op_impl/tbe/bn_infer_grad.py +4 -2
  473. mindspore/ops/_op_impl/tbe/bn_infer_grad_ds.py +40 -0
  474. mindspore/ops/_op_impl/tbe/bn_training_update.py +0 -1
  475. mindspore/ops/_op_impl/tbe/bn_training_update_ds.py +0 -1
  476. mindspore/ops/_op_impl/tbe/broadcast_to_ds.py +6 -4
  477. mindspore/ops/_op_impl/tbe/cast.py +0 -2
  478. mindspore/ops/_op_impl/tbe/cast_ds.py +3 -3
  479. mindspore/ops/_op_impl/tbe/ctc_loss_v2.py +0 -2
  480. mindspore/ops/_op_impl/tbe/ctc_loss_v2_grad.py +0 -2
  481. mindspore/ops/_op_impl/tbe/data_format_dim_map_ds.py +1 -0
  482. mindspore/ops/_op_impl/tbe/deformable_offsets.py +1 -0
  483. mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +1 -1
  484. mindspore/ops/_op_impl/tbe/dynamic_atomic_addr_clean.py +1 -1
  485. mindspore/ops/_op_impl/tbe/gather_nd.py +1 -0
  486. mindspore/ops/_op_impl/tbe/greater.py +2 -0
  487. mindspore/ops/_op_impl/tbe/{index_add.py → inplace_index_add.py} +3 -6
  488. mindspore/ops/_op_impl/tbe/layer_norm_beta_gamma_backprop_v2.py +0 -1
  489. mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py +35 -0
  490. mindspore/ops/_op_impl/tbe/npu_get_float_status_v2.py +35 -0
  491. mindspore/ops/_op_impl/tbe/one_hot_ds.py +0 -6
  492. mindspore/ops/_op_impl/tbe/{greater_ds.py → reduce_all_ds.py} +13 -16
  493. mindspore/ops/_op_impl/tbe/reduce_any_ds.py +39 -0
  494. mindspore/ops/_op_impl/tbe/roi_align_ds.py +44 -0
  495. mindspore/ops/_op_impl/tbe/roi_align_grad_ds.py +44 -0
  496. mindspore/ops/_op_impl/tbe/scatter_add.py +2 -0
  497. mindspore/ops/_op_impl/tbe/scatter_nd_add.py +2 -2
  498. mindspore/ops/_op_impl/tbe/slice.py +26 -15
  499. mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
  500. mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +1 -1
  501. mindspore/ops/_op_impl/tbe/strided_slice_grad_d.py +1 -0
  502. mindspore/ops/_op_impl/tbe/trans_data_ds.py +15 -5
  503. mindspore/ops/_op_impl/tbe/unsorted_segment_sum.py +1 -1
  504. mindspore/ops/_op_impl/tbe/unsorted_segment_sum_ds.py +2 -0
  505. mindspore/ops/_primitive_cache.py +3 -2
  506. mindspore/ops/_register_for_op.py +11 -0
  507. mindspore/ops/_utils/__init__.py +1 -1
  508. mindspore/ops/_utils/utils.py +20 -41
  509. mindspore/ops/_vmap/__init__.py +2 -2
  510. mindspore/ops/_vmap/vmap_array_ops.py +170 -78
  511. mindspore/ops/_vmap/vmap_base.py +24 -10
  512. mindspore/ops/_vmap/vmap_convolution_ops.py +7 -10
  513. mindspore/ops/_vmap/vmap_grad_math_ops.py +4 -4
  514. mindspore/ops/_vmap/vmap_grad_nn_ops.py +41 -9
  515. mindspore/ops/_vmap/vmap_image_ops.py +52 -0
  516. mindspore/ops/_vmap/vmap_math_ops.py +77 -6
  517. mindspore/ops/_vmap/vmap_nn_ops.py +78 -29
  518. mindspore/ops/_vmap/vmap_other_ops.py +3 -1
  519. mindspore/ops/_vmap/vmap_random_ops.py +55 -3
  520. mindspore/ops/_vmap/vmap_sparse_ops.py +1 -0
  521. mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
  522. mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
  523. mindspore/ops/bprop_mindir/ApproximateEqual_bprop.mindir +18 -19
  524. mindspore/ops/bprop_mindir/Argmax_bprop.mindir +13 -12
  525. mindspore/ops/bprop_mindir/Argmin_bprop.mindir +14 -13
  526. mindspore/ops/bprop_mindir/AssignSub_bprop.mindir +17 -18
  527. mindspore/ops/bprop_mindir/Assign_bprop.mindir +16 -16
  528. mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +150 -0
  529. mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +66 -0
  530. mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
  531. mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +13 -12
  532. mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
  533. mindspore/ops/bprop_mindir/BatchToSpaceND_bprop.mindir +28 -0
  534. mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
  535. mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +33 -0
  536. mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +306 -0
  537. mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +12 -8
  538. mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
  539. mindspore/ops/bprop_mindir/Concat_bprop.mindir +0 -0
  540. mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +240 -0
  541. mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +247 -0
  542. mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +247 -0
  543. mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +315 -0
  544. mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +278 -0
  545. mindspore/ops/bprop_mindir/DType_bprop.mindir +12 -12
  546. mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +58 -0
  547. mindspore/ops/bprop_mindir/Depend_bprop.mindir +12 -13
  548. mindspore/ops/bprop_mindir/DepthToSpace_bprop.mindir +23 -0
  549. mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +138 -0
  550. mindspore/ops/bprop_mindir/DiagPart_bprop.mindir +15 -0
  551. mindspore/ops/bprop_mindir/Dropout2D_bprop.mindir +0 -0
  552. mindspore/ops/bprop_mindir/Dropout3D_bprop.mindir +0 -0
  553. mindspore/ops/bprop_mindir/DropoutDoMask_bprop.mindir +22 -24
  554. mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +16 -14
  555. mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +27 -0
  556. mindspore/ops/bprop_mindir/Dropout_bprop.mindir +0 -0
  557. mindspore/ops/bprop_mindir/DynamicGRUV2_bprop.mindir +0 -0
  558. mindspore/ops/bprop_mindir/DynamicRNN_bprop.mindir +0 -0
  559. mindspore/ops/bprop_mindir/DynamicShape_bprop.mindir +12 -12
  560. mindspore/ops/bprop_mindir/Elu_bprop.mindir +16 -0
  561. mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
  562. mindspore/ops/bprop_mindir/Equal_bprop.mindir +18 -19
  563. mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +58 -0
  564. mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +16 -0
  565. mindspore/ops/bprop_mindir/Flatten_bprop.mindir +54 -0
  566. mindspore/ops/bprop_mindir/FloorDiv_bprop.mindir +18 -15
  567. mindspore/ops/bprop_mindir/GatherD_bprop.mindir +26 -0
  568. mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +57 -0
  569. mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
  570. mindspore/ops/bprop_mindir/GreaterEqual_bprop.mindir +17 -18
  571. mindspore/ops/bprop_mindir/Greater_bprop.mindir +18 -19
  572. mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +16 -0
  573. mindspore/ops/bprop_mindir/HSwish_bprop.mindir +16 -0
  574. mindspore/ops/bprop_mindir/IOU_bprop.mindir +18 -19
  575. mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
  576. mindspore/ops/bprop_mindir/IsFinite_bprop.mindir +13 -12
  577. mindspore/ops/bprop_mindir/IsInf_bprop.mindir +13 -10
  578. mindspore/ops/bprop_mindir/IsNan_bprop.mindir +14 -11
  579. mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +126 -0
  580. mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +15 -0
  581. mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +30 -0
  582. mindspore/ops/bprop_mindir/LRN_bprop.mindir +43 -0
  583. mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
  584. mindspore/ops/bprop_mindir/LessEqual_bprop.mindir +18 -19
  585. mindspore/ops/bprop_mindir/Less_bprop.mindir +17 -18
  586. mindspore/ops/bprop_mindir/LinSpace_bprop.mindir +22 -19
  587. mindspore/ops/bprop_mindir/Load_bprop.mindir +12 -13
  588. mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +23 -0
  589. mindspore/ops/bprop_mindir/LogicalAnd_bprop.mindir +17 -18
  590. mindspore/ops/bprop_mindir/LogicalNot_bprop.mindir +14 -13
  591. mindspore/ops/bprop_mindir/MaskedSelect_bprop.mindir +21 -0
  592. mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +74 -0
  593. mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +74 -0
  594. mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +75 -0
  595. mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +65 -0
  596. mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
  597. mindspore/ops/bprop_mindir/Maximum_bprop.mindir +0 -0
  598. mindspore/ops/bprop_mindir/Minimum_bprop.mindir +0 -0
  599. mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +27 -0
  600. mindspore/ops/bprop_mindir/Mish_bprop.mindir +35 -0
  601. mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
  602. mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
  603. mindspore/ops/bprop_mindir/NonZero_bprop.mindir +14 -0
  604. mindspore/ops/bprop_mindir/NotEqual_bprop.mindir +18 -19
  605. mindspore/ops/bprop_mindir/OneHot_bprop.mindir +25 -23
  606. mindspore/ops/bprop_mindir/OnesLike_bprop.mindir +13 -13
  607. mindspore/ops/bprop_mindir/PReLU_bprop.mindir +0 -0
  608. mindspore/ops/bprop_mindir/Pad_bprop.mindir +0 -0
  609. mindspore/ops/bprop_mindir/Padding_bprop.mindir +0 -0
  610. mindspore/ops/bprop_mindir/RNNTLoss_bprop.mindir +29 -0
  611. mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +82 -0
  612. mindspore/ops/bprop_mindir/Range_bprop.mindir +21 -19
  613. mindspore/ops/bprop_mindir/Rank_bprop.mindir +11 -11
  614. mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +16 -0
  615. mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
  616. mindspore/ops/bprop_mindir/ReduceAll_bprop.mindir +18 -17
  617. mindspore/ops/bprop_mindir/ReduceAny_bprop.mindir +18 -17
  618. mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +19 -23
  619. mindspore/ops/bprop_mindir/Reshape_bprop.mindir +60 -0
  620. mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +29 -0
  621. mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +89 -0
  622. mindspore/ops/bprop_mindir/ReverseSequence_bprop.mindir +52 -0
  623. mindspore/ops/bprop_mindir/ReverseV2_bprop.mindir +22 -0
  624. mindspore/ops/bprop_mindir/Round_bprop.mindir +14 -13
  625. mindspore/ops/bprop_mindir/ScatterMax_bprop.mindir +0 -0
  626. mindspore/ops/bprop_mindir/ScatterMin_bprop.mindir +0 -0
  627. mindspore/ops/bprop_mindir/ScatterNdUpdate_bprop.mindir +22 -0
  628. mindspore/ops/bprop_mindir/ScatterNd_bprop.mindir +24 -0
  629. mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +22 -0
  630. mindspore/ops/bprop_mindir/ScatterUpdate_bprop.mindir +0 -0
  631. mindspore/ops/bprop_mindir/SeLU_bprop.mindir +21 -0
  632. mindspore/ops/bprop_mindir/Select_bprop.mindir +30 -34
  633. mindspore/ops/bprop_mindir/Shape_bprop.mindir +12 -12
  634. mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +21 -0
  635. mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
  636. mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +16 -0
  637. mindspore/ops/bprop_mindir/Sign_bprop.mindir +13 -12
  638. mindspore/ops/bprop_mindir/Slice_bprop.mindir +26 -0
  639. mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +36 -0
  640. mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
  641. mindspore/ops/bprop_mindir/Softplus_bprop.mindir +16 -0
  642. mindspore/ops/bprop_mindir/Softsign_bprop.mindir +33 -0
  643. mindspore/ops/bprop_mindir/Sort_bprop.mindir +0 -0
  644. mindspore/ops/bprop_mindir/SpaceToBatchND_bprop.mindir +28 -0
  645. mindspore/ops/bprop_mindir/SpaceToDepth_bprop.mindir +23 -0
  646. mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
  647. mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
  648. mindspore/ops/bprop_mindir/Split_bprop.mindir +22 -0
  649. mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +54 -0
  650. mindspore/ops/bprop_mindir/StridedSliceGrad_bprop.mindir +95 -0
  651. mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +98 -0
  652. mindspore/ops/bprop_mindir/Switch_bprop.mindir +28 -32
  653. mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
  654. mindspore/ops/bprop_mindir/Tanh_bprop.mindir +66 -0
  655. mindspore/ops/bprop_mindir/TensorScatterAdd_bprop.mindir +22 -0
  656. mindspore/ops/bprop_mindir/TensorScatterUpdate_bprop.mindir +29 -0
  657. mindspore/ops/bprop_mindir/TensorShape_bprop.mindir +14 -0
  658. mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
  659. mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
  660. mindspore/ops/bprop_mindir/TransShape_bprop.mindir +23 -0
  661. mindspore/ops/bprop_mindir/TruncateDiv_bprop.mindir +18 -15
  662. mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +11 -13
  663. mindspore/ops/bprop_mindir/Unique_bprop.mindir +16 -0
  664. mindspore/ops/bprop_mindir/Unstack_bprop.mindir +22 -0
  665. mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +32 -0
  666. mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +38 -0
  667. mindspore/ops/bprop_mindir/ZerosLike_bprop.mindir +13 -12
  668. mindspore/ops/bprop_mindir/__init__.py +1 -4
  669. mindspore/ops/bprop_mindir/generate_mindir.py +32 -20
  670. mindspore/ops/composite/__init__.py +12 -13
  671. mindspore/ops/composite/base.py +261 -254
  672. mindspore/ops/composite/env_ops.py +41 -0
  673. mindspore/ops/composite/math_ops.py +197 -156
  674. mindspore/ops/composite/multitype_ops/_compile_utils.py +428 -176
  675. mindspore/ops/composite/multitype_ops/_constexpr_utils.py +188 -87
  676. mindspore/ops/composite/multitype_ops/add_impl.py +23 -1
  677. mindspore/ops/composite/multitype_ops/div_impl.py +3 -3
  678. mindspore/ops/composite/multitype_ops/equal_impl.py +1 -0
  679. mindspore/ops/composite/multitype_ops/floordiv_impl.py +1 -1
  680. mindspore/ops/composite/multitype_ops/getitem_impl.py +52 -5
  681. mindspore/ops/composite/multitype_ops/greater_equal_impl.py +31 -0
  682. mindspore/ops/composite/multitype_ops/greater_impl.py +31 -0
  683. mindspore/ops/composite/multitype_ops/in_impl.py +15 -3
  684. mindspore/ops/composite/multitype_ops/less_equal_impl.py +33 -2
  685. mindspore/ops/composite/multitype_ops/less_impl.py +33 -0
  686. mindspore/ops/composite/multitype_ops/logical_and_impl.py +2 -2
  687. mindspore/ops/composite/multitype_ops/logical_or_impl.py +2 -1
  688. mindspore/ops/composite/multitype_ops/mod_impl.py +1 -1
  689. mindspore/ops/composite/multitype_ops/mul_impl.py +21 -7
  690. mindspore/ops/composite/multitype_ops/not_in_impl.py +15 -3
  691. mindspore/ops/composite/multitype_ops/ones_like_impl.py +2 -4
  692. mindspore/ops/composite/multitype_ops/pow_impl.py +1 -0
  693. mindspore/ops/composite/multitype_ops/setitem_impl.py +62 -70
  694. mindspore/ops/composite/multitype_ops/sub_impl.py +3 -3
  695. mindspore/ops/composite/multitype_ops/zeros_like_impl.py +41 -4
  696. mindspore/ops/function/__init__.py +323 -8
  697. mindspore/ops/function/array_func.py +3511 -780
  698. mindspore/ops/function/clip_func.py +329 -0
  699. mindspore/ops/function/debug_func.py +6 -6
  700. mindspore/ops/function/grad/__init__.py +5 -1
  701. mindspore/ops/function/grad/grad_func.py +736 -65
  702. mindspore/ops/function/image_func.py +270 -0
  703. mindspore/ops/function/linalg_func.py +268 -8
  704. mindspore/ops/function/math_func.py +8032 -3164
  705. mindspore/ops/function/nn_func.py +5619 -1855
  706. mindspore/ops/function/other_func.py +115 -0
  707. mindspore/ops/function/parameter_func.py +11 -10
  708. mindspore/ops/function/random_func.py +939 -77
  709. mindspore/ops/function/sparse_func.py +249 -84
  710. mindspore/ops/function/sparse_unary_func.py +2303 -0
  711. mindspore/ops/function/spectral_func.py +146 -0
  712. mindspore/ops/function/vmap_func.py +114 -0
  713. mindspore/ops/functional.py +182 -254
  714. mindspore/ops/op_info_register.py +79 -34
  715. mindspore/ops/operations/__init__.py +210 -118
  716. mindspore/ops/operations/_csr_ops.py +7 -7
  717. mindspore/ops/operations/_embedding_cache_ops.py +25 -15
  718. mindspore/ops/operations/_grad_ops.py +447 -322
  719. mindspore/ops/operations/_inner_ops.py +547 -176
  720. mindspore/ops/operations/_map_tensor_ops.py +112 -0
  721. mindspore/ops/operations/_ms_kernel.py +29 -27
  722. mindspore/ops/operations/_ocr_ops.py +11 -11
  723. mindspore/ops/operations/_opaque_predicate_registry.py +41 -0
  724. mindspore/ops/operations/_quant_ops.py +186 -101
  725. mindspore/ops/operations/_rl_inner_ops.py +122 -61
  726. mindspore/ops/operations/_scalar_ops.py +466 -0
  727. mindspore/ops/operations/_sequence_ops.py +1047 -0
  728. mindspore/ops/operations/_tensor_array.py +10 -11
  729. mindspore/ops/operations/_thor_ops.py +4 -4
  730. mindspore/ops/operations/array_ops.py +1428 -1226
  731. mindspore/ops/operations/comm_ops.py +180 -117
  732. mindspore/ops/operations/control_ops.py +4 -2
  733. mindspore/ops/operations/custom_ops.py +185 -98
  734. mindspore/ops/operations/debug_ops.py +92 -54
  735. mindspore/ops/operations/image_ops.py +406 -211
  736. mindspore/ops/operations/inner_ops.py +42 -53
  737. mindspore/ops/operations/linalg_ops.py +32 -29
  738. mindspore/ops/operations/math_ops.py +2076 -897
  739. mindspore/ops/operations/nn_ops.py +1282 -1252
  740. mindspore/ops/operations/other_ops.py +124 -278
  741. mindspore/ops/operations/random_ops.py +345 -178
  742. mindspore/ops/operations/rl_ops.py +8 -9
  743. mindspore/ops/operations/sparse_ops.py +502 -157
  744. mindspore/ops/operations/spectral_ops.py +107 -0
  745. mindspore/ops/primitive.py +192 -15
  746. mindspore/ops/vm_impl_registry.py +23 -2
  747. mindspore/parallel/__init__.py +6 -1
  748. mindspore/parallel/_auto_parallel_context.py +199 -92
  749. mindspore/parallel/_cell_wrapper.py +4 -2
  750. mindspore/parallel/_cost_model_context.py +3 -0
  751. mindspore/parallel/_dp_allreduce_fusion.py +2 -1
  752. mindspore/parallel/_offload_context.py +185 -0
  753. mindspore/parallel/_parallel_serialization.py +167 -28
  754. mindspore/parallel/_ps_context.py +9 -5
  755. mindspore/parallel/_recovery_context.py +1 -1
  756. mindspore/parallel/_tensor.py +9 -1
  757. mindspore/{nn/transformer → parallel/_transformer}/__init__.py +6 -6
  758. mindspore/{nn/transformer → parallel/_transformer}/layers.py +59 -37
  759. mindspore/{nn/transformer → parallel/_transformer}/loss.py +4 -7
  760. mindspore/{nn/transformer → parallel/_transformer}/moe.py +160 -35
  761. mindspore/{nn/transformer → parallel/_transformer}/op_parallel_config.py +3 -3
  762. mindspore/{nn/transformer → parallel/_transformer}/transformer.py +235 -196
  763. mindspore/parallel/_utils.py +47 -7
  764. mindspore/parallel/algo_parameter_config.py +5 -1
  765. mindspore/parallel/checkpoint_transform.py +329 -0
  766. mindspore/parallel/shard.py +229 -0
  767. mindspore/profiler/__init__.py +2 -1
  768. mindspore/profiler/common/util.py +4 -3
  769. mindspore/profiler/common/validator/validate_path.py +2 -2
  770. mindspore/profiler/envprofiling.py +249 -0
  771. mindspore/profiler/parser/aicpu_data_parser.py +38 -39
  772. mindspore/profiler/parser/ascend_timeline_generator.py +497 -0
  773. mindspore/profiler/parser/base_timeline_generator.py +471 -0
  774. mindspore/profiler/parser/cpu_gpu_timeline_generator.py +684 -0
  775. mindspore/profiler/parser/framework_parser.py +42 -16
  776. mindspore/profiler/parser/hccl_parser.py +158 -158
  777. mindspore/profiler/parser/hwts_log_parser.py +7 -6
  778. mindspore/profiler/parser/integrator.py +18 -1579
  779. mindspore/profiler/parser/minddata_analyzer.py +8 -8
  780. mindspore/profiler/parser/msadvisor_analyzer.py +14 -27
  781. mindspore/profiler/parser/msadvisor_parser.py +2 -4
  782. mindspore/profiler/parser/optime_parser.py +17 -18
  783. mindspore/profiler/parser/profiler_info.py +108 -0
  784. mindspore/profiler/parser/step_trace_parser.py +1 -1
  785. mindspore/profiler/profiling.py +396 -194
  786. mindspore/rewrite/__init__.py +6 -2
  787. mindspore/rewrite/api/node.py +51 -110
  788. mindspore/rewrite/api/node_type.py +10 -6
  789. mindspore/rewrite/api/pattern_engine.py +51 -7
  790. mindspore/rewrite/api/scoped_value.py +64 -53
  791. mindspore/rewrite/api/symbol_tree.py +108 -61
  792. mindspore/rewrite/api/tree_node_helper.py +2 -3
  793. mindspore/{compression/quant/__init__.py → rewrite/ast_creator_register.py} +20 -11
  794. mindspore/rewrite/ast_helpers/__init__.py +6 -3
  795. mindspore/rewrite/ast_helpers/ast_creator.py +115 -0
  796. mindspore/rewrite/ast_helpers/ast_finder.py +99 -1
  797. mindspore/rewrite/ast_helpers/ast_modifier.py +17 -4
  798. mindspore/rewrite/ast_helpers/ast_replacer.py +1 -1
  799. mindspore/rewrite/ast_transformers/__init__.py +0 -1
  800. mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +46 -5
  801. mindspore/rewrite/ast_transformers/remove_return_out_of_if.py +6 -3
  802. mindspore/rewrite/common/__init__.py +2 -0
  803. mindspore/rewrite/common/event.py +1 -1
  804. mindspore/rewrite/common/observable.py +1 -1
  805. mindspore/rewrite/common/observer.py +1 -1
  806. mindspore/rewrite/common/rewrite_elog.py +35 -0
  807. mindspore/rewrite/namer.py +2 -2
  808. mindspore/rewrite/namespace.py +14 -4
  809. mindspore/rewrite/node.py +161 -13
  810. mindspore/rewrite/parser.py +0 -1
  811. mindspore/rewrite/parser_register.py +0 -1
  812. mindspore/rewrite/parsers/arguments_parser.py +3 -2
  813. mindspore/rewrite/parsers/assign_parser.py +267 -67
  814. mindspore/rewrite/parsers/attribute_parser.py +56 -0
  815. mindspore/rewrite/parsers/class_def_parser.py +191 -108
  816. mindspore/rewrite/parsers/constant_parser.py +101 -0
  817. mindspore/rewrite/parsers/container_parser.py +88 -0
  818. mindspore/rewrite/parsers/for_parser.py +28 -15
  819. mindspore/rewrite/parsers/function_def_parser.py +21 -5
  820. mindspore/rewrite/parsers/if_parser.py +11 -28
  821. mindspore/rewrite/parsers/module_parser.py +9 -6
  822. mindspore/rewrite/parsers/return_parser.py +3 -2
  823. mindspore/rewrite/sparsify/__init__.py +0 -0
  824. mindspore/rewrite/sparsify/sparse_transformer.py +448 -0
  825. mindspore/rewrite/sparsify/sparsify.py +109 -0
  826. mindspore/rewrite/sparsify/utils.py +173 -0
  827. mindspore/rewrite/symbol_tree.py +322 -109
  828. mindspore/rewrite/symbol_tree_builder.py +45 -8
  829. mindspore/rewrite/symbol_tree_dumper.py +0 -1
  830. mindspore/rewrite/topological_manager.py +1 -2
  831. mindspore/run_check/_check_version.py +209 -112
  832. mindspore/run_check/run_check.py +2 -1
  833. mindspore/scipy/linalg.py +13 -117
  834. mindspore/scipy/ops.py +5 -71
  835. mindspore/scipy/ops_grad.py +1 -25
  836. mindspore/scipy/ops_wrapper.py +1 -1
  837. mindspore/scipy/optimize/_bfgs.py +1 -1
  838. mindspore/scipy/optimize/_lagrange.py +200 -0
  839. mindspore/scipy/optimize/line_search.py +3 -2
  840. mindspore/scipy/optimize/minimize.py +43 -6
  841. mindspore/scipy/sparse/__init__.py +2 -2
  842. mindspore/scipy/sparse/linalg.py +5 -465
  843. mindspore/scipy/utils.py +2 -1
  844. mindspore/scipy/utils_const.py +7 -1
  845. mindspore/train/__init__.py +6 -4
  846. mindspore/train/_utils.py +28 -5
  847. mindspore/train/amp.py +321 -50
  848. mindspore/train/callback/__init__.py +3 -1
  849. mindspore/train/callback/_backup_and_restore.py +120 -0
  850. mindspore/train/callback/_callback.py +8 -8
  851. mindspore/train/callback/_checkpoint.py +12 -9
  852. mindspore/train/callback/_early_stop.py +13 -7
  853. mindspore/train/callback/_history.py +8 -8
  854. mindspore/train/callback/_lambda_callback.py +6 -6
  855. mindspore/train/callback/_landscape.py +36 -38
  856. mindspore/train/callback/_loss_monitor.py +12 -6
  857. mindspore/train/callback/_lr_scheduler_callback.py +2 -4
  858. mindspore/train/callback/_on_request_exit.py +212 -0
  859. mindspore/train/callback/_reduce_lr_on_plateau.py +13 -7
  860. mindspore/train/callback/_summary_collector.py +27 -19
  861. mindspore/train/callback/_time_monitor.py +13 -7
  862. mindspore/train/checkpoint_pb2.py +68 -8
  863. mindspore/train/data_sink.py +122 -33
  864. mindspore/train/dataset_helper.py +28 -87
  865. mindspore/train/loss_scale_manager.py +4 -7
  866. mindspore/{nn → train}/metrics/__init__.py +20 -20
  867. mindspore/{nn → train}/metrics/accuracy.py +12 -10
  868. mindspore/{nn → train}/metrics/auc.py +4 -4
  869. mindspore/{nn → train}/metrics/bleu_score.py +4 -4
  870. mindspore/{nn → train}/metrics/confusion_matrix.py +10 -8
  871. mindspore/{nn → train}/metrics/cosine_similarity.py +4 -4
  872. mindspore/{nn → train}/metrics/dice.py +6 -5
  873. mindspore/{nn → train}/metrics/error.py +7 -5
  874. mindspore/{nn → train}/metrics/fbeta.py +9 -7
  875. mindspore/{nn → train}/metrics/hausdorff_distance.py +8 -6
  876. mindspore/{nn → train}/metrics/loss.py +4 -3
  877. mindspore/{nn → train}/metrics/mean_surface_distance.py +6 -5
  878. mindspore/{nn → train}/metrics/metric.py +6 -5
  879. mindspore/{nn → train}/metrics/occlusion_sensitivity.py +4 -3
  880. mindspore/{nn → train}/metrics/perplexity.py +5 -4
  881. mindspore/{nn → train}/metrics/precision.py +5 -4
  882. mindspore/{nn → train}/metrics/recall.py +5 -4
  883. mindspore/{nn → train}/metrics/roc.py +7 -6
  884. mindspore/{nn → train}/metrics/root_mean_square_surface_distance.py +6 -5
  885. mindspore/{nn → train}/metrics/topk.py +7 -5
  886. mindspore/train/mind_ir_pb2.py +339 -32
  887. mindspore/train/model.py +113 -84
  888. mindspore/train/serialization.py +547 -167
  889. mindspore/train/summary/_summary_adapter.py +1 -1
  890. mindspore/train/summary/summary_record.py +43 -12
  891. mindspore/train/train_thor/convert_utils.py +7 -1
  892. mindspore/train/train_thor/dataset_helper.py +3 -3
  893. mindspore/train/train_thor/model_thor.py +0 -4
  894. mindspore/version.py +1 -1
  895. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/METADATA +4 -3
  896. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/RECORD +899 -675
  897. mindspore/compression/common/constant.py +0 -124
  898. mindspore/compression/export/__init__.py +0 -19
  899. mindspore/compression/export/quant_export.py +0 -514
  900. mindspore/compression/quant/qat.py +0 -636
  901. mindspore/compression/quant/quant_utils.py +0 -462
  902. mindspore/compression/quant/quantizer.py +0 -68
  903. mindspore/nn/layer/quant.py +0 -1868
  904. mindspore/nn/layer/rnn_utils.py +0 -90
  905. mindspore/nn/probability/dpn/__init__.py +0 -22
  906. mindspore/nn/probability/dpn/vae/__init__.py +0 -25
  907. mindspore/nn/probability/dpn/vae/cvae.py +0 -138
  908. mindspore/nn/probability/dpn/vae/vae.py +0 -122
  909. mindspore/nn/probability/infer/__init__.py +0 -22
  910. mindspore/nn/probability/infer/variational/elbo.py +0 -70
  911. mindspore/nn/probability/infer/variational/svi.py +0 -84
  912. mindspore/nn/probability/toolbox/__init__.py +0 -22
  913. mindspore/nn/probability/toolbox/anomaly_detection.py +0 -99
  914. mindspore/nn/probability/toolbox/uncertainty_evaluation.py +0 -363
  915. mindspore/nn/probability/transforms/__init__.py +0 -22
  916. mindspore/nn/probability/transforms/transform_bnn.py +0 -262
  917. mindspore/nn/probability/zhusuan/__init__.py +0 -18
  918. mindspore/nn/probability/zhusuan/framework/__init__.py +0 -18
  919. mindspore/nn/probability/zhusuan/framework/bn.py +0 -95
  920. mindspore/nn/probability/zhusuan/variational/__init__.py +0 -18
  921. mindspore/nn/probability/zhusuan/variational/elbo.py +0 -46
  922. mindspore/ops/_op_impl/tbe/bias_add_grad_ds.py +0 -52
  923. mindspore/ops/_op_impl/tbe/scatter_nd_add_ds.py +0 -43
  924. mindspore/ops/bprop_mindir/AssignAdd_bprop.mindir +0 -20
  925. mindspore/ops/bprop_mindir/Identity_bprop.mindir +0 -9
  926. mindspore/ops/bprop_mindir/LogicalOr_bprop.mindir +0 -20
  927. mindspore/ops/bprop_mindir/ReLU_bprop.mindir +0 -16
  928. mindspore/ops/bprop_mindir/UpdateState_bprop.mindir +0 -17
  929. mindspore/ops/bprop_mindir/stop_gradient_bprop.mindir +0 -12
  930. mindspore/ops/composite/array_ops.py +0 -210
  931. mindspore/ops/composite/clip_ops.py +0 -238
  932. mindspore/ops/composite/random_ops.py +0 -426
  933. mindspore/ops/composite/vmap_ops.py +0 -38
  934. mindspore/ops/operations/sponge_ops.py +0 -3531
  935. mindspore/ops/operations/sponge_update_ops.py +0 -2546
  936. mindspore/parallel/nn/__init__.py +0 -42
  937. mindspore/parallel/nn/loss.py +0 -22
  938. mindspore/parallel/nn/moe.py +0 -21
  939. mindspore/parallel/nn/op_parallel_config.py +0 -22
  940. mindspore/parallel/nn/transformer.py +0 -31
  941. mindspore/run_check/_check_deps_version.py +0 -84
  942. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/WHEEL +0 -0
  943. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/entry_points.txt +0 -0
  944. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/top_level.txt +0 -0
mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir ADDED
@@ -0,0 +1,315 @@
1
+
2
+ 0.1.1 MindSpore*2.0.0:�_
3
+ �+get_bprop_conv3d_transpose.718:[CNode]719:1+get_bprop_conv3d_transpose.718:[CNode]719:1"REF::bprop.720:Default/bprop.720-op571
4
+ C[ValueNode]732[ValueNode]732"Constant*
5
+ value* out_channel�
6
+ C[ValueNode]733[ValueNode]733"Constant*
7
+ value* kernel_size�
8
+ <[ValueNode]734[ValueNode]734"Constant*
9
+ value*mode�
10
+ @[ValueNode]735[ValueNode]735"Constant*
11
+ value*pad_mode�
12
+ ;[ValueNode]736[ValueNode]736"Constant*
13
+ value*pad�
14
+ >[ValueNode]737[ValueNode]737"Constant*
15
+ value*stride�
16
+ @[ValueNode]738[ValueNode]738"Constant*
17
+ value*dilation�
18
+ =[ValueNode]739[ValueNode]739"Constant*
19
+ value*group�
20
+ C[ValueNode]740[ValueNode]740"Constant*
21
+ value* data_format�
22
+ �
23
+ [ValueNode]732
24
+ [ValueNode]733
25
+ [ValueNode]734
26
+ [ValueNode]735
27
+ [ValueNode]736
28
+ [ValueNode]737
29
+ [ValueNode]738
30
+ [ValueNode]739
31
+ [ValueNode]740,get_bprop_conv3d_transpose.718:[CNode]741:10,get_bprop_conv3d_transpose.718:[CNode]741:10"REF::S-Prim-MakeTuple:11:Default/S-Prim-MakeTuple-op578
32
+ B[ValueNode]742[ValueNode]742"Constant*
33
+ value*
34
+ in_channel�
35
+ �
36
+ #get_bprop_conv3d_transpose.718:self
37
+ [ValueNode]742,get_bprop_conv3d_transpose.718:[CNode]743:12,get_bprop_conv3d_transpose.718:[CNode]743:12"REF::getattr:13:Default/getattr-op579
38
+ C[ValueNode]744[ValueNode]744"Constant*
39
+ value* kernel_size�
40
+ �
41
+ #get_bprop_conv3d_transpose.718:self
42
+ [ValueNode]744,get_bprop_conv3d_transpose.718:[CNode]745:14,get_bprop_conv3d_transpose.718:[CNode]745:14"REF::getattr:13:Default/getattr-op580
43
+ <[ValueNode]746[ValueNode]746"Constant*
44
+ value*mode�
45
+ �
46
+ #get_bprop_conv3d_transpose.718:self
47
+ [ValueNode]746,get_bprop_conv3d_transpose.718:[CNode]747:15,get_bprop_conv3d_transpose.718:[CNode]747:15"REF::getattr:13:Default/getattr-op581
48
+ E[ValueNode]748[ValueNode]748"Constant*
49
+ value*
50
+ �
51
+ #get_bprop_conv3d_transpose.718:self
52
+ [ValueNode]748,get_bprop_conv3d_transpose.718:[CNode]749:16,get_bprop_conv3d_transpose.718:[CNode]749:16"REF::getattr:13:Default/getattr-op582
53
+ �,get_bprop_conv3d_transpose.718:[CNode]750:17,get_bprop_conv3d_transpose.718:[CNode]750:17"1REF::get_bprop_conv3d_transpose.718:[CNode]749:16:583
54
+ @[ValueNode]751[ValueNode]751"Constant*
55
+ value*pad_list�
56
+ �
57
+ ,get_bprop_conv3d_transpose.718:[CNode]750:17
58
+ [ValueNode]751*get_bprop_conv3d_transpose.718:pad_list:18*get_bprop_conv3d_transpose.718:pad_list:18"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op584
59
+ >[ValueNode]752[ValueNode]752"Constant*
60
+ value*stride�
61
+ �
62
+ #get_bprop_conv3d_transpose.718:self
63
+ [ValueNode]752,get_bprop_conv3d_transpose.718:[CNode]753:19,get_bprop_conv3d_transpose.718:[CNode]753:19"REF::getattr:13:Default/getattr-op585
64
+ 8[ValueNode]754[ValueNode]754"Constant*
65
+ value�
66
+ �
67
+ ,get_bprop_conv3d_transpose.718:[CNode]753:19
68
+ [ValueNode]754,get_bprop_conv3d_transpose.718:[CNode]755:20,get_bprop_conv3d_transpose.718:[CNode]755:20"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op586
69
+ >[ValueNode]756[ValueNode]756"Constant*
70
+ value*stride�
71
+ �
72
+ #get_bprop_conv3d_transpose.718:self
73
+ [ValueNode]756,get_bprop_conv3d_transpose.718:[CNode]757:21,get_bprop_conv3d_transpose.718:[CNode]757:21"REF::getattr:13:Default/getattr-op587
74
+ 8[ValueNode]758[ValueNode]758"Constant*
75
+ value�
76
+ �
77
+ ,get_bprop_conv3d_transpose.718:[CNode]757:21
78
+ [ValueNode]758,get_bprop_conv3d_transpose.718:[CNode]759:22,get_bprop_conv3d_transpose.718:[CNode]759:22"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op588
79
+ >[ValueNode]760[ValueNode]760"Constant*
80
+ value*stride�
81
+ �
82
+ #get_bprop_conv3d_transpose.718:self
83
+ [ValueNode]760,get_bprop_conv3d_transpose.718:[CNode]761:23,get_bprop_conv3d_transpose.718:[CNode]761:23"REF::getattr:13:Default/getattr-op589
84
+ 8[ValueNode]762[ValueNode]762"Constant*
85
+ value�
86
+ �
87
+ ,get_bprop_conv3d_transpose.718:[CNode]761:23
88
+ [ValueNode]762,get_bprop_conv3d_transpose.718:[CNode]763:24,get_bprop_conv3d_transpose.718:[CNode]763:24"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op590
89
+ �
90
+ ,get_bprop_conv3d_transpose.718:[CNode]755:20
91
+ ,get_bprop_conv3d_transpose.718:[CNode]759:22
92
+ ,get_bprop_conv3d_transpose.718:[CNode]763:24(get_bprop_conv3d_transpose.718:stride:25(get_bprop_conv3d_transpose.718:stride:25"REF::S-Prim-MakeTuple:26:Default/S-Prim-MakeTuple-op591
93
+ @[ValueNode]764[ValueNode]764"Constant*
94
+ value*dilation�
95
+ �
96
+ #get_bprop_conv3d_transpose.718:self
97
+ [ValueNode]764,get_bprop_conv3d_transpose.718:[CNode]765:27,get_bprop_conv3d_transpose.718:[CNode]765:27"REF::getattr:13:Default/getattr-op592
98
+ 8[ValueNode]766[ValueNode]766"Constant*
99
+ value�
100
+ �
101
+ ,get_bprop_conv3d_transpose.718:[CNode]765:27
102
+ [ValueNode]766,get_bprop_conv3d_transpose.718:[CNode]767:28,get_bprop_conv3d_transpose.718:[CNode]767:28"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op593
103
+ @[ValueNode]768[ValueNode]768"Constant*
104
+ value*dilation�
105
+ �
106
+ #get_bprop_conv3d_transpose.718:self
107
+ [ValueNode]768,get_bprop_conv3d_transpose.718:[CNode]769:29,get_bprop_conv3d_transpose.718:[CNode]769:29"REF::getattr:13:Default/getattr-op594
108
+ 8[ValueNode]770[ValueNode]770"Constant*
109
+ value�
110
+ �
111
+ ,get_bprop_conv3d_transpose.718:[CNode]769:29
112
+ [ValueNode]770,get_bprop_conv3d_transpose.718:[CNode]771:30,get_bprop_conv3d_transpose.718:[CNode]771:30"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op595
113
+ @[ValueNode]772[ValueNode]772"Constant*
114
+ value*dilation�
115
+ �
116
+ #get_bprop_conv3d_transpose.718:self
117
+ [ValueNode]772,get_bprop_conv3d_transpose.718:[CNode]773:31,get_bprop_conv3d_transpose.718:[CNode]773:31"REF::getattr:13:Default/getattr-op596
118
+ 8[ValueNode]774[ValueNode]774"Constant*
119
+ value�
120
+ �
121
+ ,get_bprop_conv3d_transpose.718:[CNode]773:31
122
+ [ValueNode]774,get_bprop_conv3d_transpose.718:[CNode]775:32,get_bprop_conv3d_transpose.718:[CNode]775:32"REF::MetaFuncGraph::getitem:Default/S-Prim-getitem-op597
123
+ �
124
+ ,get_bprop_conv3d_transpose.718:[CNode]767:28
125
+ ,get_bprop_conv3d_transpose.718:[CNode]771:30
126
+ ,get_bprop_conv3d_transpose.718:[CNode]775:32*get_bprop_conv3d_transpose.718:dilation:33*get_bprop_conv3d_transpose.718:dilation:33"REF::S-Prim-MakeTuple:34:Default/S-Prim-MakeTuple-op598
127
+ =[ValueNode]776[ValueNode]776"Constant*
128
+ value*group�
129
+ �
130
+ #get_bprop_conv3d_transpose.718:self
131
+ [ValueNode]776,get_bprop_conv3d_transpose.718:[CNode]777:35,get_bprop_conv3d_transpose.718:[CNode]777:35"REF::getattr:13:Default/getattr-op599
132
+ C[ValueNode]778[ValueNode]778"Constant*
133
+ value* data_format�
134
+ �
135
+ #get_bprop_conv3d_transpose.718:self
136
+ [ValueNode]778,get_bprop_conv3d_transpose.718:[CNode]779:36,get_bprop_conv3d_transpose.718:[CNode]779:36"REF::getattr:13:Default/getattr-op600
137
+ ;[ValueNode]780[ValueNode]780"Constant*
138
+ value*pad�
139
+ �
140
+ ,get_bprop_conv3d_transpose.718:[CNode]743:12
141
+ ,get_bprop_conv3d_transpose.718:[CNode]745:14
142
+ ,get_bprop_conv3d_transpose.718:[CNode]747:15
143
+ [ValueNode]780
144
+ *get_bprop_conv3d_transpose.718:pad_list:18
145
+ (get_bprop_conv3d_transpose.718:stride:25
146
+ *get_bprop_conv3d_transpose.718:dilation:33
147
+ ,get_bprop_conv3d_transpose.718:[CNode]777:35
148
+ ,get_bprop_conv3d_transpose.718:[CNode]779:36,get_bprop_conv3d_transpose.718:[CNode]781:37,get_bprop_conv3d_transpose.718:[CNode]781:37"REF::S-Prim-MakeTuple:38:Default/S-Prim-MakeTuple-op601
149
+ �
150
+ ,get_bprop_conv3d_transpose.718:[CNode]741:10
151
+ ,get_bprop_conv3d_transpose.718:[CNode]781:37,get_bprop_conv3d_transpose.718:[CNode]782:39,get_bprop_conv3d_transpose.718:[CNode]782:39"REF::S-Prim-make_dict:40:Default/S-Prim-make_dict-op602
152
+ H783783"Constant*2
153
+ value*&mindspore.ops.operations.nn_ops.Conv3D�!
154
+ �
155
+ 783
156
+ ,get_bprop_conv3d_transpose.718:[CNode]782:39,get_bprop_conv3d_transpose.718:input_grad:41,get_bprop_conv3d_transpose.718:input_grad:41"REF::MetaFuncGraph::unpack_call:Default/unpack_call.784-op603
157
+ C[ValueNode]785[ValueNode]785"Constant*
158
+ value* out_channel�
159
+ C[ValueNode]786[ValueNode]786"Constant*
160
+ value* kernel_size�
161
+ <[ValueNode]787[ValueNode]787"Constant*
162
+ value*mode�
163
+ @[ValueNode]788[ValueNode]788"Constant*
164
+ value*pad_mode�
165
+ ;[ValueNode]789[ValueNode]789"Constant*
166
+ value*pad�
167
+ >[ValueNode]790[ValueNode]790"Constant*
168
+ value*stride�
169
+ @[ValueNode]791[ValueNode]791"Constant*
170
+ value*dilation�
171
+ =[ValueNode]792[ValueNode]792"Constant*
172
+ value*group�
173
+ C[ValueNode]793[ValueNode]793"Constant*
174
+ value* data_format�
175
+ �
176
+ [ValueNode]785
177
+ [ValueNode]786
178
+ [ValueNode]787
179
+ [ValueNode]788
180
+ [ValueNode]789
181
+ [ValueNode]790
182
+ [ValueNode]791
183
+ [ValueNode]792
184
+ [ValueNode]793,get_bprop_conv3d_transpose.718:[CNode]794:43,get_bprop_conv3d_transpose.718:[CNode]794:43"REF::S-Prim-MakeTuple:44:Default/S-Prim-MakeTuple-op605
185
+ B[ValueNode]795[ValueNode]795"Constant*
186
+ value*
187
+ in_channel�
188
+ �
189
+ #get_bprop_conv3d_transpose.718:self
190
+ [ValueNode]795,get_bprop_conv3d_transpose.718:[CNode]796:45,get_bprop_conv3d_transpose.718:[CNode]796:45"REF::getattr:13:Default/getattr-op606
191
+ C[ValueNode]797[ValueNode]797"Constant*
192
+ value* kernel_size�
193
+ �
194
+ #get_bprop_conv3d_transpose.718:self
195
+ [ValueNode]797,get_bprop_conv3d_transpose.718:[CNode]798:46,get_bprop_conv3d_transpose.718:[CNode]798:46"REF::getattr:13:Default/getattr-op607
196
+ <[ValueNode]799[ValueNode]799"Constant*
197
+ value*mode�
198
+ �
199
+ #get_bprop_conv3d_transpose.718:self
200
+ [ValueNode]799,get_bprop_conv3d_transpose.718:[CNode]800:47,get_bprop_conv3d_transpose.718:[CNode]800:47"REF::getattr:13:Default/getattr-op608
201
+ >[ValueNode]801[ValueNode]801"Constant*
202
+ value*stride�
203
+ �
204
+ #get_bprop_conv3d_transpose.718:self
205
+ [ValueNode]801,get_bprop_conv3d_transpose.718:[CNode]802:48,get_bprop_conv3d_transpose.718:[CNode]802:48"REF::getattr:13:Default/getattr-op609
206
+ @[ValueNode]803[ValueNode]803"Constant*
207
+ value*dilation�
208
+ �
209
+ #get_bprop_conv3d_transpose.718:self
210
+ [ValueNode]803,get_bprop_conv3d_transpose.718:[CNode]804:49,get_bprop_conv3d_transpose.718:[CNode]804:49"REF::getattr:13:Default/getattr-op610
211
+ =[ValueNode]805[ValueNode]805"Constant*
212
+ value*group�
213
+ �
214
+ #get_bprop_conv3d_transpose.718:self
215
+ [ValueNode]805,get_bprop_conv3d_transpose.718:[CNode]806:50,get_bprop_conv3d_transpose.718:[CNode]806:50"REF::getattr:13:Default/getattr-op611
216
+ C[ValueNode]807[ValueNode]807"Constant*
217
+ value* data_format�
218
+ �
219
+ #get_bprop_conv3d_transpose.718:self
220
+ [ValueNode]807,get_bprop_conv3d_transpose.718:[CNode]808:51,get_bprop_conv3d_transpose.718:[CNode]808:51"REF::getattr:13:Default/getattr-op612
221
+ ;[ValueNode]809[ValueNode]809"Constant*
222
+ value*pad�
223
+ �
224
+ ,get_bprop_conv3d_transpose.718:[CNode]796:45
225
+ ,get_bprop_conv3d_transpose.718:[CNode]798:46
226
+ ,get_bprop_conv3d_transpose.718:[CNode]800:47
227
+ [ValueNode]809
228
+ *get_bprop_conv3d_transpose.718:pad_list:18
229
+ ,get_bprop_conv3d_transpose.718:[CNode]802:48
230
+ ,get_bprop_conv3d_transpose.718:[CNode]804:49
231
+ ,get_bprop_conv3d_transpose.718:[CNode]806:50
232
+ ,get_bprop_conv3d_transpose.718:[CNode]808:51,get_bprop_conv3d_transpose.718:[CNode]810:52,get_bprop_conv3d_transpose.718:[CNode]810:52"REF::S-Prim-MakeTuple:53:Default/S-Prim-MakeTuple-op613
233
+ �
234
+ ,get_bprop_conv3d_transpose.718:[CNode]794:43
235
+ ,get_bprop_conv3d_transpose.718:[CNode]810:52,get_bprop_conv3d_transpose.718:[CNode]811:54,get_bprop_conv3d_transpose.718:[CNode]811:54"REF::S-Prim-make_dict:55:Default/S-Prim-make_dict-op614
236
+ Y812812"Constant*C
237
+ value*7mindspore.ops.operations._grad_ops.Conv3DBackpropFilter�!
238
+ �
239
+ 812
240
+ ,get_bprop_conv3d_transpose.718:[CNode]811:54-get_bprop_conv3d_transpose.718:filter_grad:56-get_bprop_conv3d_transpose.718:filter_grad:56"REF::MetaFuncGraph::unpack_call:Default/unpack_call.813-op615
241
+ �/get_bprop_conv3d_transpose.718:get_dyn_shape:64/get_bprop_conv3d_transpose.718:get_dyn_shape:64">REF::ClassType::mindspore.ops.operations.array_ops.TensorShape:TDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.TensorShape'-op621get_bprop_conv3d_transpose.718*%
242
+ #get_bprop_conv3d_transpose.718:self*"
243
+ get_bprop_conv3d_transpose.718:x*"
244
+ get_bprop_conv3d_transpose.718:w*$
245
+ "get_bprop_conv3d_transpose.718:out*%
246
+ #get_bprop_conv3d_transpose.718:dout2-
247
+ +get_bprop_conv3d_transpose.718:[CNode]719:1:@6f53e4cf9d4faa542e3d4a39f9f8a939715c510ba393e1ceb2cff4ddd2d81729J/grad_nn_ops.pyB�
248
+ {
249
+ get_bprop_conv3d_transpose.718:wbprop.720:[CNode]721:2bprop.720:[CNode]721:2"REF::shape.722:Default/shape.722-op572
250
+ �
251
+ bprop.720:[CNode]721:2bprop.720:[CNode]723:3bprop.720:[CNode]723:3"REF::S-Prim-IsShapeUnKnown:4:#Default/S-Prim-IsShapeUnKnown-op573
252
+ q
253
+ bprop.720:[CNode]723:3bprop.720:[CNode]724:5bprop.720:[CNode]724:5"REF::bool_.725:Default/bool_.725-op574
254
+ �
255
+ bprop.720:[CNode]724:5
256
+ ✓bprop.726
257
+ ✗bprop.727bprop.720:[CNode]728:6bprop.720:[CNode]728:6"
258
+ Rbprop.720:[CNode]729:8bprop.720:[CNode]729:8"REF::bprop.720:[CNode]728:6:576 bprop.7202
259
+ bprop.720:[CNode]729:8B�
260
+ e✗bprop.727:[CNode]730:9✗bprop.727:[CNode]730:9"REF::↓bprop.731:Default/↓bprop.731-op577 ✗bprop.7272
261
+ ✗bprop.727:[CNode]730:9B�
262
+ �
263
+ #get_bprop_conv3d_transpose.718:dout
264
+ get_bprop_conv3d_transpose.718:w↓bprop.731:dx:42↓bprop.731:dx:42"1REF::get_bprop_conv3d_transpose.718:input_grad:41:604
265
+ �
266
+ get_bprop_conv3d_transpose.718:w↓bprop.731:[CNode]814:57↓bprop.731:[CNode]814:57"REF::shape.722:Default/shape.722-op616
267
+ �
268
+ #get_bprop_conv3d_transpose.718:dout
269
+ get_bprop_conv3d_transpose.718:x
270
+ ↓bprop.731:[CNode]814:57↓bprop.731:dw:58↓bprop.731:dw:58"2REF::get_bprop_conv3d_transpose.718:filter_grad:56:617
271
+ �
272
+ ↓bprop.731:dx:42
273
+ ↓bprop.731:dw:58↓bprop.731:[CNode]815:59↓bprop.731:[CNode]815:59"REF::S-Prim-MakeTuple:60:Default/S-Prim-MakeTuple-op618 ↓bprop.7312
274
+ ↓bprop.731:[CNode]815:59B
275
+ after_block� B�
276
+ w
277
+ shape.722:input_xshape.722:[CNode]816:61shape.722:[CNode]816:61"REF::S-Prim-Shape:62:Default/S-Prim-Shape-op619 shape.722*
278
+ shape.722:input_x2
279
+ shape.722:[CNode]816:61B�
280
+ �
281
+ #get_bprop_conv3d_transpose.718:dout
282
+ get_bprop_conv3d_transpose.718:w✓bprop.726:dx:63✓bprop.726:dx:63"1REF::get_bprop_conv3d_transpose.718:input_grad:41:620
283
+ �
284
+ get_bprop_conv3d_transpose.718:w✓bprop.726:[CNode]817:65✓bprop.726:[CNode]817:65"4REF::get_bprop_conv3d_transpose.718:get_dyn_shape:64:622
285
+ �
286
+ #get_bprop_conv3d_transpose.718:dout
287
+ get_bprop_conv3d_transpose.718:x
288
+ ✓bprop.726:[CNode]817:65✓bprop.726:dw:66✓bprop.726:dw:66"2REF::get_bprop_conv3d_transpose.718:filter_grad:56:623
289
+ �
290
+ ✓bprop.726:dx:63
291
+ ✓bprop.726:dw:66✓bprop.726:[CNode]818:67✓bprop.726:[CNode]818:67"REF::S-Prim-MakeTuple:68:Default/S-Prim-MakeTuple-op624 ✓bprop.7262
292
+ ✓bprop.726:[CNode]818:67B�
293
+ @[ValueNode]819[ValueNode]819"Constant*
294
+ value*__bool__�
295
+ w
296
+ bool_.725:x
297
+ [ValueNode]819bool_.725:[CNode]820:69bool_.725:[CNode]820:69"REF::getattr:13:Default/getattr-op625
298
+ Ubool_.725:[CNode]821:70bool_.725:[CNode]821:70"REF::bool_.725:[CNode]820:69:626 bool_.725*
299
+ bool_.725:x2
300
+ bool_.725:[CNode]821:70Pb
301
+ Switch:7Switchb
302
+
303
+ getattr:13getattrb
304
+ S-Prim-Shape:62 S-Prim-Shapeb'
305
+ S-Prim-make_dict:40S-Prim-make_dictb'
306
+ S-Prim-MakeTuple:11S-Prim-MakeTupleb'
307
+ S-Prim-make_dict:55S-Prim-make_dictb'
308
+ S-Prim-MakeTuple:60S-Prim-MakeTupleb0
309
+ S-Prim-IsShapeUnKnown:4S-Prim-IsShapeUnKnownb'
310
+ S-Prim-MakeTuple:34S-Prim-MakeTupleb'
311
+ S-Prim-MakeTuple:68S-Prim-MakeTupleb'
312
+ S-Prim-MakeTuple:38S-Prim-MakeTupleb'
313
+ S-Prim-MakeTuple:26S-Prim-MakeTupleb'
314
+ S-Prim-MakeTuple:44S-Prim-MakeTupleb'
315
+ S-Prim-MakeTuple:53S-Prim-MakeTupleh
mindspore/ops/bprop_mindir/Conv3D_bprop.mindir ADDED
@@ -0,0 +1,278 @@
1
+
2
+ 0.1.1 MindSpore*2.0.0:�@
3
+ aget_bprop_conv3d.1:[CNode]2:1get_bprop_conv3d.1:[CNode]2:1" REF::bprop.3:Default/bprop.3-op0
4
+ �get_bprop_conv3d.1:get_shape:2get_bprop_conv3d.1:get_shape:2"8REF::ClassType::mindspore.ops.operations.array_ops.Shape:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Shape'-op1
5
+ A
6
+ value* out_channel�
7
+ �
8
+ get_bprop_conv3d.1:self
9
+
10
+ A
11
+ value* kernel_size�
12
+ �
13
+ get_bprop_conv3d.1:self
14
+
15
+ :
16
+ value*mode�
17
+ �
18
+ get_bprop_conv3d.1:self
19
+
20
+ �
21
+ get_bprop_conv3d.1:[CNode]20:14
22
+ get_bprop_conv3d.1:[CNode]22:16
23
+ get_bprop_conv3d.1:[CNode]24:17get_bprop_conv3d.1:[CNode]25:18get_bprop_conv3d.1:[CNode]25:18"REF::S-Prim-MakeTuple:19:Default/S-Prim-MakeTuple-op14
24
+ >
25
+ value*pad_mode�
26
+ 9
27
+ value*pad�
28
+ <
29
+ value*stride�
30
+ >
31
+ value*dilation�
32
+ ;
33
+ value*group�
34
+ A
35
+ value* data_format�
36
+ �
37
+
38
+
39
+
40
+
41
+
42
+
43
+ >
44
+ value*pad_mode�
45
+ �
46
+ get_bprop_conv3d.1:self
47
+
48
+ 9
49
+ value*pad�
50
+ �
51
+ get_bprop_conv3d.1:self
52
+
53
+ <
54
+ value*stride�
55
+ �
56
+ get_bprop_conv3d.1:self
57
+
58
+ >
59
+ value*dilation�
60
+ �
61
+ get_bprop_conv3d.1:self
62
+
63
+ ;
64
+ value*group�
65
+ �
66
+ get_bprop_conv3d.1:self
67
+
68
+ A
69
+ value* data_format�
70
+ �
71
+ get_bprop_conv3d.1:self
72
+
73
+ �
74
+ get_bprop_conv3d.1:[CNode]34:22
75
+ get_bprop_conv3d.1:[CNode]36:23
76
+ get_bprop_conv3d.1:[CNode]38:24
77
+ get_bprop_conv3d.1:[CNode]40:25
78
+ get_bprop_conv3d.1:[CNode]42:26
79
+ get_bprop_conv3d.1:[CNode]44:27get_bprop_conv3d.1:[CNode]45:28get_bprop_conv3d.1:[CNode]45:28"REF::S-Prim-MakeTuple:29:Default/S-Prim-MakeTuple-op22
80
+ �
81
+ get_bprop_conv3d.1:[CNode]32:20
82
+ get_bprop_conv3d.1:[CNode]45:28get_bprop_conv3d.1:[CNode]46:30get_bprop_conv3d.1:[CNode]46:30"REF::S-Prim-make_dict:31:Default/S-Prim-make_dict-op23
83
+ S4747"Constant*?
84
+ value*3mindspore.ops.operations.nn_ops.Conv3DBackpropInput�!
85
+ �
86
+ 47
87
+ get_bprop_conv3d.1:[CNode]25:18
88
+ get_bprop_conv3d.1:[CNode]46:30 get_bprop_conv3d.1:input_grad:32 get_bprop_conv3d.1:input_grad:32"REF::MetaFuncGraph::unpack_call:Default/unpack_call.48-op24
89
+ �get_bprop_conv3d.1:cast:35get_bprop_conv3d.1:cast:35"7REF::ClassType::mindspore.ops.operations.array_ops.Cast:LDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.Cast'-op27
90
+ A
91
+ value* out_channel�
92
+ �
93
+ get_bprop_conv3d.1:self
94
+
95
+ A
96
+ value* kernel_size�
97
+ �
98
+ get_bprop_conv3d.1:self
99
+
100
+ :
101
+ value*mode�
102
+ �
103
+ get_bprop_conv3d.1:self
104
+
105
+ �
106
+ get_bprop_conv3d.1:[CNode]51:36
107
+ get_bprop_conv3d.1:[CNode]53:37
108
+ get_bprop_conv3d.1:[CNode]55:38get_bprop_conv3d.1:[CNode]56:39get_bprop_conv3d.1:[CNode]56:39"REF::S-Prim-MakeTuple:40:Default/S-Prim-MakeTuple-op31
109
+ >
110
+ value*pad_mode�
111
+ 9
112
+ value*pad�
113
+ <
114
+ value*stride�
115
+ >
116
+ value*dilation�
117
+ ;
118
+ value*group�
119
+ A
120
+ value* data_format�
121
+ �
122
+
123
+
124
+
125
+
126
+
127
+
128
+ >
129
+ value*pad_mode�
130
+ �
131
+ get_bprop_conv3d.1:self
132
+
133
+ 9
134
+ value*pad�
135
+ �
136
+ get_bprop_conv3d.1:self
137
+
138
+ <
139
+ value*stride�
140
+ �
141
+ get_bprop_conv3d.1:self
142
+
143
+ >
144
+ value*dilation�
145
+ �
146
+ get_bprop_conv3d.1:self
147
+
148
+ ;
149
+ value*group�
150
+ �
151
+ get_bprop_conv3d.1:self
152
+
153
+ A
154
+ value* data_format�
155
+ �
156
+ get_bprop_conv3d.1:self
157
+
158
+ �
159
+ get_bprop_conv3d.1:[CNode]65:43
160
+ get_bprop_conv3d.1:[CNode]67:44
161
+ get_bprop_conv3d.1:[CNode]69:45
162
+ get_bprop_conv3d.1:[CNode]71:46
163
+ get_bprop_conv3d.1:[CNode]73:47
164
+ get_bprop_conv3d.1:[CNode]75:48get_bprop_conv3d.1:[CNode]76:49get_bprop_conv3d.1:[CNode]76:49"REF::S-Prim-MakeTuple:50:Default/S-Prim-MakeTuple-op39
165
+ �
166
+ get_bprop_conv3d.1:[CNode]63:41
167
+ get_bprop_conv3d.1:[CNode]76:49get_bprop_conv3d.1:[CNode]77:51get_bprop_conv3d.1:[CNode]77:51"REF::S-Prim-make_dict:52:Default/S-Prim-make_dict-op40
168
+ W7878"Constant*C
169
+ value*7mindspore.ops.operations._grad_ops.Conv3DBackpropFilter�!
170
+ �
171
+ 78
172
+ get_bprop_conv3d.1:[CNode]56:39
173
+ get_bprop_conv3d.1:[CNode]77:51!get_bprop_conv3d.1:filter_grad:53!get_bprop_conv3d.1:filter_grad:53"REF::MetaFuncGraph::unpack_call:Default/unpack_call.79-op41
174
+ �get_bprop_conv3d.1:get_dtype:56get_bprop_conv3d.1:get_dtype:56"8REF::ClassType::mindspore.ops.operations.array_ops.DType:MDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.DType'-op44
175
+ �#get_bprop_conv3d.1:get_dyn_shape:61#get_bprop_conv3d.1:get_dyn_shape:61">REF::ClassType::mindspore.ops.operations.array_ops.TensorShape:SDefault/MindIRClassType:class 'mindspore.ops.operations.array_ops.TensorShape'-op48get_bprop_conv3d.1*
176
+ get_bprop_conv3d.1:self*
177
+ get_bprop_conv3d.1:x*
178
+ get_bprop_conv3d.1:w*
179
+ get_bprop_conv3d.1:out*
180
+ get_bprop_conv3d.1:dout2
181
+ get_bprop_conv3d.1:[CNode]2:1:@ac358b5dcf264c042115b19fd775801a9b232b96f41720e2d3073b71bec3c441J/grad_nn_ops.pyB�
182
+ f
183
+ get_bprop_conv3d.1:xbprop.3:[CNode]4:3bprop.3:[CNode]4:3"#REF::get_bprop_conv3d.1:get_shape:2:2
184
+ }
185
+ bprop.3:[CNode]4:3bprop.3:[CNode]5:4bprop.3:[CNode]5:4"REF::S-Prim-IsShapeUnKnown:5:!Default/S-Prim-IsShapeUnKnown-op3
186
+ _
187
+ bprop.3:[CNode]5:4bprop.3:[CNode]6:6bprop.3:[CNode]6:6" REF::bool_.7:Default/bool_.7-op4
188
+ y
189
+ bprop.3:[CNode]6:6
190
+
191
+ ↰bprop.8
192
+
193
+ ↱bprop.9bprop.3:[CNode]10:7bprop.3:[CNode]10:7"
194
+ Gbprop.3:[CNode]11:9bprop.3:[CNode]11:9"REF::bprop.3:[CNode]10:7:6
195
+ d
196
+ bprop.3:[CNode]11:9bprop.3:[CNode]12:10bprop.3:[CNode]12:10" REF::bool_.7:Default/bool_.7-op7
197
+ 
198
+ bprop.3:[CNode]12:10
199
+ ✓bprop.13
200
+ ✗bprop.14bprop.3:[CNode]15:11bprop.3:[CNode]15:11"
201
+ Jbprop.3:[CNode]16:12bprop.3:[CNode]16:12"REF::bprop.3:[CNode]15:11:9bprop.32
202
+ bprop.3:[CNode]16:12B�
203
+ `✗bprop.14:[CNode]17:13✗bprop.14:[CNode]17:13"REF::↓bprop.18:Default/↓bprop.18-op10 ✗bprop.142
204
+ ✗bprop.14:[CNode]17:13B�
205
+ s
206
+ get_bprop_conv3d.1:x↓bprop.18:[CNode]49:33↓bprop.18:[CNode]49:33"#REF::get_bprop_conv3d.1:get_shape:2:25
207
+ �
208
+ get_bprop_conv3d.1:w
209
+ get_bprop_conv3d.1:dout
210
+ ↓bprop.18:[CNode]49:33↓bprop.18:dx:34↓bprop.18:dx:34"%REF::get_bprop_conv3d.1:input_grad:32:26
211
+ s
212
+ get_bprop_conv3d.1:w↓bprop.18:[CNode]80:54↓bprop.18:[CNode]80:54"#REF::get_bprop_conv3d.1:get_shape:2:42
213
+ �
214
+ get_bprop_conv3d.1:x
215
+ get_bprop_conv3d.1:dout
216
+ ↓bprop.18:[CNode]80:54↓bprop.18:[CNode]81:55↓bprop.18:[CNode]81:55"&REF::get_bprop_conv3d.1:filter_grad:53:43
217
+ t
218
+ get_bprop_conv3d.1:x↓bprop.18:[CNode]82:57↓bprop.18:[CNode]82:57"$REF::get_bprop_conv3d.1:get_dtype:56:45
219
+ 
220
+ ↓bprop.18:[CNode]81:55
221
+ ↓bprop.18:[CNode]82:57↓bprop.18:dw:58↓bprop.18:dw:58"REF::get_bprop_conv3d.1:cast:35:46
222
+ �
223
+ ↓bprop.18:dx:34
224
+ ↓bprop.18:dw:58↓bprop.18:[CNode]83:59↓bprop.18:[CNode]83:59"REF::S-Prim-MakeTuple:60:Default/S-Prim-MakeTuple-op47 ↓bprop.182
225
+ ↓bprop.18:[CNode]83:59B
226
+ after_block� B�
227
+ x
228
+ get_bprop_conv3d.1:x✓bprop.13:[CNode]84:62✓bprop.13:[CNode]84:62"(REF::get_bprop_conv3d.1:get_dyn_shape:61:49
229
+ �
230
+ get_bprop_conv3d.1:w
231
+ get_bprop_conv3d.1:dout
232
+ ✓bprop.13:[CNode]84:62✓bprop.13:dx:63✓bprop.13:dx:63"%REF::get_bprop_conv3d.1:input_grad:32:50
233
+ x
234
+ get_bprop_conv3d.1:w✓bprop.13:[CNode]85:64✓bprop.13:[CNode]85:64"(REF::get_bprop_conv3d.1:get_dyn_shape:61:51
235
+ �
236
+ get_bprop_conv3d.1:x
237
+ get_bprop_conv3d.1:dout
238
+ ✓bprop.13:[CNode]85:64✓bprop.13:[CNode]86:65✓bprop.13:[CNode]86:65"&REF::get_bprop_conv3d.1:filter_grad:53:52
239
+ t
240
+ get_bprop_conv3d.1:x✓bprop.13:[CNode]87:66✓bprop.13:[CNode]87:66"$REF::get_bprop_conv3d.1:get_dtype:56:53
241
+ 
242
+ ✓bprop.13:[CNode]86:65
243
+ ✓bprop.13:[CNode]87:66✓bprop.13:dw:67✓bprop.13:dw:67"REF::get_bprop_conv3d.1:cast:35:54
244
+ �
245
+ ✓bprop.13:dx:63
246
+ ✓bprop.13:dw:67✓bprop.13:[CNode]88:68✓bprop.13:[CNode]88:68"REF::S-Prim-MakeTuple:69:Default/S-Prim-MakeTuple-op55 ✓bprop.132
247
+ ✓bprop.13:[CNode]88:68B�
248
+ >
249
+ value*__bool__�
250
+ m
251
+ bool_.7:x
252
+
253
+ Kbool_.7:[CNode]91:71bool_.7:[CNode]91:71"REF::bool_.7:[CNode]90:70:57bool_.7*
254
+ bool_.7:x2
255
+ bool_.7:[CNode]91:71B�
256
+ q
257
+ get_bprop_conv3d.1:w↱bprop.9:[CNode]92:72↱bprop.9:[CNode]92:72"#REF::get_bprop_conv3d.1:get_shape:2:58
258
+ �
259
+ ↱bprop.9:[CNode]92:72↱bprop.9:[CNode]93:73↱bprop.9:[CNode]93:73"REF::S-Prim-IsShapeUnKnown:74:"Default/S-Prim-IsShapeUnKnown-op59
260
+ ↱bprop.92
261
+ ↱bprop.9:[CNode]93:73B"
262
+ ↰bprop.82
263
+ bprop.3:[CNode]5:4Pb
264
+ Switch:8Switchb
265
+
266
+ getattr:15getattrb'
267
+ S-Prim-MakeTuple:29S-Prim-MakeTupleb0
268
+ S-Prim-IsShapeUnKnown:5S-Prim-IsShapeUnKnownb1
269
+ S-Prim-IsShapeUnKnown:74S-Prim-IsShapeUnKnownb'
270
+ S-Prim-make_dict:31S-Prim-make_dictb'
271
+ S-Prim-MakeTuple:21S-Prim-MakeTupleb'
272
+ S-Prim-MakeTuple:50S-Prim-MakeTupleb'
273
+ S-Prim-MakeTuple:40S-Prim-MakeTupleb'
274
+ S-Prim-make_dict:52S-Prim-make_dictb'
275
+ S-Prim-MakeTuple:69S-Prim-MakeTupleb'
276
+ S-Prim-MakeTuple:60S-Prim-MakeTupleb'
277
+ S-Prim-MakeTuple:42S-Prim-MakeTupleb'
278
+ S-Prim-MakeTuple:19S-Prim-MakeTupleh
mindspore/ops/bprop_mindir/DType_bprop.mindir CHANGED
@@ -1,14 +1,14 @@
1
1
 
2
- 0.1.1 MindSpore*1.9.0:�
3
- �
4
-
5
- bprop.37:xbprop.37:[CNode]38:1bprop.37:[CNode]38:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op27
6
- z
7
- bprop.37:[CNode]38:1bprop.37:[CNode]39:3bprop.37:[CNode]39:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op28bprop.37*
8
-
9
- bprop.37:x*
10
- bprop.37:out*
11
-
12
- bprop.37:[CNode]39:3:@030ffb7e684c8e2604e69bfc02a6e32af82543a97f4aa0a6bc7cab6275ee9b9dPbH
13
- #S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
2
+ 0.1.1 MindSpore*2.0.0:�
3
+ _get_bprop_dtype.1:[CNode]2:1get_bprop_dtype.1:[CNode]2:1" REF::bprop.3:Default/bprop.3-op0get_bprop_dtype.1*
4
+ get_bprop_dtype.1:self*
5
+ get_bprop_dtype.1:x*
6
+ get_bprop_dtype.1:out*
7
+ get_bprop_dtype.1:dout2
8
+ get_bprop_dtype.1:[CNode]2:1:@54a9dbba51e886fa876dd646b337aec70b1ecbfcf799d96be0dbb699f1deea5eJ/grad_array_ops.pyB�
9
+ �
10
+ get_bprop_dtype.1:xbprop.3:[CNode]4:2bprop.3:[CNode]4:2".REF::MetaFuncGraph::hyper_map[zeros_like_leaf]:-Default/S-Prim-hyper_map[zeros_like_leaf]-op1
11
+ s
12
+ bprop.3:[CNode]4:2bprop.3:[CNode]5:3bprop.3:[CNode]5:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op2bprop.32
13
+ bprop.3:[CNode]5:3Pb&
14
14
  S-Prim-MakeTuple:4S-Prim-MakeTupleh
mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir ADDED
@@ -0,0 +1,58 @@
1
+
2
+ 0.1.1 MindSpore*2.0.0:�
3
+ {*get_bprop_deformable_offsets.8:[CNode]16:1*get_bprop_deformable_offsets.8:[CNode]16:1" REF::bprop.9:Default/bprop.9-op0
4
+ =
5
+ value*strides�
6
+ �
7
+ #get_bprop_deformable_offsets.8:self
8
+
9
+ :
10
+ value*pads�
11
+ �
12
+ #get_bprop_deformable_offsets.8:self
13
+
14
+ ;
15
+ value*ksize�
16
+ �
17
+ #get_bprop_deformable_offsets.8:self
18
+
19
+ ?
20
+ value* dilations�
21
+ �
22
+ #get_bprop_deformable_offsets.8:self
23
+
24
+ A
25
+ value* data_format�
26
+ �
27
+ #get_bprop_deformable_offsets.8:self
28
+
29
+ G
30
+ value*deformable_groups�
31
+ �
32
+ #get_bprop_deformable_offsets.8:self
33
+
34
+ ?
35
+ value* modulated�
36
+ �
37
+ #get_bprop_deformable_offsets.8:self
38
+
39
+ �
40
+ )get_bprop_deformable_offsets.8:[CNode]1:2
41
+ )get_bprop_deformable_offsets.8:[CNode]2:4
42
+ )get_bprop_deformable_offsets.8:[CNode]3:5
43
+ )get_bprop_deformable_offsets.8:[CNode]4:6
44
+ )get_bprop_deformable_offsets.8:[CNode]5:7
45
+ )get_bprop_deformable_offsets.8:[CNode]6:8
46
+ )get_bprop_deformable_offsets.8:[CNode]7:9&get_bprop_deformable_offsets.8:grad:10&get_bprop_deformable_offsets.8:grad:10"HREF::ClassType::mindspore.ops.operations._grad_ops.DeformableOffsetsGrad:\Default/MindIRClassType:class 'mindspore.ops.operations._grad_ops.DeformableOffsetsGrad'-op8get_bprop_deformable_offsets.8*%
47
+ #get_bprop_deformable_offsets.8:self*"
48
+ get_bprop_deformable_offsets.8:x*(
49
+ &get_bprop_deformable_offsets.8:offsets*$
50
+ "get_bprop_deformable_offsets.8:out*%
51
+ #get_bprop_deformable_offsets.8:dout2,
52
+ *get_bprop_deformable_offsets.8:[CNode]16:1:@c8294d4d00342872d591bac5f3fc0747aca6ca338151bc930e0f21f59288bce1J/grad_nn_ops.pyB�
53
+ �
54
+ #get_bprop_deformable_offsets.8:dout
55
+ get_bprop_deformable_offsets.8:x
56
+ &get_bprop_deformable_offsets.8:offsetsbprop.9:out_grad:11bprop.9:out_grad:11"+REF::get_bprop_deformable_offsets.8:grad:10:9bprop.92
57
+ bprop.9:out_grad:11Pb
58
+ getattr:3getattrh