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
@@ -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'])