mindspore 1.10.0__cp39-cp39-win_amd64.whl → 2.0.0rc1__cp39-cp39-win_amd64.whl

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

Potentially problematic release.


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

Files changed (966) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/ConcurrencyCheck.dll +0 -0
  3. mindspore/CppBuildInsights.dll +0 -0
  4. mindspore/CppCoreCheck.dll +0 -0
  5. mindspore/EnumIndex.dll +0 -0
  6. mindspore/EspXEngine.dll +0 -0
  7. mindspore/HResultCheck.dll +0 -0
  8. mindspore/KernelTraceControl.dll +0 -0
  9. mindspore/LocalESPC.dll +0 -0
  10. mindspore/Microsoft.Diagnostics.Tracing.EventSource.dll +0 -0
  11. mindspore/Microsoft.VisualStudio.RemoteControl.dll +0 -0
  12. mindspore/Microsoft.VisualStudio.Telemetry.dll +0 -0
  13. mindspore/Microsoft.VisualStudio.Utilities.Internal.dll +0 -0
  14. mindspore/Newtonsoft.Json.dll +0 -0
  15. mindspore/System.Runtime.CompilerServices.Unsafe.dll +0 -0
  16. mindspore/VariantClear.dll +0 -0
  17. mindspore/__init__.py +9 -4
  18. mindspore/_c_dataengine.cp39-win_amd64.pyd +0 -0
  19. mindspore/_c_expression.cp39-win_amd64.pyd +0 -0
  20. mindspore/_c_mindrecord.cp39-win_amd64.pyd +0 -0
  21. mindspore/_check_jit_forbidden_api.py +102 -0
  22. mindspore/_checkparam.py +1066 -1001
  23. mindspore/_extends/builtin_operations.py +32 -4
  24. mindspore/_extends/graph_kernel/model/graph_split.py +66 -222
  25. mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +12 -9
  26. mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +119 -26
  27. mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +50 -50
  28. mindspore/_extends/parallel_compile/akg_compiler/util.py +9 -6
  29. mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +4 -25
  30. mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +9 -4
  31. mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -27
  32. mindspore/_extends/parse/__init__.py +5 -3
  33. mindspore/_extends/parse/namespace.py +17 -2
  34. mindspore/_extends/parse/parser.py +193 -34
  35. mindspore/_extends/parse/resources.py +7 -8
  36. mindspore/_extends/parse/standard_method.py +1780 -435
  37. mindspore/_extends/parse/trope.py +3 -1
  38. mindspore/amp.py +53 -58
  39. mindspore/atlprov.dll +0 -0
  40. mindspore/boost/adasum.py +3 -2
  41. mindspore/boost/boost.py +2 -2
  42. mindspore/boost/boost_cell_wrapper.py +46 -26
  43. mindspore/boost/dim_reduce.py +6 -5
  44. mindspore/boost/grad_accumulation.py +2 -1
  45. mindspore/boost/group_loss_scale_manager.py +1 -1
  46. mindspore/c1.dll +0 -0
  47. mindspore/c1xx.dll +0 -0
  48. mindspore/c2.dll +0 -0
  49. mindspore/cfgpersist.dll +0 -0
  50. mindspore/clang_rt.asan_dbg_dynamic-x86_64.dll +0 -0
  51. mindspore/clang_rt.asan_dynamic-x86_64.dll +0 -0
  52. mindspore/common/__init__.py +11 -10
  53. mindspore/common/_decorator.py +2 -0
  54. mindspore/common/_register_for_adapter.py +55 -0
  55. mindspore/common/_stub_tensor.py +201 -0
  56. mindspore/common/_utils.py +57 -0
  57. mindspore/common/api.py +582 -297
  58. mindspore/common/dtype.py +66 -18
  59. mindspore/common/dump.py +2 -2
  60. mindspore/common/initializer.py +38 -1
  61. mindspore/common/jit_config.py +25 -13
  62. mindspore/common/mutable.py +53 -24
  63. mindspore/common/parameter.py +60 -37
  64. mindspore/common/seed.py +8 -24
  65. mindspore/common/sparse_tensor.py +927 -0
  66. mindspore/common/tensor.py +1627 -3900
  67. mindspore/communication/__init__.py +10 -5
  68. mindspore/communication/_comm_helper.py +78 -214
  69. mindspore/communication/_hccl_management.py +2 -1
  70. mindspore/communication/management.py +136 -47
  71. mindspore/config/op_info.config +501 -1008
  72. mindspore/context.py +291 -56
  73. mindspore/d3dcompiler_47.dll +0 -0
  74. mindspore/dataset/__init__.py +12 -8
  75. mindspore/dataset/audio/__init__.py +9 -9
  76. mindspore/dataset/audio/transforms.py +1090 -228
  77. mindspore/dataset/audio/utils.py +87 -39
  78. mindspore/dataset/audio/validators.py +223 -1
  79. mindspore/dataset/callback/ds_callback.py +17 -15
  80. mindspore/dataset/core/config.py +246 -17
  81. mindspore/dataset/core/py_util_helpers.py +4 -3
  82. mindspore/dataset/core/validator_helpers.py +10 -10
  83. mindspore/{parallel/nn/layers.py → dataset/debug/__init__.py} +7 -8
  84. mindspore/dataset/debug/debug_hook.py +65 -0
  85. mindspore/dataset/debug/pre_defined_hook.py +67 -0
  86. mindspore/dataset/engine/__init__.py +7 -3
  87. mindspore/dataset/engine/cache_client.py +9 -9
  88. mindspore/dataset/engine/datasets.py +648 -477
  89. mindspore/dataset/engine/datasets_audio.py +165 -167
  90. mindspore/dataset/engine/datasets_standard_format.py +93 -67
  91. mindspore/dataset/engine/datasets_text.py +492 -342
  92. mindspore/dataset/engine/datasets_user_defined.py +85 -50
  93. mindspore/dataset/engine/datasets_vision.py +1224 -699
  94. mindspore/dataset/engine/graphdata.py +134 -69
  95. mindspore/dataset/engine/iterators.py +50 -9
  96. mindspore/dataset/engine/offload.py +52 -31
  97. mindspore/dataset/engine/samplers.py +27 -24
  98. mindspore/dataset/engine/serializer_deserializer.py +14 -15
  99. mindspore/dataset/engine/validators.py +213 -52
  100. mindspore/dataset/text/__init__.py +10 -8
  101. mindspore/dataset/text/transforms.py +152 -57
  102. mindspore/dataset/text/utils.py +98 -49
  103. mindspore/dataset/text/validators.py +25 -0
  104. mindspore/dataset/transforms/__init__.py +4 -2
  105. mindspore/dataset/transforms/c_transforms.py +11 -13
  106. mindspore/dataset/transforms/py_transforms.py +2 -2
  107. mindspore/dataset/transforms/py_transforms_util.py +10 -0
  108. mindspore/dataset/transforms/transforms.py +13 -15
  109. mindspore/dataset/transforms/validators.py +7 -7
  110. mindspore/dataset/utils/__init__.py +2 -1
  111. mindspore/dataset/utils/browse_dataset.py +13 -13
  112. mindspore/dataset/utils/line_reader.py +121 -0
  113. mindspore/dataset/vision/__init__.py +8 -7
  114. mindspore/dataset/vision/c_transforms.py +125 -126
  115. mindspore/dataset/vision/py_transforms.py +37 -37
  116. mindspore/dataset/vision/py_transforms_util.py +23 -20
  117. mindspore/dataset/vision/transforms.py +316 -315
  118. mindspore/dataset/vision/utils.py +313 -17
  119. mindspore/dataset/vision/validators.py +6 -6
  120. mindspore/default_config.py +0 -1
  121. mindspore/dpcmi.dll +0 -0
  122. mindspore/{compression → experimental}/__init__.py +6 -5
  123. mindspore/experimental/map_parameter.py +275 -0
  124. mindspore/include/OWNERS +0 -1
  125. mindspore/include/api/callback/callback.h +9 -13
  126. mindspore/include/api/callback/ckpt_saver.h +2 -2
  127. mindspore/include/api/callback/loss_monitor.h +2 -2
  128. mindspore/include/api/callback/lr_scheduler.h +5 -5
  129. mindspore/include/api/callback/time_monitor.h +2 -2
  130. mindspore/include/api/callback/train_accuracy.h +4 -6
  131. mindspore/include/api/cfg.h +19 -6
  132. mindspore/include/api/context.h +70 -9
  133. mindspore/include/api/delegate.h +8 -1
  134. mindspore/include/api/dual_abi_helper.h +8 -24
  135. mindspore/include/api/metrics/accuracy.h +2 -2
  136. mindspore/include/api/metrics/metrics.h +4 -3
  137. mindspore/include/api/model.h +9 -4
  138. mindspore/include/api/model_group.h +68 -0
  139. mindspore/include/api/model_parallel_runner.h +17 -17
  140. mindspore/include/api/net.h +12 -11
  141. mindspore/include/api/serialization.h +20 -4
  142. mindspore/include/api/status.h +7 -1
  143. mindspore/include/api/types.h +25 -21
  144. mindspore/include/api/visible.h +4 -0
  145. mindspore/include/c_api/model_c.h +5 -0
  146. mindspore/include/c_api/status_c.h +1 -1
  147. mindspore/include/dataset/config.h +1 -1
  148. mindspore/include/dataset/constants.h +14 -0
  149. mindspore/include/dataset/text.h +59 -0
  150. mindspore/include/dataset/vision.h +56 -117
  151. mindspore/include/dataset/vision_lite.h +102 -0
  152. mindspore/jpeg62.dll +0 -0
  153. mindspore/log.py +28 -28
  154. mindspore/mindrecord/common/exceptions.py +2 -4
  155. mindspore/mindrecord/filereader.py +19 -1
  156. mindspore/mindrecord/filewriter.py +250 -88
  157. mindspore/mindrecord/mindpage.py +13 -13
  158. mindspore/mindrecord/shardheader.py +15 -15
  159. mindspore/mindrecord/shardreader.py +9 -0
  160. mindspore/mindrecord/shardwriter.py +29 -29
  161. mindspore/mindrecord/tools/cifar100_to_mr.py +9 -9
  162. mindspore/mindrecord/tools/cifar10_to_mr.py +9 -9
  163. mindspore/mindrecord/tools/csv_to_mr.py +4 -4
  164. mindspore/mindrecord/tools/imagenet_to_mr.py +70 -65
  165. mindspore/mindrecord/tools/mnist_to_mr.py +41 -41
  166. mindspore/mindrecord/tools/tfrecord_to_mr.py +6 -6
  167. mindspore/{libmindspore_backend.dll → mindspore_backend.dll} +0 -0
  168. mindspore/mindspore_common.dll +0 -0
  169. mindspore/mindspore_core.dll +0 -0
  170. mindspore/mindspore_glog.dll +0 -0
  171. mindspore/mindspore_shared_lib.dll +0 -0
  172. mindspore/msobj140.dll +0 -0
  173. mindspore/mspdb140.dll +0 -0
  174. mindspore/mspdbcore.dll +0 -0
  175. mindspore/mspdbst.dll +0 -0
  176. mindspore/mspft140.dll +0 -0
  177. mindspore/msvcdis140.dll +0 -0
  178. mindspore/msvcp140_1.dll +0 -0
  179. mindspore/msvcp140_2.dll +0 -0
  180. mindspore/msvcp140_atomic_wait.dll +0 -0
  181. mindspore/msvcp140_codecvt_ids.dll +0 -0
  182. mindspore/nn/__init__.py +1 -5
  183. mindspore/nn/cell.py +297 -234
  184. mindspore/nn/dynamic_lr.py +1 -1
  185. mindspore/nn/grad/cell_grad.py +17 -42
  186. mindspore/nn/layer/__init__.py +7 -4
  187. mindspore/nn/layer/activation.py +131 -88
  188. mindspore/nn/layer/basic.py +313 -613
  189. mindspore/nn/layer/channel_shuffle.py +103 -0
  190. mindspore/nn/layer/combined.py +1 -1
  191. mindspore/nn/layer/container.py +52 -6
  192. mindspore/nn/layer/conv.py +112 -43
  193. mindspore/nn/layer/dense.py +10 -9
  194. mindspore/nn/layer/embedding.py +36 -34
  195. mindspore/nn/layer/image.py +123 -27
  196. mindspore/nn/layer/math.py +108 -107
  197. mindspore/nn/layer/normalization.py +212 -366
  198. mindspore/nn/layer/padding.py +370 -42
  199. mindspore/nn/layer/pooling.py +1443 -219
  200. mindspore/nn/layer/rnn_cells.py +11 -16
  201. mindspore/nn/layer/rnns.py +38 -39
  202. mindspore/nn/layer/thor_layer.py +24 -25
  203. mindspore/nn/layer/timedistributed.py +5 -5
  204. mindspore/nn/layer/transformer.py +701 -0
  205. mindspore/nn/learning_rate_schedule.py +8 -8
  206. mindspore/nn/loss/__init__.py +9 -6
  207. mindspore/nn/loss/loss.py +678 -142
  208. mindspore/nn/metrics.py +53 -0
  209. mindspore/nn/optim/_dist_optimizer_registry.py +2 -2
  210. mindspore/nn/optim/ada_grad.py +8 -8
  211. mindspore/nn/optim/adadelta.py +2 -3
  212. mindspore/nn/optim/adafactor.py +18 -14
  213. mindspore/nn/optim/adam.py +429 -87
  214. mindspore/nn/optim/adamax.py +5 -6
  215. mindspore/nn/optim/adasum.py +10 -8
  216. mindspore/nn/optim/asgd.py +7 -7
  217. mindspore/nn/optim/ftrl.py +81 -11
  218. mindspore/nn/optim/lamb.py +7 -8
  219. mindspore/nn/optim/lars.py +4 -4
  220. mindspore/nn/optim/lazyadam.py +82 -7
  221. mindspore/nn/optim/momentum.py +8 -7
  222. mindspore/nn/optim/optimizer.py +19 -10
  223. mindspore/nn/optim/proximal_ada_grad.py +6 -5
  224. mindspore/nn/optim/rmsprop.py +3 -3
  225. mindspore/nn/optim/rprop.py +20 -16
  226. mindspore/nn/optim/sgd.py +21 -15
  227. mindspore/nn/optim/thor.py +23 -21
  228. mindspore/nn/probability/__init__.py +0 -2
  229. mindspore/nn/probability/bijector/bijector.py +7 -6
  230. mindspore/nn/probability/bijector/invert.py +4 -2
  231. mindspore/nn/probability/bijector/softplus.py +2 -2
  232. mindspore/nn/probability/bnn_layers/dense_variational.py +1 -1
  233. mindspore/nn/probability/bnn_layers/layer_distribution.py +2 -2
  234. mindspore/nn/probability/distribution/__init__.py +6 -0
  235. mindspore/nn/probability/distribution/_utils/custom_ops.py +3 -2
  236. mindspore/nn/probability/distribution/_utils/utils.py +11 -17
  237. mindspore/nn/probability/distribution/bernoulli.py +6 -6
  238. mindspore/nn/probability/distribution/beta.py +1 -1
  239. mindspore/nn/probability/distribution/categorical.py +9 -9
  240. mindspore/nn/probability/distribution/cauchy.py +8 -8
  241. mindspore/nn/probability/distribution/distribution.py +12 -6
  242. mindspore/nn/probability/distribution/exponential.py +5 -5
  243. mindspore/nn/probability/distribution/gamma.py +3 -3
  244. mindspore/nn/probability/distribution/geometric.py +6 -5
  245. mindspore/nn/probability/distribution/gumbel.py +5 -5
  246. mindspore/nn/probability/distribution/half_normal.py +133 -0
  247. mindspore/nn/probability/distribution/laplace.py +128 -0
  248. mindspore/nn/probability/distribution/log_normal.py +0 -1
  249. mindspore/nn/probability/distribution/logistic.py +4 -5
  250. mindspore/nn/probability/distribution/normal.py +11 -15
  251. mindspore/nn/probability/distribution/poisson.py +6 -2
  252. mindspore/nn/probability/distribution/student_t.py +150 -0
  253. mindspore/nn/probability/distribution/transformed_distribution.py +4 -4
  254. mindspore/nn/probability/distribution/uniform.py +5 -5
  255. mindspore/nn/reinforcement/_tensors_queue.py +3 -3
  256. mindspore/nn/reinforcement/tensor_array.py +2 -2
  257. mindspore/nn/sparse/sparse.py +8 -1
  258. mindspore/nn/wrap/cell_wrapper.py +55 -27
  259. mindspore/nn/wrap/grad_reducer.py +20 -11
  260. mindspore/nn/wrap/loss_scale.py +47 -30
  261. mindspore/numpy/array_creations.py +33 -22
  262. mindspore/numpy/array_ops.py +46 -42
  263. mindspore/numpy/logic_ops.py +6 -27
  264. mindspore/numpy/math_ops.py +26 -19
  265. mindspore/numpy/utils.py +1 -8
  266. mindspore/numpy/utils_const.py +112 -62
  267. mindspore/opencv_core452.dll +0 -0
  268. mindspore/opencv_imgcodecs452.dll +0 -0
  269. mindspore/opencv_imgproc452.dll +0 -0
  270. mindspore/ops/__init__.py +6 -3
  271. mindspore/ops/_constants.py +0 -6
  272. mindspore/ops/_grad/__init__.py +2 -1
  273. mindspore/ops/_grad/grad_array_ops.py +209 -152
  274. mindspore/ops/_grad/grad_base.py +55 -17
  275. mindspore/ops/_grad/grad_clip_ops.py +11 -3
  276. mindspore/ops/_grad/grad_comm_ops.py +58 -47
  277. mindspore/ops/_grad/grad_implementations.py +21 -61
  278. mindspore/ops/_grad/grad_inner_ops.py +48 -6
  279. mindspore/ops/_grad/grad_math_ops.py +306 -161
  280. mindspore/ops/_grad/grad_nn_ops.py +192 -181
  281. mindspore/ops/_grad/grad_other_ops.py +1 -1
  282. mindspore/ops/_grad/grad_quant_ops.py +5 -5
  283. mindspore/ops/_grad/grad_sequence_ops.py +296 -0
  284. mindspore/ops/_grad/grad_sparse.py +15 -9
  285. mindspore/ops/_grad_experimental/__init__.py +1 -0
  286. mindspore/ops/_grad_experimental/grad_array_ops.py +441 -55
  287. mindspore/ops/_grad_experimental/grad_image_ops.py +25 -7
  288. mindspore/ops/_grad_experimental/grad_inner_ops.py +3 -44
  289. mindspore/ops/_grad_experimental/grad_linalg_ops.py +16 -21
  290. mindspore/ops/_grad_experimental/grad_math_ops.py +979 -49
  291. mindspore/ops/_grad_experimental/grad_nn_ops.py +78 -8
  292. mindspore/ops/_grad_experimental/grad_scalar_ops.py +112 -0
  293. mindspore/ops/_grad_experimental/grad_sparse_ops.py +197 -13
  294. mindspore/ops/_op_impl/__init__.py +3 -3
  295. mindspore/ops/_op_impl/_custom_op/__init__.py +0 -1
  296. mindspore/ops/_op_impl/_custom_op/_basic.py +0 -1
  297. mindspore/ops/_op_impl/_custom_op/batch_matmul_impl.py +1 -1
  298. mindspore/ops/_op_impl/_custom_op/batchnorm_fold.py +4 -2
  299. mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py +2 -2
  300. mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py +2 -2
  301. mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py +5 -5
  302. mindspore/ops/_op_impl/_custom_op/batchnorm_fold_grad.py +3 -3
  303. mindspore/ops/_op_impl/_custom_op/cholesky_trsm_impl.py +1 -1
  304. mindspore/ops/_op_impl/_custom_op/correction_mul.py +3 -3
  305. mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py +2 -2
  306. mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +4 -8
  307. mindspore/ops/_op_impl/_custom_op/dsd_impl.py +1 -1
  308. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py +2 -2
  309. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py +2 -2
  310. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py +2 -2
  311. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py +2 -2
  312. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py +2 -2
  313. mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py +2 -2
  314. mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py +2 -2
  315. mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py +2 -2
  316. mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py +2 -2
  317. mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py +2 -2
  318. mindspore/ops/_op_impl/_custom_op/fused_abs_max1_impl.py +1 -1
  319. mindspore/ops/_op_impl/_custom_op/img2col_impl.py +1 -1
  320. mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
  321. mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_right_impl.py +1 -1
  322. mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py +1 -1
  323. mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py +1 -1
  324. mindspore/ops/_op_impl/_custom_op/matmul_cube_impl.py +2 -2
  325. mindspore/ops/_op_impl/_custom_op/matmul_dds_grad_impl.py +0 -1
  326. mindspore/ops/_op_impl/_custom_op/matmul_dds_impl.py +0 -1
  327. mindspore/ops/_op_impl/_custom_op/matrix_combine_impl.py +1 -1
  328. mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py +2 -2
  329. mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py +2 -2
  330. mindspore/ops/_op_impl/_custom_op/transpose02314_impl.py +1 -1
  331. mindspore/ops/_op_impl/aicpu/__init__.py +238 -3
  332. mindspore/ops/_op_impl/aicpu/abs.py +36 -0
  333. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d.py +34 -0
  334. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d_grad.py +34 -0
  335. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d.py +39 -0
  336. mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_3d_grad.py +39 -0
  337. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_2d_grad.py +37 -0
  338. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d.py +42 -0
  339. mindspore/ops/_op_impl/aicpu/adaptive_max_pool_3d_grad.py +152 -0
  340. mindspore/ops/_op_impl/aicpu/add.py +43 -0
  341. mindspore/ops/_op_impl/aicpu/addcdiv.py +0 -32
  342. mindspore/ops/_op_impl/aicpu/addcmul.py +0 -84
  343. mindspore/ops/_op_impl/aicpu/affine_grid_grad.py +35 -0
  344. mindspore/ops/_op_impl/aicpu/arg_max.py +75 -0
  345. mindspore/ops/_op_impl/aicpu/arg_min.py +75 -0
  346. mindspore/ops/_op_impl/aicpu/argmin_with_value.py +43 -0
  347. mindspore/ops/_op_impl/aicpu/batch_matmul.py +43 -0
  348. mindspore/ops/_op_impl/aicpu/batch_norm_grad_grad.py +49 -0
  349. mindspore/ops/_op_impl/aicpu/bernoulli.py +48 -0
  350. mindspore/ops/_op_impl/aicpu/bessel_i0.py +31 -0
  351. mindspore/ops/_op_impl/aicpu/bias_add.py +44 -0
  352. mindspore/ops/_op_impl/aicpu/bias_add_grad.py +43 -0
  353. mindspore/ops/_op_impl/aicpu/bincount.py +33 -0
  354. mindspore/{nn/probability/infer/variational/__init__.py → ops/_op_impl/aicpu/cauchy.py} +17 -10
  355. mindspore/ops/_op_impl/aicpu/channel_shuffle.py +40 -0
  356. mindspore/ops/_op_impl/aicpu/cholesky.py +1 -1
  357. mindspore/ops/_op_impl/{cpu/bias_add.py → aicpu/choleskygrad.py} +9 -7
  358. mindspore/ops/_op_impl/aicpu/combined_non_max_suppression.py +42 -0
  359. mindspore/ops/_op_impl/aicpu/concat_offset.py +42 -0
  360. mindspore/ops/_op_impl/aicpu/concat_offset_v1.py +31 -0
  361. mindspore/ops/_op_impl/aicpu/conj.py +11 -0
  362. mindspore/ops/_op_impl/aicpu/crop_and_resize_grad_image.py +38 -0
  363. mindspore/ops/_op_impl/aicpu/cumulative_logsumexp.py +36 -0
  364. mindspore/ops/_op_impl/aicpu/deformable_offsets.py +38 -0
  365. mindspore/ops/_op_impl/aicpu/deformable_offsets_grad.py +2 -2
  366. mindspore/ops/_op_impl/aicpu/dense_to_sparse_set_operation.py +48 -0
  367. mindspore/ops/_op_impl/aicpu/diag.py +36 -0
  368. mindspore/ops/_op_impl/aicpu/diag_part.py +36 -0
  369. mindspore/ops/_op_impl/aicpu/diagonal.py +35 -0
  370. mindspore/ops/_op_impl/{cpu/bias_add_grad.py → aicpu/digamma.py} +9 -7
  371. mindspore/ops/_op_impl/aicpu/eig.py +35 -0
  372. mindspore/ops/_op_impl/aicpu/fft_with_size.py +41 -0
  373. mindspore/ops/_op_impl/aicpu/flatten.py +1 -0
  374. mindspore/ops/_op_impl/aicpu/fmax.py +36 -0
  375. mindspore/ops/_op_impl/aicpu/fmin.py +37 -0
  376. mindspore/ops/_op_impl/aicpu/fractional_max_pool3d_with_fixed_ksize.py +1 -1
  377. mindspore/ops/_op_impl/aicpu/fse_decode.py +43 -0
  378. mindspore/ops/_op_impl/aicpu/glu.py +33 -0
  379. mindspore/ops/_op_impl/aicpu/glu_grad.py +34 -0
  380. mindspore/ops/_op_impl/aicpu/greater.py +41 -0
  381. mindspore/ops/_op_impl/aicpu/greater_equal.py +41 -0
  382. mindspore/ops/_op_impl/aicpu/index_put.py +50 -0
  383. mindspore/ops/_op_impl/{tbe/scatter_add_ds.py → aicpu/inplace_index_add.py} +17 -21
  384. mindspore/ops/_op_impl/aicpu/instance_norm_v2.py +41 -0
  385. mindspore/ops/_op_impl/aicpu/instance_norm_v2_grad.py +44 -0
  386. mindspore/ops/_op_impl/aicpu/layer_norm_grad_grad.py +47 -0
  387. mindspore/ops/_op_impl/aicpu/less.py +41 -0
  388. mindspore/ops/_op_impl/aicpu/less_equal.py +41 -0
  389. mindspore/ops/_op_impl/aicpu/lgamma.py +32 -0
  390. mindspore/ops/_op_impl/aicpu/log_normal_reverse.py +33 -0
  391. mindspore/ops/_op_impl/aicpu/logit.py +33 -0
  392. mindspore/ops/_op_impl/aicpu/logit_grad.py +34 -0
  393. mindspore/ops/_op_impl/aicpu/masked_fill.py +42 -0
  394. mindspore/ops/_op_impl/aicpu/masked_scatter.py +39 -0
  395. mindspore/ops/_op_impl/aicpu/matmul.py +39 -0
  396. mindspore/ops/_op_impl/aicpu/matrix_logarithm.py +31 -0
  397. mindspore/ops/_op_impl/aicpu/matrix_power.py +32 -0
  398. mindspore/ops/_op_impl/aicpu/matrix_solve_ls.py +36 -0
  399. mindspore/ops/_op_impl/aicpu/matrix_triangular_solve.py +36 -0
  400. mindspore/ops/_op_impl/aicpu/mirror_pad.py +2 -0
  401. mindspore/ops/_op_impl/aicpu/mirror_pad_grad.py +0 -4
  402. mindspore/ops/_op_impl/aicpu/mul.py +3 -1
  403. mindspore/ops/_op_impl/aicpu/multinomial.py +14 -6
  404. mindspore/ops/_op_impl/aicpu/multinomial_with_replacement.py +35 -0
  405. mindspore/ops/_op_impl/aicpu/nan_to_num.py +34 -0
  406. mindspore/ops/_op_impl/aicpu/nllloss.py +38 -0
  407. mindspore/ops/_op_impl/aicpu/nllloss_grad.py +39 -0
  408. mindspore/ops/_op_impl/aicpu/ones_like.py +0 -2
  409. mindspore/ops/_op_impl/aicpu/polar.py +32 -0
  410. mindspore/ops/_op_impl/aicpu/polygamma.py +34 -0
  411. mindspore/ops/_op_impl/aicpu/qr.py +36 -0
  412. mindspore/ops/_op_impl/aicpu/quant_dtype_cast.py +40 -0
  413. mindspore/ops/_op_impl/aicpu/quantile.py +35 -0
  414. mindspore/ops/_op_impl/aicpu/ragged_tensor_to_sparse.py +73 -0
  415. mindspore/ops/_op_impl/aicpu/ragged_tensor_to_tensor.py +74 -0
  416. mindspore/ops/_op_impl/aicpu/random_shuffle.py +3 -0
  417. mindspore/ops/_op_impl/aicpu/randperm_v2.py +41 -0
  418. mindspore/ops/_op_impl/aicpu/range.py +36 -0
  419. mindspore/ops/_op_impl/aicpu/reciprocal.py +34 -0
  420. mindspore/ops/_op_impl/aicpu/reciprocal_grad.py +35 -0
  421. mindspore/ops/_op_impl/aicpu/reduce_sum.py +57 -0
  422. mindspore/ops/_op_impl/aicpu/resize_bicubic.py +2 -8
  423. mindspore/ops/_op_impl/aicpu/resize_bicubic_grad.py +1 -1
  424. mindspore/ops/_op_impl/aicpu/resize_v2.py +68 -0
  425. mindspore/ops/_op_impl/aicpu/resize_v2_grad.py +68 -0
  426. mindspore/ops/_op_impl/aicpu/scatter_elements.py +4 -0
  427. mindspore/ops/_op_impl/aicpu/scatter_nd_update.py +2 -0
  428. mindspore/ops/_op_impl/aicpu/search_sorted.py +12 -6
  429. mindspore/ops/_op_impl/aicpu/self_adjoint_eig.py +34 -0
  430. mindspore/ops/_op_impl/aicpu/sequence_add.py +34 -0
  431. mindspore/ops/_op_impl/aicpu/sequence_add_offset.py +34 -0
  432. mindspore/ops/_op_impl/aicpu/sequence_addn.py +38 -0
  433. mindspore/ops/_op_impl/aicpu/slice_grad.py +76 -0
  434. mindspore/ops/_op_impl/aicpu/smooth_l1_loss.py +35 -0
  435. mindspore/ops/_op_impl/aicpu/smooth_l1_loss_grad.py +37 -0
  436. mindspore/ops/_op_impl/aicpu/sort.py +39 -0
  437. mindspore/ops/_op_impl/aicpu/sparse_apply_adagrad_da.py +0 -24
  438. mindspore/ops/_op_impl/aicpu/sparse_cross.py +42 -0
  439. mindspore/ops/_op_impl/aicpu/sparse_fill_empty_rows.py +63 -0
  440. mindspore/ops/_op_impl/aicpu/sparse_fill_empty_rows_grad.py +45 -0
  441. mindspore/ops/_op_impl/aicpu/sparse_matrix_mat_mul.py +56 -0
  442. mindspore/ops/_op_impl/{tbe/slice_ds.py → aicpu/sparse_segment_sum.py} +16 -24
  443. mindspore/ops/_op_impl/aicpu/sparse_segment_sum_with_num_segments.py +68 -0
  444. mindspore/ops/_op_impl/aicpu/sparse_slice.py +63 -0
  445. mindspore/ops/_op_impl/aicpu/sparse_slice_grad.py +61 -0
  446. mindspore/ops/_op_impl/aicpu/squared_difference.py +2 -0
  447. mindspore/ops/_op_impl/aicpu/strided_slice_v2.py +93 -0
  448. mindspore/ops/_op_impl/aicpu/strided_slice_v2_grad.py +66 -0
  449. mindspore/ops/_op_impl/aicpu/tensor_scatter_update.py +59 -0
  450. mindspore/ops/_op_impl/{tbe/gather_v2.py → aicpu/tile.py} +24 -24
  451. mindspore/ops/_op_impl/aicpu/tridiagonal_solve.py +35 -0
  452. mindspore/ops/_op_impl/aicpu/tril_indices.py +34 -0
  453. mindspore/ops/_op_impl/aicpu/triu_indices.py +34 -0
  454. mindspore/ops/_op_impl/aicpu/uniform.py +34 -0
  455. mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +1 -0
  456. mindspore/ops/_op_impl/aicpu/unique_consecutive.py +10 -2
  457. mindspore/ops/_op_impl/cpu/__init__.py +1 -2
  458. mindspore/ops/_op_impl/cpu/dynamic_shape.py +5 -1
  459. mindspore/ops/_op_impl/cpu/maximum_grad.py +2 -0
  460. mindspore/{compression/common/__init__.py → ops/_op_impl/cpu/pyexecute.py} +13 -8
  461. mindspore/ops/_op_impl/cpu/reduce_sum.py +8 -0
  462. mindspore/ops/_op_impl/cpu/sparse_slice.py +62 -0
  463. mindspore/ops/_op_impl/cpu/sparse_slice_grad.py +60 -0
  464. mindspore/ops/_op_impl/cpu/tensor_shape.py +5 -1
  465. mindspore/ops/_op_impl/tbe/__init__.py +27 -608
  466. mindspore/ops/_op_impl/tbe/addcdiv_ds.py +42 -0
  467. mindspore/ops/_op_impl/tbe/addcmul_ds.py +44 -0
  468. mindspore/ops/_op_impl/tbe/assign_add_ds.py +1 -0
  469. mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
  470. mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +1 -1
  471. mindspore/ops/_op_impl/tbe/basic_lstm_cell_c_state_grad_v2.py +0 -1
  472. mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
  473. mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +1 -1
  474. mindspore/ops/_op_impl/tbe/batch_to_space_nd_v2.py +41 -0
  475. mindspore/ops/_op_impl/tbe/bce_with_logits_loss.py +1 -0
  476. mindspore/ops/_op_impl/tbe/bias_add_grad.py +2 -0
  477. mindspore/ops/_op_impl/tbe/bn_infer_grad.py +4 -2
  478. mindspore/ops/_op_impl/tbe/bn_infer_grad_ds.py +40 -0
  479. mindspore/ops/_op_impl/tbe/bn_training_update.py +0 -1
  480. mindspore/ops/_op_impl/tbe/bn_training_update_ds.py +0 -1
  481. mindspore/ops/_op_impl/tbe/broadcast_to_ds.py +6 -4
  482. mindspore/ops/_op_impl/tbe/cast.py +0 -2
  483. mindspore/ops/_op_impl/tbe/cast_ds.py +3 -3
  484. mindspore/ops/_op_impl/tbe/ctc_loss_v2.py +0 -2
  485. mindspore/ops/_op_impl/tbe/ctc_loss_v2_grad.py +0 -2
  486. mindspore/ops/_op_impl/tbe/data_format_dim_map_ds.py +1 -0
  487. mindspore/ops/_op_impl/tbe/deformable_offsets.py +1 -0
  488. mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +1 -1
  489. mindspore/ops/_op_impl/tbe/dynamic_atomic_addr_clean.py +1 -1
  490. mindspore/ops/_op_impl/tbe/gather_nd.py +1 -0
  491. mindspore/ops/_op_impl/tbe/greater.py +2 -0
  492. mindspore/ops/_op_impl/tbe/{index_add.py → inplace_index_add.py} +3 -6
  493. mindspore/ops/_op_impl/tbe/layer_norm_beta_gamma_backprop_v2.py +0 -1
  494. mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py +35 -0
  495. mindspore/ops/_op_impl/tbe/npu_get_float_status_v2.py +35 -0
  496. mindspore/ops/_op_impl/tbe/one_hot_ds.py +0 -6
  497. mindspore/ops/_op_impl/tbe/{greater_ds.py → reduce_all_ds.py} +13 -16
  498. mindspore/ops/_op_impl/tbe/reduce_any_ds.py +39 -0
  499. mindspore/ops/_op_impl/tbe/roi_align_ds.py +44 -0
  500. mindspore/ops/_op_impl/tbe/roi_align_grad_ds.py +44 -0
  501. mindspore/ops/_op_impl/tbe/scatter_add.py +2 -0
  502. mindspore/ops/_op_impl/tbe/scatter_nd_add.py +2 -2
  503. mindspore/ops/_op_impl/tbe/slice.py +26 -15
  504. mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
  505. mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +1 -1
  506. mindspore/ops/_op_impl/tbe/strided_slice_grad_d.py +1 -0
  507. mindspore/ops/_op_impl/tbe/trans_data_ds.py +15 -5
  508. mindspore/ops/_op_impl/tbe/unsorted_segment_sum.py +1 -1
  509. mindspore/ops/_op_impl/tbe/unsorted_segment_sum_ds.py +2 -0
  510. mindspore/ops/_primitive_cache.py +3 -2
  511. mindspore/ops/_register_for_op.py +11 -0
  512. mindspore/ops/_utils/__init__.py +1 -1
  513. mindspore/ops/_utils/utils.py +20 -41
  514. mindspore/ops/_vmap/__init__.py +2 -2
  515. mindspore/ops/_vmap/vmap_array_ops.py +170 -78
  516. mindspore/ops/_vmap/vmap_base.py +24 -10
  517. mindspore/ops/_vmap/vmap_convolution_ops.py +7 -10
  518. mindspore/ops/_vmap/vmap_grad_math_ops.py +4 -4
  519. mindspore/ops/_vmap/vmap_grad_nn_ops.py +41 -9
  520. mindspore/ops/_vmap/vmap_image_ops.py +52 -0
  521. mindspore/ops/_vmap/vmap_math_ops.py +77 -6
  522. mindspore/ops/_vmap/vmap_nn_ops.py +78 -29
  523. mindspore/ops/_vmap/vmap_other_ops.py +3 -1
  524. mindspore/ops/_vmap/vmap_random_ops.py +55 -3
  525. mindspore/ops/_vmap/vmap_sparse_ops.py +1 -0
  526. mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
  527. mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
  528. mindspore/ops/bprop_mindir/ApproximateEqual_bprop.mindir +18 -19
  529. mindspore/ops/bprop_mindir/Argmax_bprop.mindir +13 -12
  530. mindspore/ops/bprop_mindir/Argmin_bprop.mindir +14 -13
  531. mindspore/ops/bprop_mindir/AssignSub_bprop.mindir +17 -18
  532. mindspore/ops/bprop_mindir/Assign_bprop.mindir +16 -16
  533. mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +150 -0
  534. mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +66 -0
  535. mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
  536. mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +13 -12
  537. mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
  538. mindspore/ops/bprop_mindir/BatchToSpaceND_bprop.mindir +28 -0
  539. mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
  540. mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +33 -0
  541. mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +306 -0
  542. mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +12 -8
  543. mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
  544. mindspore/ops/bprop_mindir/Concat_bprop.mindir +0 -0
  545. mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +240 -0
  546. mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +247 -0
  547. mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +247 -0
  548. mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +315 -0
  549. mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +278 -0
  550. mindspore/ops/bprop_mindir/DType_bprop.mindir +12 -12
  551. mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +58 -0
  552. mindspore/ops/bprop_mindir/Depend_bprop.mindir +12 -13
  553. mindspore/ops/bprop_mindir/DepthToSpace_bprop.mindir +23 -0
  554. mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +138 -0
  555. mindspore/ops/bprop_mindir/DiagPart_bprop.mindir +15 -0
  556. mindspore/ops/bprop_mindir/Dropout2D_bprop.mindir +0 -0
  557. mindspore/ops/bprop_mindir/Dropout3D_bprop.mindir +0 -0
  558. mindspore/ops/bprop_mindir/DropoutDoMask_bprop.mindir +22 -24
  559. mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +16 -14
  560. mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +27 -0
  561. mindspore/ops/bprop_mindir/Dropout_bprop.mindir +0 -0
  562. mindspore/ops/bprop_mindir/DynamicGRUV2_bprop.mindir +0 -0
  563. mindspore/ops/bprop_mindir/DynamicRNN_bprop.mindir +0 -0
  564. mindspore/ops/bprop_mindir/DynamicShape_bprop.mindir +12 -12
  565. mindspore/ops/bprop_mindir/Elu_bprop.mindir +16 -0
  566. mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
  567. mindspore/ops/bprop_mindir/Equal_bprop.mindir +18 -19
  568. mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +58 -0
  569. mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +16 -0
  570. mindspore/ops/bprop_mindir/Flatten_bprop.mindir +54 -0
  571. mindspore/ops/bprop_mindir/FloorDiv_bprop.mindir +18 -15
  572. mindspore/ops/bprop_mindir/GatherD_bprop.mindir +26 -0
  573. mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +57 -0
  574. mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
  575. mindspore/ops/bprop_mindir/GreaterEqual_bprop.mindir +17 -18
  576. mindspore/ops/bprop_mindir/Greater_bprop.mindir +18 -19
  577. mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +16 -0
  578. mindspore/ops/bprop_mindir/HSwish_bprop.mindir +16 -0
  579. mindspore/ops/bprop_mindir/IOU_bprop.mindir +18 -19
  580. mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
  581. mindspore/ops/bprop_mindir/IsFinite_bprop.mindir +13 -12
  582. mindspore/ops/bprop_mindir/IsInf_bprop.mindir +13 -10
  583. mindspore/ops/bprop_mindir/IsNan_bprop.mindir +14 -11
  584. mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +126 -0
  585. mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +15 -0
  586. mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +30 -0
  587. mindspore/ops/bprop_mindir/LRN_bprop.mindir +43 -0
  588. mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
  589. mindspore/ops/bprop_mindir/LessEqual_bprop.mindir +18 -19
  590. mindspore/ops/bprop_mindir/Less_bprop.mindir +17 -18
  591. mindspore/ops/bprop_mindir/LinSpace_bprop.mindir +22 -19
  592. mindspore/ops/bprop_mindir/Load_bprop.mindir +12 -13
  593. mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +23 -0
  594. mindspore/ops/bprop_mindir/LogicalAnd_bprop.mindir +17 -18
  595. mindspore/ops/bprop_mindir/LogicalNot_bprop.mindir +14 -13
  596. mindspore/ops/bprop_mindir/MaskedSelect_bprop.mindir +21 -0
  597. mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +74 -0
  598. mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +74 -0
  599. mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +75 -0
  600. mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +65 -0
  601. mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
  602. mindspore/ops/bprop_mindir/Maximum_bprop.mindir +0 -0
  603. mindspore/ops/bprop_mindir/Minimum_bprop.mindir +0 -0
  604. mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +27 -0
  605. mindspore/ops/bprop_mindir/Mish_bprop.mindir +35 -0
  606. mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
  607. mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
  608. mindspore/ops/bprop_mindir/NonZero_bprop.mindir +14 -0
  609. mindspore/ops/bprop_mindir/NotEqual_bprop.mindir +18 -19
  610. mindspore/ops/bprop_mindir/OneHot_bprop.mindir +25 -23
  611. mindspore/ops/bprop_mindir/OnesLike_bprop.mindir +13 -13
  612. mindspore/ops/bprop_mindir/PReLU_bprop.mindir +0 -0
  613. mindspore/ops/bprop_mindir/Pad_bprop.mindir +0 -0
  614. mindspore/ops/bprop_mindir/Padding_bprop.mindir +0 -0
  615. mindspore/ops/bprop_mindir/RNNTLoss_bprop.mindir +29 -0
  616. mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +82 -0
  617. mindspore/ops/bprop_mindir/Range_bprop.mindir +21 -19
  618. mindspore/ops/bprop_mindir/Rank_bprop.mindir +11 -11
  619. mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +16 -0
  620. mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
  621. mindspore/ops/bprop_mindir/ReduceAll_bprop.mindir +18 -17
  622. mindspore/ops/bprop_mindir/ReduceAny_bprop.mindir +18 -17
  623. mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +19 -23
  624. mindspore/ops/bprop_mindir/Reshape_bprop.mindir +60 -0
  625. mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +29 -0
  626. mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +89 -0
  627. mindspore/ops/bprop_mindir/ReverseSequence_bprop.mindir +52 -0
  628. mindspore/ops/bprop_mindir/ReverseV2_bprop.mindir +22 -0
  629. mindspore/ops/bprop_mindir/Round_bprop.mindir +14 -13
  630. mindspore/ops/bprop_mindir/ScatterMax_bprop.mindir +0 -0
  631. mindspore/ops/bprop_mindir/ScatterMin_bprop.mindir +0 -0
  632. mindspore/ops/bprop_mindir/ScatterNdUpdate_bprop.mindir +22 -0
  633. mindspore/ops/bprop_mindir/ScatterNd_bprop.mindir +24 -0
  634. mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +22 -0
  635. mindspore/ops/bprop_mindir/ScatterUpdate_bprop.mindir +0 -0
  636. mindspore/ops/bprop_mindir/SeLU_bprop.mindir +21 -0
  637. mindspore/ops/bprop_mindir/Select_bprop.mindir +30 -34
  638. mindspore/ops/bprop_mindir/Shape_bprop.mindir +12 -12
  639. mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +21 -0
  640. mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
  641. mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +16 -0
  642. mindspore/ops/bprop_mindir/Sign_bprop.mindir +13 -12
  643. mindspore/ops/bprop_mindir/Slice_bprop.mindir +26 -0
  644. mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +36 -0
  645. mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
  646. mindspore/ops/bprop_mindir/Softplus_bprop.mindir +16 -0
  647. mindspore/ops/bprop_mindir/Softsign_bprop.mindir +33 -0
  648. mindspore/ops/bprop_mindir/Sort_bprop.mindir +0 -0
  649. mindspore/ops/bprop_mindir/SpaceToBatchND_bprop.mindir +28 -0
  650. mindspore/ops/bprop_mindir/SpaceToDepth_bprop.mindir +23 -0
  651. mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
  652. mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
  653. mindspore/ops/bprop_mindir/Split_bprop.mindir +22 -0
  654. mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +54 -0
  655. mindspore/ops/bprop_mindir/StridedSliceGrad_bprop.mindir +95 -0
  656. mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +98 -0
  657. mindspore/ops/bprop_mindir/Switch_bprop.mindir +28 -32
  658. mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
  659. mindspore/ops/bprop_mindir/Tanh_bprop.mindir +66 -0
  660. mindspore/ops/bprop_mindir/TensorScatterAdd_bprop.mindir +22 -0
  661. mindspore/ops/bprop_mindir/TensorScatterUpdate_bprop.mindir +29 -0
  662. mindspore/ops/bprop_mindir/TensorShape_bprop.mindir +14 -0
  663. mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
  664. mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
  665. mindspore/ops/bprop_mindir/TransShape_bprop.mindir +23 -0
  666. mindspore/ops/bprop_mindir/TruncateDiv_bprop.mindir +18 -15
  667. mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +11 -13
  668. mindspore/ops/bprop_mindir/Unique_bprop.mindir +16 -0
  669. mindspore/ops/bprop_mindir/Unstack_bprop.mindir +22 -0
  670. mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +32 -0
  671. mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +38 -0
  672. mindspore/ops/bprop_mindir/ZerosLike_bprop.mindir +13 -12
  673. mindspore/ops/bprop_mindir/__init__.py +1 -4
  674. mindspore/ops/bprop_mindir/generate_mindir.py +32 -20
  675. mindspore/ops/composite/__init__.py +12 -13
  676. mindspore/ops/composite/base.py +261 -254
  677. mindspore/ops/composite/env_ops.py +41 -0
  678. mindspore/ops/composite/math_ops.py +197 -156
  679. mindspore/ops/composite/multitype_ops/_compile_utils.py +428 -176
  680. mindspore/ops/composite/multitype_ops/_constexpr_utils.py +188 -87
  681. mindspore/ops/composite/multitype_ops/add_impl.py +23 -1
  682. mindspore/ops/composite/multitype_ops/div_impl.py +3 -3
  683. mindspore/ops/composite/multitype_ops/equal_impl.py +1 -0
  684. mindspore/ops/composite/multitype_ops/floordiv_impl.py +1 -1
  685. mindspore/ops/composite/multitype_ops/getitem_impl.py +52 -5
  686. mindspore/ops/composite/multitype_ops/greater_equal_impl.py +31 -0
  687. mindspore/ops/composite/multitype_ops/greater_impl.py +31 -0
  688. mindspore/ops/composite/multitype_ops/in_impl.py +15 -3
  689. mindspore/ops/composite/multitype_ops/less_equal_impl.py +33 -2
  690. mindspore/ops/composite/multitype_ops/less_impl.py +33 -0
  691. mindspore/ops/composite/multitype_ops/logical_and_impl.py +2 -2
  692. mindspore/ops/composite/multitype_ops/logical_or_impl.py +2 -1
  693. mindspore/ops/composite/multitype_ops/mod_impl.py +1 -1
  694. mindspore/ops/composite/multitype_ops/mul_impl.py +21 -7
  695. mindspore/ops/composite/multitype_ops/not_in_impl.py +15 -3
  696. mindspore/ops/composite/multitype_ops/ones_like_impl.py +2 -4
  697. mindspore/ops/composite/multitype_ops/pow_impl.py +1 -0
  698. mindspore/ops/composite/multitype_ops/setitem_impl.py +62 -70
  699. mindspore/ops/composite/multitype_ops/sub_impl.py +3 -3
  700. mindspore/ops/composite/multitype_ops/zeros_like_impl.py +41 -4
  701. mindspore/ops/function/__init__.py +323 -8
  702. mindspore/ops/function/array_func.py +3511 -780
  703. mindspore/ops/function/clip_func.py +329 -0
  704. mindspore/ops/function/debug_func.py +6 -6
  705. mindspore/ops/function/grad/__init__.py +5 -1
  706. mindspore/ops/function/grad/grad_func.py +736 -65
  707. mindspore/ops/function/image_func.py +270 -0
  708. mindspore/ops/function/linalg_func.py +268 -8
  709. mindspore/ops/function/math_func.py +8032 -3164
  710. mindspore/ops/function/nn_func.py +5619 -1855
  711. mindspore/ops/function/other_func.py +115 -0
  712. mindspore/ops/function/parameter_func.py +11 -10
  713. mindspore/ops/function/random_func.py +939 -77
  714. mindspore/ops/function/sparse_func.py +249 -84
  715. mindspore/ops/function/sparse_unary_func.py +2303 -0
  716. mindspore/ops/function/spectral_func.py +146 -0
  717. mindspore/ops/function/vmap_func.py +114 -0
  718. mindspore/ops/functional.py +182 -254
  719. mindspore/ops/op_info_register.py +79 -34
  720. mindspore/ops/operations/__init__.py +210 -118
  721. mindspore/ops/operations/_csr_ops.py +7 -7
  722. mindspore/ops/operations/_embedding_cache_ops.py +25 -15
  723. mindspore/ops/operations/_grad_ops.py +447 -322
  724. mindspore/ops/operations/_inner_ops.py +547 -176
  725. mindspore/ops/operations/_map_tensor_ops.py +112 -0
  726. mindspore/ops/operations/_ms_kernel.py +29 -27
  727. mindspore/ops/operations/_ocr_ops.py +11 -11
  728. mindspore/ops/operations/_opaque_predicate_registry.py +41 -0
  729. mindspore/ops/operations/_quant_ops.py +186 -101
  730. mindspore/ops/operations/_rl_inner_ops.py +122 -61
  731. mindspore/ops/operations/_scalar_ops.py +466 -0
  732. mindspore/ops/operations/_sequence_ops.py +1047 -0
  733. mindspore/ops/operations/_tensor_array.py +10 -11
  734. mindspore/ops/operations/_thor_ops.py +4 -4
  735. mindspore/ops/operations/array_ops.py +1428 -1226
  736. mindspore/ops/operations/comm_ops.py +180 -117
  737. mindspore/ops/operations/control_ops.py +4 -2
  738. mindspore/ops/operations/custom_ops.py +185 -98
  739. mindspore/ops/operations/debug_ops.py +92 -54
  740. mindspore/ops/operations/image_ops.py +406 -211
  741. mindspore/ops/operations/inner_ops.py +42 -53
  742. mindspore/ops/operations/linalg_ops.py +32 -29
  743. mindspore/ops/operations/math_ops.py +2076 -897
  744. mindspore/ops/operations/nn_ops.py +1282 -1252
  745. mindspore/ops/operations/other_ops.py +124 -278
  746. mindspore/ops/operations/random_ops.py +345 -178
  747. mindspore/ops/operations/rl_ops.py +8 -9
  748. mindspore/ops/operations/sparse_ops.py +502 -157
  749. mindspore/ops/operations/spectral_ops.py +107 -0
  750. mindspore/ops/primitive.py +192 -15
  751. mindspore/ops/vm_impl_registry.py +23 -2
  752. mindspore/parallel/__init__.py +6 -1
  753. mindspore/parallel/_auto_parallel_context.py +199 -92
  754. mindspore/parallel/_cell_wrapper.py +4 -2
  755. mindspore/parallel/_cost_model_context.py +3 -0
  756. mindspore/parallel/_dp_allreduce_fusion.py +2 -1
  757. mindspore/parallel/_offload_context.py +185 -0
  758. mindspore/parallel/_parallel_serialization.py +167 -28
  759. mindspore/parallel/_ps_context.py +9 -5
  760. mindspore/parallel/_recovery_context.py +1 -1
  761. mindspore/parallel/_tensor.py +9 -1
  762. mindspore/{nn/transformer → parallel/_transformer}/__init__.py +6 -6
  763. mindspore/{nn/transformer → parallel/_transformer}/layers.py +59 -37
  764. mindspore/{nn/transformer → parallel/_transformer}/loss.py +4 -7
  765. mindspore/{nn/transformer → parallel/_transformer}/moe.py +160 -35
  766. mindspore/{nn/transformer → parallel/_transformer}/op_parallel_config.py +3 -3
  767. mindspore/{nn/transformer → parallel/_transformer}/transformer.py +235 -196
  768. mindspore/parallel/_utils.py +47 -7
  769. mindspore/parallel/algo_parameter_config.py +5 -1
  770. mindspore/parallel/checkpoint_transform.py +329 -0
  771. mindspore/parallel/shard.py +229 -0
  772. mindspore/perf_msvcbuildinsights.dll +0 -0
  773. mindspore/pgodb140.dll +0 -0
  774. mindspore/pgort140.dll +0 -0
  775. mindspore/profiler/__init__.py +2 -1
  776. mindspore/profiler/common/util.py +4 -3
  777. mindspore/profiler/common/validator/validate_path.py +2 -2
  778. mindspore/profiler/envprofiling.py +249 -0
  779. mindspore/profiler/parser/aicpu_data_parser.py +38 -39
  780. mindspore/profiler/parser/ascend_timeline_generator.py +497 -0
  781. mindspore/profiler/parser/base_timeline_generator.py +471 -0
  782. mindspore/profiler/parser/cpu_gpu_timeline_generator.py +684 -0
  783. mindspore/profiler/parser/framework_parser.py +42 -16
  784. mindspore/profiler/parser/hccl_parser.py +158 -158
  785. mindspore/profiler/parser/hwts_log_parser.py +7 -6
  786. mindspore/profiler/parser/integrator.py +18 -1579
  787. mindspore/profiler/parser/minddata_analyzer.py +8 -8
  788. mindspore/profiler/parser/msadvisor_analyzer.py +14 -27
  789. mindspore/profiler/parser/msadvisor_parser.py +2 -4
  790. mindspore/profiler/parser/optime_parser.py +17 -18
  791. mindspore/profiler/parser/profiler_info.py +108 -0
  792. mindspore/profiler/parser/step_trace_parser.py +1 -1
  793. mindspore/profiler/profiling.py +396 -194
  794. mindspore/rewrite/__init__.py +6 -2
  795. mindspore/rewrite/api/node.py +51 -110
  796. mindspore/rewrite/api/node_type.py +10 -6
  797. mindspore/rewrite/api/pattern_engine.py +51 -7
  798. mindspore/rewrite/api/scoped_value.py +64 -53
  799. mindspore/rewrite/api/symbol_tree.py +108 -61
  800. mindspore/rewrite/api/tree_node_helper.py +2 -3
  801. mindspore/{compression/quant/__init__.py → rewrite/ast_creator_register.py} +20 -11
  802. mindspore/rewrite/ast_helpers/__init__.py +6 -3
  803. mindspore/rewrite/ast_helpers/ast_creator.py +115 -0
  804. mindspore/rewrite/ast_helpers/ast_finder.py +99 -1
  805. mindspore/rewrite/ast_helpers/ast_modifier.py +17 -4
  806. mindspore/rewrite/ast_helpers/ast_replacer.py +1 -1
  807. mindspore/rewrite/ast_transformers/__init__.py +0 -1
  808. mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +46 -5
  809. mindspore/rewrite/ast_transformers/remove_return_out_of_if.py +6 -3
  810. mindspore/rewrite/common/__init__.py +2 -0
  811. mindspore/rewrite/common/event.py +1 -1
  812. mindspore/rewrite/common/observable.py +1 -1
  813. mindspore/rewrite/common/observer.py +1 -1
  814. mindspore/rewrite/common/rewrite_elog.py +35 -0
  815. mindspore/rewrite/namer.py +2 -2
  816. mindspore/rewrite/namespace.py +14 -4
  817. mindspore/rewrite/node.py +161 -13
  818. mindspore/rewrite/parser.py +0 -1
  819. mindspore/rewrite/parser_register.py +0 -1
  820. mindspore/rewrite/parsers/arguments_parser.py +3 -2
  821. mindspore/rewrite/parsers/assign_parser.py +267 -67
  822. mindspore/rewrite/parsers/attribute_parser.py +56 -0
  823. mindspore/rewrite/parsers/class_def_parser.py +191 -108
  824. mindspore/rewrite/parsers/constant_parser.py +101 -0
  825. mindspore/rewrite/parsers/container_parser.py +88 -0
  826. mindspore/rewrite/parsers/for_parser.py +28 -15
  827. mindspore/rewrite/parsers/function_def_parser.py +21 -5
  828. mindspore/rewrite/parsers/if_parser.py +11 -28
  829. mindspore/rewrite/parsers/module_parser.py +9 -6
  830. mindspore/rewrite/parsers/return_parser.py +3 -2
  831. mindspore/rewrite/sparsify/__init__.py +0 -0
  832. mindspore/rewrite/sparsify/sparse_transformer.py +448 -0
  833. mindspore/rewrite/sparsify/sparsify.py +109 -0
  834. mindspore/rewrite/sparsify/utils.py +173 -0
  835. mindspore/rewrite/symbol_tree.py +322 -109
  836. mindspore/rewrite/symbol_tree_builder.py +45 -8
  837. mindspore/rewrite/symbol_tree_dumper.py +0 -1
  838. mindspore/rewrite/topological_manager.py +1 -2
  839. mindspore/run_check/_check_version.py +209 -112
  840. mindspore/run_check/run_check.py +2 -1
  841. mindspore/tbbmalloc.dll +0 -0
  842. mindspore/tinyxml2.dll +0 -0
  843. mindspore/train/__init__.py +6 -4
  844. mindspore/train/_utils.py +28 -5
  845. mindspore/train/amp.py +321 -50
  846. mindspore/train/callback/__init__.py +3 -1
  847. mindspore/train/callback/_backup_and_restore.py +120 -0
  848. mindspore/train/callback/_callback.py +8 -8
  849. mindspore/train/callback/_checkpoint.py +12 -9
  850. mindspore/train/callback/_early_stop.py +13 -7
  851. mindspore/train/callback/_history.py +8 -8
  852. mindspore/train/callback/_lambda_callback.py +6 -6
  853. mindspore/train/callback/_landscape.py +36 -38
  854. mindspore/train/callback/_loss_monitor.py +12 -6
  855. mindspore/train/callback/_lr_scheduler_callback.py +2 -4
  856. mindspore/train/callback/_on_request_exit.py +212 -0
  857. mindspore/train/callback/_reduce_lr_on_plateau.py +13 -7
  858. mindspore/train/callback/_summary_collector.py +27 -19
  859. mindspore/train/callback/_time_monitor.py +13 -7
  860. mindspore/train/checkpoint_pb2.py +68 -8
  861. mindspore/train/data_sink.py +122 -33
  862. mindspore/train/dataset_helper.py +28 -87
  863. mindspore/train/loss_scale_manager.py +4 -7
  864. mindspore/{nn → train}/metrics/__init__.py +20 -20
  865. mindspore/{nn → train}/metrics/accuracy.py +12 -10
  866. mindspore/{nn → train}/metrics/auc.py +4 -4
  867. mindspore/{nn → train}/metrics/bleu_score.py +4 -4
  868. mindspore/{nn → train}/metrics/confusion_matrix.py +10 -8
  869. mindspore/{nn → train}/metrics/cosine_similarity.py +4 -4
  870. mindspore/{nn → train}/metrics/dice.py +6 -5
  871. mindspore/{nn → train}/metrics/error.py +7 -5
  872. mindspore/{nn → train}/metrics/fbeta.py +9 -7
  873. mindspore/{nn → train}/metrics/hausdorff_distance.py +8 -6
  874. mindspore/{nn → train}/metrics/loss.py +4 -3
  875. mindspore/{nn → train}/metrics/mean_surface_distance.py +6 -5
  876. mindspore/{nn → train}/metrics/metric.py +6 -5
  877. mindspore/{nn → train}/metrics/occlusion_sensitivity.py +4 -3
  878. mindspore/{nn → train}/metrics/perplexity.py +5 -4
  879. mindspore/{nn → train}/metrics/precision.py +5 -4
  880. mindspore/{nn → train}/metrics/recall.py +5 -4
  881. mindspore/{nn → train}/metrics/roc.py +7 -6
  882. mindspore/{nn → train}/metrics/root_mean_square_surface_distance.py +6 -5
  883. mindspore/{nn → train}/metrics/topk.py +7 -5
  884. mindspore/train/mind_ir_pb2.py +339 -32
  885. mindspore/train/model.py +113 -84
  886. mindspore/train/serialization.py +547 -167
  887. mindspore/train/summary/_summary_adapter.py +1 -1
  888. mindspore/train/summary/summary_record.py +43 -12
  889. mindspore/train/train_thor/convert_utils.py +7 -1
  890. mindspore/train/train_thor/dataset_helper.py +3 -3
  891. mindspore/train/train_thor/model_thor.py +0 -4
  892. mindspore/turbojpeg.dll +0 -0
  893. mindspore/vcmeta.dll +0 -0
  894. mindspore/vcruntime140.dll +0 -0
  895. mindspore/vcruntime140_1.dll +0 -0
  896. mindspore/version.py +1 -1
  897. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/METADATA +4 -3
  898. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/RECORD +901 -660
  899. mindspore/compression/common/constant.py +0 -124
  900. mindspore/compression/export/__init__.py +0 -19
  901. mindspore/compression/export/quant_export.py +0 -514
  902. mindspore/compression/quant/qat.py +0 -636
  903. mindspore/compression/quant/quant_utils.py +0 -462
  904. mindspore/compression/quant/quantizer.py +0 -68
  905. mindspore/libatomic-1.dll +0 -0
  906. mindspore/libgcc_s_seh-1.dll +0 -0
  907. mindspore/libgfortran-4.dll +0 -0
  908. mindspore/libgomp-1.dll +0 -0
  909. mindspore/libjpeg-62.dll +0 -0
  910. mindspore/libmindspore.dll +0 -0
  911. mindspore/libmindspore_common.dll +0 -0
  912. mindspore/libmindspore_core.dll +0 -0
  913. mindspore/libmindspore_glog.dll +0 -0
  914. mindspore/libnnacl.dll +0 -0
  915. mindspore/libopencv_core452.dll +0 -0
  916. mindspore/libopencv_imgcodecs452.dll +0 -0
  917. mindspore/libopencv_imgproc452.dll +0 -0
  918. mindspore/libquadmath-0.dll +0 -0
  919. mindspore/libsqlite3.dll +0 -0
  920. mindspore/libssp-0.dll +0 -0
  921. mindspore/libstdc++-6.dll +0 -0
  922. mindspore/libtinyxml2.dll +0 -0
  923. mindspore/libturbojpeg.dll +0 -0
  924. mindspore/libwinpthread-1.dll +0 -0
  925. mindspore/nn/layer/quant.py +0 -1868
  926. mindspore/nn/layer/rnn_utils.py +0 -90
  927. mindspore/nn/probability/dpn/__init__.py +0 -22
  928. mindspore/nn/probability/dpn/vae/__init__.py +0 -25
  929. mindspore/nn/probability/dpn/vae/cvae.py +0 -138
  930. mindspore/nn/probability/dpn/vae/vae.py +0 -122
  931. mindspore/nn/probability/infer/__init__.py +0 -22
  932. mindspore/nn/probability/infer/variational/elbo.py +0 -70
  933. mindspore/nn/probability/infer/variational/svi.py +0 -84
  934. mindspore/nn/probability/toolbox/__init__.py +0 -22
  935. mindspore/nn/probability/toolbox/anomaly_detection.py +0 -99
  936. mindspore/nn/probability/toolbox/uncertainty_evaluation.py +0 -363
  937. mindspore/nn/probability/transforms/__init__.py +0 -22
  938. mindspore/nn/probability/transforms/transform_bnn.py +0 -262
  939. mindspore/nn/probability/zhusuan/__init__.py +0 -18
  940. mindspore/nn/probability/zhusuan/framework/__init__.py +0 -18
  941. mindspore/nn/probability/zhusuan/framework/bn.py +0 -95
  942. mindspore/nn/probability/zhusuan/variational/__init__.py +0 -18
  943. mindspore/nn/probability/zhusuan/variational/elbo.py +0 -46
  944. mindspore/ops/_op_impl/tbe/bias_add_grad_ds.py +0 -52
  945. mindspore/ops/_op_impl/tbe/scatter_nd_add_ds.py +0 -43
  946. mindspore/ops/bprop_mindir/AssignAdd_bprop.mindir +0 -20
  947. mindspore/ops/bprop_mindir/Identity_bprop.mindir +0 -9
  948. mindspore/ops/bprop_mindir/LogicalOr_bprop.mindir +0 -20
  949. mindspore/ops/bprop_mindir/ReLU_bprop.mindir +0 -16
  950. mindspore/ops/bprop_mindir/UpdateState_bprop.mindir +0 -17
  951. mindspore/ops/bprop_mindir/stop_gradient_bprop.mindir +0 -12
  952. mindspore/ops/composite/array_ops.py +0 -210
  953. mindspore/ops/composite/clip_ops.py +0 -238
  954. mindspore/ops/composite/random_ops.py +0 -426
  955. mindspore/ops/composite/vmap_ops.py +0 -38
  956. mindspore/ops/operations/sponge_ops.py +0 -3531
  957. mindspore/ops/operations/sponge_update_ops.py +0 -2546
  958. mindspore/parallel/nn/__init__.py +0 -42
  959. mindspore/parallel/nn/loss.py +0 -22
  960. mindspore/parallel/nn/moe.py +0 -21
  961. mindspore/parallel/nn/op_parallel_config.py +0 -22
  962. mindspore/parallel/nn/transformer.py +0 -31
  963. mindspore/run_check/_check_deps_version.py +0 -84
  964. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/WHEEL +0 -0
  965. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/entry_points.txt +0 -0
  966. {mindspore-1.10.0.dist-info → mindspore-2.0.0rc1.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,1047 @@
1
+ # Copyright 2022-2023 Huawei Technologies Co., Ltd
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ============================================================================
15
+ """Operations for sequence"""
16
+
17
+ from mindspore.ops.primitive import Primitive, PrimitiveWithCheck, prim_attr_register
18
+ from mindspore.common import Tensor
19
+ from mindspore._c_expression import Tensor as Tensor_
20
+
21
+
22
+ class ListAppend(Primitive):
23
+ r"""
24
+ Append element to the end of list.
25
+
26
+ .. note::
27
+ This operation is used for dynamic length list and this it is only for internal used.
28
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
29
+
30
+ Inputs:
31
+ - **input_data** (List) - The list for target to append. Must be dynamic length sequence
32
+ - **target** (Any Object) - The target element to be appended. The shape and type of target must be the same as
33
+ as the element within 'input_data'.
34
+
35
+ Outputs:
36
+ Dynamic length list after append.
37
+
38
+ Raises:
39
+ TypeError: The 'input_data' is not dynamic length list.
40
+ ValueError: The shape or type of 'target' is not the same as the element within 'input_data'.
41
+
42
+ Supported Platforms:
43
+ ``Ascend`` ``GPU`` ``CPU``
44
+ """
45
+ @prim_attr_register
46
+ def __init__(self):
47
+ """Initialize ListAppend"""
48
+ self.init_prim_io_names(inputs=['input_data', 'target'], outputs=['output_data'])
49
+
50
+
51
+ class ListInsert(Primitive):
52
+ r"""
53
+ Insert element to the index of list.
54
+
55
+ .. note::
56
+ This operation is used for dynamic length list and this it is only for internal used.
57
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
58
+
59
+ Inputs:
60
+ - **input_data** (List) - The list for target to append. Must be dynamic length sequence
61
+ - **index** (Int) - The list for index to insert.
62
+ - **target** (Any Object) - The target element to be inserted. The shape and type of target must be the same as
63
+ as the element within 'input_data'.
64
+
65
+ Outputs:
66
+ Dynamic length list after insert.
67
+
68
+ Raises:
69
+ TypeError: The 'input_data' is not dynamic length list.
70
+ ValueError: The shape or type of 'target' is not the same as the element within 'input_data'.
71
+
72
+ Supported Platforms:
73
+ ``Ascend`` ``GPU`` ``CPU``
74
+ """
75
+ @prim_attr_register
76
+ def __init__(self):
77
+ """Initialize ListInsert"""
78
+ self.init_prim_io_names(inputs=['input_data', 'index', 'target'], outputs=['output_data'])
79
+
80
+
81
+ class ListAppendAndInsertGrad(Primitive):
82
+ r"""
83
+ Pop the end of element from list.
84
+
85
+ .. note::
86
+ This operation is used for dynamic length list and this it is only for internal used.
87
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
88
+
89
+ Inputs:
90
+ - **input_data** (List) - The list for target to pop. Must be dynamic length sequence
91
+ - **index** (Int) - The list for index to pop.
92
+
93
+ Outputs:
94
+ Dynamic length list after pop and the pop value.
95
+
96
+ Raises:
97
+ TypeError: The 'input_data' is not dynamic length list.
98
+
99
+ Supported Platforms:
100
+ ``Ascend`` ``GPU`` ``CPU``
101
+ """
102
+ @prim_attr_register
103
+ def __init__(self):
104
+ """Initialize ListAppendAndInsertGrad"""
105
+ self.init_prim_io_names(inputs=['input_data', 'index'], outputs=['output_data'])
106
+
107
+
108
+ class ShapeMulGrad(Primitive):
109
+ r"""
110
+ shape_mul grad.
111
+
112
+ .. note::
113
+ This operation is used for dynamic length list and this it is only for internal used.
114
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
115
+
116
+ Inputs:
117
+ - **input_data** (List) - The list for target to pop. Must be dynamic length sequence
118
+ - **dout** (Int) - The dout of shape_mul.
119
+
120
+ Outputs:
121
+ Dynamic length tuple.
122
+
123
+ Raises:
124
+ TypeError: The 'input_data' is not dynamic length list.
125
+
126
+ Supported Platforms:
127
+ ``Ascend`` ``GPU`` ``CPU``
128
+ """
129
+ @prim_attr_register
130
+ def __init__(self):
131
+ """Initialize ListAppendAndInsertGrad"""
132
+ self.init_prim_io_names(inputs=['input_data', 'dout'], outputs=['output_data'])
133
+
134
+
135
+ class shape_mul(Primitive):
136
+ r"""
137
+ shape_mul.
138
+
139
+ .. note::
140
+ This operation is used for dynamic length list and this it is only for internal used.
141
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
142
+
143
+ Inputs:
144
+ - **input_data** (List) - The list to shape_mul. Must be dynamic length sequence
145
+
146
+ Outputs:
147
+ Scalar.
148
+
149
+ Raises:
150
+ TypeError: The 'input_data' is not dynamic length list.
151
+
152
+ Supported Platforms:
153
+ ``Ascend`` ``GPU`` ``CPU``
154
+ """
155
+ @prim_attr_register
156
+ def __init__(self):
157
+ """Initialize ListAppendAndInsertGrad"""
158
+ self.init_prim_io_names(inputs=['input_data'], outputs=['output_data'])
159
+
160
+ def __call__(self, input_data):
161
+ if not isinstance(input_data, tuple):
162
+ raise ValueError("For primitive [shape_mul], input must be tuple")
163
+ out = 1
164
+ for x in input_data:
165
+ out *= x
166
+ return out
167
+
168
+
169
+ class SequenceSlice(Primitive):
170
+ r"""
171
+ Sequence slice operation.
172
+
173
+ .. note::
174
+ This it is only for internal used. The sequence input should be dynamic length sequence or at least one of
175
+ start/end/step should be variable.
176
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
177
+
178
+ Inputs:
179
+ - **seq** (Union[List, Tuple]) - The sequence to slice.
180
+ - **start** (int) - start index of slice.
181
+ - **stop** (int) - stop index of slice.
182
+ - **step** (int) - step of slice.
183
+
184
+ Outputs:
185
+ Dynamic length sequence after slice.
186
+
187
+ Raises:
188
+ TypeError: The 'seq' input is neither list or tuple.
189
+ ValueError: The 'seq' input is not dynamic length and none of start/end/step is variable.
190
+
191
+ Supported Platforms:
192
+ ``Ascend`` ``GPU`` ``CPU``
193
+ """
194
+ @prim_attr_register
195
+ def __init__(self):
196
+ """Initialize SequenceSlice"""
197
+ self.init_prim_io_names(inputs=['seq', 'start', 'stop', 'step'], outputs=['output_data'])
198
+
199
+
200
+ class InSequence(Primitive):
201
+ r"""
202
+ element in sequence.
203
+
204
+ .. note::
205
+ This operation is used for dynamic length list and this it is only for internal used.
206
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
207
+
208
+ Inputs:
209
+ - **element** (Any Object) - The element can be a tensor or scalar
210
+ - **input_data** (Sequence) - The sequence. Must be dynamic length sequence
211
+
212
+ Outputs:
213
+ element in sequence, True or False.
214
+
215
+ Raises:
216
+ TypeError: The 'input_data' is not dynamic length list.
217
+
218
+ Supported Platforms:
219
+ ``Ascend`` ``GPU`` ``CPU``
220
+ """
221
+ @prim_attr_register
222
+ def __init__(self):
223
+ """Initialize ListAppend"""
224
+ self.init_prim_io_names(inputs=['element', 'input_data'], outputs=['output_data'])
225
+
226
+
227
+ class SequenceSliceSetItem(Primitive):
228
+ r"""
229
+ Sequence slice setitem operation.
230
+
231
+ .. note::
232
+ This it is only for internal used. The sequence input should be dynamic length sequence or at least one of
233
+ start/end/step should be variable.
234
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
235
+
236
+ Inputs:
237
+ - **seq** (Union[List, Tuple]) - The sequence to perform slice setitem.
238
+ - **target** (Union[List, Tuple]) - The target item to set.
239
+ - **start** (int) - start index of slice.
240
+ - **stop** (int) - stop index of slice.
241
+ - **step** (int) - step of slice.
242
+
243
+ Outputs:
244
+ Dynamic length sequence after slice setitem.
245
+
246
+ Raises:
247
+ ValueError: The 'seq' and 'target' input is not dynamic length and none of start/end/step is variable.
248
+
249
+ Supported Platforms:
250
+ ``Ascend`` ``GPU`` ``CPU``
251
+ """
252
+ @prim_attr_register
253
+ def __init__(self):
254
+ """Initialize SequenceSliceSetItem"""
255
+ self.init_prim_io_names(inputs=['seq', 'target', 'start', 'stop', 'step'], outputs=['output_data'])
256
+
257
+
258
+ class SequenceSliceGrad(Primitive):
259
+ """Reverse of slice."""
260
+
261
+ @prim_attr_register
262
+ def __init__(self):
263
+ """Initialize SequenceSliceGrad"""
264
+ self.init_prim_io_names(inputs=['dy', 'x', 'start', 'stop', 'step'], outputs=['dx'])
265
+
266
+
267
+ class SequenceAdd(Primitive):
268
+ r"""
269
+ Add elements of two sequence together.
270
+
271
+ .. note::
272
+ This it is only for internal used. At least one of the sequence should be dynamic length sequence.
273
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
274
+
275
+ Inputs:
276
+ - **input_1** (Union[List, Tuple]) - The first sequence for sequence addition.
277
+ - **input_2** (Union[List, Tuple]) - The second sequence for sequence addition.
278
+
279
+ Outputs:
280
+ Dynamic length sequence after addition.
281
+
282
+ Raises:
283
+ TypeError: The 'input_1' and 'input_2' are not both list or tuple.
284
+ TypeError: Both of 'input_1' and 'input_2' are not dynamic length sequence.
285
+
286
+ Supported Platforms:
287
+ ``Ascend`` ``GPU`` ``CPU``
288
+ """
289
+ @prim_attr_register
290
+ def __init__(self):
291
+ """Initialize SequenceAdd"""
292
+ self.init_prim_io_names(inputs=['input_1', 'input_2'], outputs=['output_data'])
293
+
294
+
295
+ class SequenceAddOffset(Primitive):
296
+ r"""
297
+ Get offsets of SequenceAdd inputs within its output, refs to ConcatOffset.
298
+
299
+ .. note::
300
+ This it is only for internal used. At least one of the sequence should be dynamic length sequence.
301
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
302
+
303
+ Inputs:
304
+ - **input_0** (List, Tuple) - A Tuple/List.
305
+ - **input_1** (List, Tuple) - A Tuple/List.
306
+
307
+ Outputs:
308
+ A tuple of offsets of SequenceAdd inputs within its output.
309
+
310
+ Raises:
311
+ TypeError: The data in 'input_0' and 'input_1' are not both list or tuple.
312
+
313
+ Supported Platforms:
314
+ ``Ascend`` ``GPU`` ``CPU``
315
+ """
316
+ @prim_attr_register
317
+ def __init__(self):
318
+ """Initialize SequenceAddOffset"""
319
+ self.init_prim_io_names(inputs=['shape_0', 'shape_1'], outputs=['output'])
320
+
321
+
322
+ class TupleToTensor(PrimitiveWithCheck):
323
+ r"""
324
+ Convert tuple to tensor
325
+
326
+ If the type of the first number in the tuple is integer, the data type of the output tensor is int.
327
+ Otherwise, the data type of the output tensor is float.
328
+
329
+ .. note::
330
+ This it is only for internal used. The input_tuple can be constant/variable value or dynamic length tuple.
331
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
332
+
333
+ Inputs:
334
+ - **input_tuple** (Tuple) - A tuple of numbers. These numbers have the same type. The shape is :math:`(N,)`
335
+ - **dtype** (mindspore.dtype): The target data type. Default: mindspore.float32.
336
+
337
+ Outputs:
338
+ Tensor, if the input tuple contains `N` numbers, then the shape of the output tensor is (N,).
339
+
340
+ Raises:
341
+ TypeError: If `input_tuple` is not a tuple.
342
+ ValueError: If length of `input_tuple` is less than 0.
343
+
344
+ Supported Platforms:
345
+ ``Ascend`` ``GPU`` ``CPU``
346
+ """
347
+ @prim_attr_register
348
+ def __init__(self):
349
+ """Initialize TupleToTensor"""
350
+ self.init_prim_io_names(inputs=['input_tuple', 'dtype'], outputs=['output_data'])
351
+
352
+ def __call__(self, x, dtype):
353
+ return self.infer_value(x, dtype)
354
+
355
+ def infer_value(self, x, dtype):
356
+ """infer value"""
357
+ if x is None:
358
+ return None
359
+ if isinstance(x, range):
360
+ x = tuple(x)
361
+ if isinstance(x, tuple) and None not in x:
362
+ return Tensor(x, dtype=dtype)
363
+ return None
364
+
365
+
366
+ class ListToTensor(PrimitiveWithCheck):
367
+ r"""
368
+ Convert list to tensor
369
+
370
+ If the type of the first number in the list is integer, the data type of the output tensor is int.
371
+ Otherwise, the data type of the output tensor is float.
372
+
373
+ .. note::
374
+ This it is only for internal used. The input_list can be constant/variable value or dynamic length list.
375
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
376
+
377
+ Inputs:
378
+ - **input_list** (List) - A tuple of numbers. These numbers have the same type. The shape is :math:`(N,)`
379
+ - **dtype** (mindspore.dtype): The target data type. Default: mindspore.float32.
380
+
381
+ Outputs:
382
+ Tensor, if the input tuple contains `N` numbers, then the shape of the output tensor is (N,).
383
+
384
+ Raises:
385
+ TypeError: If `input_list` is not a list.
386
+ ValueError: If length of `input_list` is less than 0.
387
+
388
+ Supported Platforms:
389
+ ``Ascend`` ``GPU`` ``CPU``
390
+ """
391
+ @prim_attr_register
392
+ def __init__(self):
393
+ """Initialize ListToTensor"""
394
+ self.init_prim_io_names(inputs=['input_list', 'dtype'], outputs=['output_data'])
395
+
396
+ def __call__(self, x, dtype):
397
+ return self.infer_value(x, dtype)
398
+
399
+ def infer_value(self, x, dtype):
400
+ """Infer_value ListToTensor"""
401
+ if x is not None and None not in x and isinstance(x, list):
402
+ return Tensor(x, dtype)
403
+ return None
404
+
405
+
406
+ class TensorToTuple(PrimitiveWithCheck):
407
+ r"""
408
+ Convert tensor to tuple
409
+
410
+ .. note::
411
+ This it is only for internal used. The input_tensor can be constant/variable value.
412
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
413
+
414
+ Inputs:
415
+ - **input_tensor** (Tensor) - The input_tensor must be a 1-D Tensor.
416
+
417
+ Outputs:
418
+ Tuple, the length is equal to tensor shape.
419
+
420
+ Raises:
421
+ TypeError: If `input_tensor` is not a tensor.
422
+ ValueError: If rank of `input_tensor` is not 1.
423
+
424
+ Supported Platforms:
425
+ ``Ascend`` ``GPU`` ``CPU``
426
+ """
427
+ @prim_attr_register
428
+ def __init__(self):
429
+ """Initialize TensorToTuple"""
430
+ self.init_prim_io_names(inputs=['input_tensor'], outputs=['output_data'])
431
+
432
+ def __call__(self, x):
433
+ return self.infer_value(x)
434
+
435
+ def infer_value(self, x):
436
+ """Infer_value TensorToTuple"""
437
+ value = None
438
+ if x is not None and None not in x and isinstance(x, (Tensor, Tensor_)):
439
+ value = tuple(x.asnumpy().tolist())
440
+ return value
441
+
442
+
443
+ class TensorToList(PrimitiveWithCheck):
444
+ r"""
445
+ Convert tensor to list
446
+
447
+ .. note::
448
+ This it is only for internal used. The input_tensor can be constant/variable value.
449
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
450
+
451
+ Inputs:
452
+ - **input_tensor** (Tensor) - The input_tensor must be a 1-D Tensor.
453
+
454
+ Outputs:
455
+ List, the length is equal to tensor shape.
456
+
457
+ Raises:
458
+ TypeError: If `input_tensor` is not a tensor.
459
+ ValueError: If rank of `input_tensor` is not 1.
460
+
461
+ Supported Platforms:
462
+ ``Ascend`` ``GPU`` ``CPU``
463
+ """
464
+ @prim_attr_register
465
+ def __init__(self):
466
+ """Initialize TensorToList"""
467
+ self.init_prim_io_names(inputs=['input_tensor'], outputs=['output_data'])
468
+
469
+ def __call__(self, x):
470
+ return self.infer_value(x)
471
+
472
+ def infer_value(self, x):
473
+ """infer_value TensorToList"""
474
+ value = None
475
+ if x is not None and None not in x and isinstance(x, (Tensor, Tensor_)):
476
+ value = x.asnumpy().tolist()
477
+ return value
478
+
479
+
480
+ class TensorToScalar(PrimitiveWithCheck):
481
+ r"""
482
+ Convert tensor to scalar
483
+
484
+ .. note::
485
+ This it is only for internal used. The input_tensor can be constant/variable value.
486
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
487
+
488
+ Inputs:
489
+ - **input_tensor** (Tensor) - The input_tensor must be a 0-D Tensor.
490
+
491
+ Outputs:
492
+ Scalar, a constant or variable scalar.
493
+
494
+ Raises:
495
+ TypeError: If `input_tensor` is not a tensor.
496
+ ValueError: If rank of `input_tensor` is not 0.
497
+
498
+ Supported Platforms:
499
+ ``Ascend`` ``GPU`` ``CPU``
500
+ """
501
+ @prim_attr_register
502
+ def __init__(self):
503
+ """Initialize TensorToScalar"""
504
+ self.init_prim_io_names(inputs=['input_tensor'], outputs=['output_data'])
505
+
506
+ def __call__(self, x):
507
+ return self.infer_value(x)
508
+
509
+ def infer_value(self, x):
510
+ """infer_value TensorToScalar"""
511
+ value = None
512
+ if x is not None and None not in x and isinstance(x, (Tensor, Tensor_)):
513
+ value = x.asnumpy().tolist()
514
+ return value
515
+
516
+
517
+ class SequenceCount(Primitive):
518
+ r"""
519
+ Support sequence count operation 'seq.count(target)'.
520
+
521
+ .. note::
522
+ This it is only for internal used.
523
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
524
+
525
+ Inputs:
526
+ - **sequence** (Union[List, Tuple]) - The sequence to count elements.
527
+ - **target** (Any Object) - The target element to find in sequence.
528
+
529
+ Outputs:
530
+ Integer, counting of target element.
531
+
532
+ Raises:
533
+ TypeError: The 'sequence' is not list or tuple.
534
+
535
+ Supported Platforms:
536
+ ``Ascend`` ``GPU`` ``CPU``
537
+ """
538
+ @prim_attr_register
539
+ def __init__(self):
540
+ """Initialize SequenceCount"""
541
+ self.init_prim_io_names(inputs=['sequence', 'target'], outputs=['output_data'])
542
+
543
+
544
+ class SequenceIndex(Primitive):
545
+ r"""
546
+ Support sequence index operation 'seq.index(target)'.
547
+
548
+ .. note::
549
+ This it is only for internal used.
550
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
551
+
552
+ Inputs:
553
+ - **sequence** (Union[List, Tuple]) - The sequence to find the index value of the first occurrence.
554
+ - **target** (Any Object) - The target element to find in sequence.
555
+ - **start** (Integer) - The start index to find in sequence.
556
+ - **end** (Integer) - The end index to find in sequence.
557
+
558
+ Outputs:
559
+ Integer, the index value of the first occurrence of the target element.
560
+
561
+ Raises:
562
+ TypeError: The 'sequence' is not list or tuple.
563
+
564
+ Supported Platforms:
565
+ ``Ascend`` ``GPU`` ``CPU``
566
+ """
567
+ @prim_attr_register
568
+ def __init__(self):
569
+ """Initialize SequenceIndex"""
570
+ self.init_prim_io_names(inputs=['sequence', 'target', 'start', 'end'], outputs=['output_data'])
571
+
572
+
573
+ class SequenceMul(Primitive):
574
+ r"""
575
+ Support sequence multiplication operation 'seq.mul(scalar)'.
576
+
577
+ .. note::
578
+ This it is only for internal used.
579
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
580
+
581
+ Inputs:
582
+ - **sequence** (Union[List, Tuple]) - The sequence to count elements.
583
+ - **scalar** (Any Object) - The times to replicate the sequence.
584
+
585
+ Outputs:
586
+ List or tuple with 'scalar' times multiplication.
587
+
588
+ Raises:
589
+ TypeError: The 'sequence' is not list or tuple.
590
+ ValueError: Both 'sequence' and 'scalar' is constant.
591
+
592
+ Supported Platforms:
593
+ ``Ascend`` ``GPU`` ``CPU``
594
+ """
595
+ @prim_attr_register
596
+ def __init__(self):
597
+ """Initialize SequenceMul"""
598
+ self.init_prim_io_names(inputs=['sequence', 'scalar'], outputs=['output_data'])
599
+
600
+
601
+ class SequenceZerosLike(Primitive):
602
+ r"""
603
+ Returns a Sequence with a value of 0 and its shape and data type is the same as the input.
604
+
605
+ .. note::
606
+ This it is only for internal used.
607
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
608
+
609
+ Inputs:
610
+ - **sequence** (Union[List, Tuple]) - The sequence to count elements.
611
+
612
+ Outputs:
613
+ List or tuple filled with zeros.
614
+
615
+ Raises:
616
+ TypeError: The 'sequence' is not list or tuple.
617
+
618
+ Supported Platforms:
619
+ ``Ascend`` ``GPU`` ``CPU``
620
+ """
621
+ @prim_attr_register
622
+ def __init__(self):
623
+ """Initialize SequenceZerosLike"""
624
+ self.init_prim_io_names(inputs=['sequence'], outputs=['output_data'])
625
+
626
+
627
+ class make_range(Primitive):
628
+ r"""
629
+ Creates a sequence of numbers in range [start, limit) with step size delta.
630
+
631
+ .. note::
632
+ This it is only for internal used.
633
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
634
+
635
+ Inputs:
636
+ - **start** (Union[int, float]) - Start of interval.
637
+ - **limit** (Union[int, float]) - End of interval.
638
+ - **delta** (Union[int, float]) - Spacing between values.
639
+
640
+ Outputs:
641
+ A 1-D Sequence, with the same type as the inputs.
642
+
643
+ Raises:
644
+ TypeError: If datatype of `start`, `limit` or `delta` is not same.
645
+ TypeError: If datatype of `start`, `limit` or `delta` is not supported.
646
+ ValueError: If `delta` = 0.
647
+ ValueError: If `start` >= `limit` when `delta` > 0.
648
+ ValueError: If `start` <= `limit` when `delta` < 0.
649
+
650
+ Supported Platforms:
651
+ ``Ascend`` ``GPU`` ``CPU``
652
+ """
653
+ @prim_attr_register
654
+ def __init__(self):
655
+ """Initialize make_range"""
656
+ self.init_prim_io_names(inputs=['start', 'limit', 'delta'], outputs=['output_data'])
657
+
658
+ def __call__(self, start, limit, delta):
659
+ return range(start, limit, delta)
660
+
661
+
662
+ class tuple_equal(Primitive):
663
+ r"""
664
+ Support sequence equal operation 'equal(target)'.
665
+
666
+ .. note::
667
+ This it is only for internal used.
668
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
669
+
670
+ Inputs:
671
+ - **x** (Union[Tuple]) - The tuple.
672
+ - **y** (Union[Tuple]) - The tuple.
673
+
674
+ Outputs:
675
+ Bool.
676
+
677
+ Raises:
678
+ TypeError: The 'x' is not tuple.
679
+
680
+ Supported Platforms:
681
+ ``Ascend`` ``GPU`` ``CPU``
682
+ """
683
+ @prim_attr_register
684
+ def __init__(self):
685
+ """Initialize tuple_equal"""
686
+
687
+ def __call__(self, x, y):
688
+ return x == y
689
+
690
+
691
+ class list_equal(Primitive):
692
+ r"""
693
+ Support sequence equal operation 'equal(target)'.
694
+
695
+ .. note::
696
+ This it is only for internal used.
697
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
698
+
699
+ Inputs:
700
+ - **x** (Union[List]) - The list.
701
+ - **y** (Union[List]) - The list.
702
+
703
+ Outputs:
704
+ Bool.
705
+
706
+ Raises:
707
+ TypeError: The 'x' is not list.
708
+
709
+ Supported Platforms:
710
+ ``Ascend`` ``GPU`` ``CPU``
711
+ """
712
+ @prim_attr_register
713
+ def __init__(self):
714
+ """Initialize list_equal"""
715
+
716
+ def __call__(self, x, y):
717
+ return x == y
718
+
719
+
720
+ class sequence_len(Primitive):
721
+ r"""
722
+ Returns length of Sequence.
723
+
724
+ .. note::
725
+ This it is only for internal used.
726
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
727
+
728
+ Inputs:
729
+ - **sequence** (Union[List, Tuple]) - The sequence.
730
+
731
+ Outputs:
732
+ Integer, length of Sequence.
733
+
734
+ Raises:
735
+ TypeError: The 'sequence' is not list or tuple.
736
+
737
+ Supported Platforms:
738
+ ``Ascend`` ``GPU`` ``CPU``
739
+ """
740
+ @prim_attr_register
741
+ def __init__(self):
742
+ """Initialize sequence_len"""
743
+ self.init_prim_io_names(inputs=['sequence'], outputs=['output_data'])
744
+
745
+ def __call__(self, x):
746
+ return len(x)
747
+
748
+
749
+ class SequenceMax(Primitive):
750
+ r"""
751
+ Support sequence max operation 'max(target)'.
752
+
753
+ .. note::
754
+ This it is only for internal used.
755
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
756
+
757
+ Inputs:
758
+ - **sequence** (Union[List, Tuple]) - The sequence.
759
+
760
+ Outputs:
761
+ The maximum element.
762
+
763
+ Raises:
764
+ TypeError: The 'sequence' is not list or tuple.
765
+
766
+ Supported Platforms:
767
+ ``Ascend`` ``GPU`` ``CPU``
768
+ """
769
+ @prim_attr_register
770
+ def __init__(self):
771
+ """Initialize SequenceMax"""
772
+ self.init_prim_io_names(inputs=['sequence'], outputs=['output_data'])
773
+
774
+
775
+ class SequenceMin(Primitive):
776
+ r"""
777
+ Support sequence min operation 'min(target)'.
778
+
779
+ .. note::
780
+ This it is only for internal used.
781
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
782
+
783
+ Inputs:
784
+ - **sequence** (Union[List, Tuple]) - The sequence.
785
+
786
+ Outputs:
787
+ The minimum element.
788
+
789
+ Raises:
790
+ TypeError: The 'sequence' is not list or tuple.
791
+
792
+ Supported Platforms:
793
+ ``Ascend`` ``GPU`` ``CPU``
794
+ """
795
+ @prim_attr_register
796
+ def __init__(self):
797
+ """Initialize SequenceMax"""
798
+ self.init_prim_io_names(inputs=['sequence'], outputs=['output_data'])
799
+
800
+
801
+ class SequenceAddN(Primitive):
802
+ r"""
803
+ Support sequence AddN operation.
804
+
805
+ .. note::
806
+ This it is only for internal used.
807
+
808
+ Inputs:
809
+ - **sequence** (Union[List, Tuple]) - The sequence.
810
+
811
+ Outputs:
812
+ The addition of all input.
813
+
814
+ Raises:
815
+ TypeError: The 'sequence' is not list or tuple.
816
+
817
+ Supported Platforms:
818
+ ``Ascend`` ``GPU`` ``CPU``
819
+ """
820
+ @prim_attr_register
821
+ def __init__(self):
822
+ """Initialize SequenceAddN"""
823
+ self.init_prim_io_names(inputs=["inputs"], outputs=["sum"])
824
+
825
+
826
+ class tuple_greater_than(Primitive):
827
+ r"""
828
+ Support tuple_greater_than operation 'greater_than(target)'.
829
+
830
+ .. note::
831
+ This it is only for internal used.
832
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
833
+
834
+ Inputs:
835
+ - **input_0** (Union[Tuple]) - The first tuple.
836
+ - **input_1** (Union[Tuple]) - The second tuple.
837
+
838
+ Outputs:
839
+ A bool value to indicate whether tuple 'input_0' is greater than tuple 'input_1'.
840
+
841
+ Raises:
842
+ TypeError: The 'input_0' or 'input_1' is not list or tuple.
843
+
844
+ Supported Platforms:
845
+ ``Ascend`` ``GPU`` ``CPU``
846
+ """
847
+ @prim_attr_register
848
+ def __init__(self):
849
+ """Initialize tuple_greater_than"""
850
+ self.init_prim_io_names(
851
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
852
+
853
+
854
+ class tuple_greater_equal(Primitive):
855
+ r"""
856
+ Support tuple_greater_equal operation 'greater_equal(target)'.
857
+
858
+ .. note::
859
+ This it is only for internal used.
860
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
861
+
862
+ Inputs:
863
+ - **input_0** (Union[Tuple]) - The first tuple.
864
+ - **input_1** (Union[Tuple]) - The second tuple.
865
+
866
+ Outputs:
867
+ A bool value to indicate whether tuple 'input_0' is greater than or equal to tuple 'input_1'.
868
+
869
+ Raises:
870
+ TypeError: The 'input_0' or 'input_1' is not list or tuple.
871
+
872
+ Supported Platforms:
873
+ ``Ascend`` ``GPU`` ``CPU``
874
+ """
875
+ @prim_attr_register
876
+ def __init__(self):
877
+ """Initialize tuple_greater_equal"""
878
+ self.init_prim_io_names(
879
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
880
+
881
+
882
+ class list_greater_than(Primitive):
883
+ r"""
884
+ Support list_greater_than operation 'greater_than(target)'.
885
+
886
+ .. note::
887
+ This it is only for internal used.
888
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
889
+
890
+ Inputs:
891
+ - **input_0** (Union[List]) - The first list.
892
+ - **input_1** (Union[List]) - The second list.
893
+
894
+ Outputs:
895
+ A bool value to indicate whether list 'input_0' is greater than list 'input_1'.
896
+
897
+ Raises:
898
+ TypeError: The 'input_0' or 'input_1' is not list or list.
899
+
900
+ Supported Platforms:
901
+ ``Ascend`` ``GPU`` ``CPU``
902
+ """
903
+ @prim_attr_register
904
+ def __init__(self):
905
+ """Initialize list_greater_than"""
906
+ self.init_prim_io_names(
907
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
908
+
909
+
910
+ class list_greater_equal(Primitive):
911
+ r"""
912
+ Support list_greater_equal operation 'greater_equal(target)'.
913
+
914
+ .. note::
915
+ This it is only for internal used.
916
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
917
+
918
+ Inputs:
919
+ - **input_0** (Union[List]) - The first list.
920
+ - **input_1** (Union[List]) - The second list.
921
+
922
+ Outputs:
923
+ A bool value to indicate whether list 'input_0' is greater than or equal to list 'input_1'.
924
+
925
+ Raises:
926
+ TypeError: The 'input_0' or 'input_1' is not list or list.
927
+
928
+ Supported Platforms:
929
+ ``Ascend`` ``GPU`` ``CPU``
930
+ """
931
+ @prim_attr_register
932
+ def __init__(self):
933
+ """Initialize list_greater_equal"""
934
+ self.init_prim_io_names(
935
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
936
+
937
+
938
+ class tuple_lt(Primitive):
939
+ r"""
940
+ Support tuple less_than operation 'less_than(target)'.
941
+
942
+ .. note::
943
+ This it is only for internal used.
944
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
945
+
946
+ Inputs:
947
+ - **input_0** (Union[Tuple]) - The first sequence.
948
+ - **input_1** (Union[Tuple]) - The second sequence, dtype and shape should be same as 'input_0'.
949
+
950
+ Outputs:
951
+ A bool value to indicate whether every element in 'input_0' is less than element in 'input_1' correspondingly.
952
+
953
+ Raises:
954
+ TypeError: The 'input_0' or 'input_1' is not tuple.
955
+
956
+ Supported Platforms:
957
+ ``Ascend`` ``GPU`` ``CPU``
958
+ """
959
+ @prim_attr_register
960
+ def __init__(self):
961
+ """Initialize tuple_lt"""
962
+ self.init_prim_io_names(
963
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
964
+
965
+
966
+ class list_lt(Primitive):
967
+ r"""
968
+ Support list less_than operation 'less_than(target)'.
969
+
970
+ .. note::
971
+ This it is only for internal used.
972
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
973
+
974
+ Inputs:
975
+ - **input_0** (Union[List]) - The first sequence.
976
+ - **input_1** (Union[List]) - The second sequence, dtype and shape should be same as 'input_0'.
977
+
978
+ Outputs:
979
+ A bool value to indicate whether every element in 'input_0' is less than element in 'input_1' correspondingly.
980
+
981
+ Raises:
982
+ TypeError: The 'input_0' or 'input_1' is not list.
983
+
984
+ Supported Platforms:
985
+ ``Ascend`` ``GPU`` ``CPU``
986
+ """
987
+ @prim_attr_register
988
+ def __init__(self):
989
+ """Initialize list_lt"""
990
+ self.init_prim_io_names(
991
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
992
+
993
+
994
+ class tuple_le(Primitive):
995
+ r"""
996
+ Support tuple less_equal operation 'less_equal(target)'.
997
+
998
+ .. note::
999
+ This it is only for internal used.
1000
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
1001
+
1002
+ Inputs:
1003
+ - **input_0** (Union[Tuple]) - The first sequence.
1004
+ - **input_1** (Union[Tuple]) - The second sequence, dtype and shape should be same as 'input_0'.
1005
+
1006
+ Outputs:
1007
+ A bool value to indicate whether every element in 'input_0' is less equal element in 'input_1' correspondingly.
1008
+
1009
+ Raises:
1010
+ TypeError: The 'input_0' or 'input_1' is not tuple.
1011
+
1012
+ Supported Platforms:
1013
+ ``Ascend`` ``GPU`` ``CPU``
1014
+ """
1015
+ @prim_attr_register
1016
+ def __init__(self):
1017
+ """Initialize tuple_le"""
1018
+ self.init_prim_io_names(
1019
+ inputs=['input_0', 'input_1'], outputs=['output_data'])
1020
+
1021
+
1022
+ class list_le(Primitive):
1023
+ r"""
1024
+ Support list less equal operation 'less_equal(target)'.
1025
+
1026
+ .. note::
1027
+ This it is only for internal used.
1028
+ This primitive only have 'CPU' implementation, for other platform, it runs using heterogeneous.
1029
+
1030
+ Inputs:
1031
+ - **input_0** (Union[List]) - The first sequence.
1032
+ - **input_1** (Union[List]) - The second sequence, dtype and shape should be same as 'input_0'.
1033
+
1034
+ Outputs:
1035
+ A bool value to indicate whether every element in 'input_0' is less equal element in 'input_1' correspondingly.
1036
+
1037
+ Raises:
1038
+ TypeError: The 'input_0' or 'input_1' is not list.
1039
+
1040
+ Supported Platforms:
1041
+ ``Ascend`` ``GPU`` ``CPU``
1042
+ """
1043
+ @prim_attr_register
1044
+ def __init__(self):
1045
+ """Initialize list_le"""
1046
+ self.init_prim_io_names(
1047
+ inputs=['input_0', 'input_1'], outputs=['output_data'])