mindstudio-probe 1.0.1__py3-none-any.whl → 1.0.4__py3-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.
Files changed (323) hide show
  1. {mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/LICENSE +201 -201
  2. {mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/METADATA +36 -30
  3. mindstudio_probe-1.0.4.dist-info/RECORD +276 -0
  4. {mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/WHEEL +1 -1
  5. {mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/entry_points.txt +1 -0
  6. msprobe/README.md +101 -182
  7. msprobe/__init__.py +1 -0
  8. msprobe/{config/config.json → config.json} +49 -27
  9. msprobe/core/__init__.py +0 -0
  10. msprobe/{pytorch → core}/advisor/advisor.py +124 -124
  11. msprobe/{pytorch → core}/advisor/advisor_const.py +59 -59
  12. msprobe/{pytorch → core}/advisor/advisor_result.py +58 -58
  13. msprobe/core/common/const.py +341 -241
  14. msprobe/core/common/exceptions.py +100 -88
  15. msprobe/core/common/{file_check.py → file_utils.py} +478 -265
  16. msprobe/core/common/log.py +76 -55
  17. msprobe/core/common/utils.py +385 -516
  18. msprobe/core/common_config.py +85 -58
  19. msprobe/core/compare/acc_compare.py +300 -0
  20. msprobe/core/compare/check.py +95 -0
  21. msprobe/core/compare/compare_cli.py +49 -0
  22. msprobe/core/compare/highlight.py +223 -0
  23. msprobe/core/compare/multiprocessing_compute.py +149 -0
  24. msprobe/{pytorch → core}/compare/npy_compare.py +295 -244
  25. msprobe/core/compare/utils.py +430 -0
  26. msprobe/core/data_dump/data_collector.py +154 -140
  27. msprobe/core/data_dump/data_processor/base.py +314 -245
  28. msprobe/core/data_dump/data_processor/factory.py +59 -61
  29. msprobe/core/data_dump/data_processor/mindspore_processor.py +186 -0
  30. msprobe/core/data_dump/data_processor/pytorch_processor.py +366 -346
  31. msprobe/core/data_dump/json_writer.py +96 -116
  32. msprobe/core/data_dump/scope.py +178 -178
  33. msprobe/core/grad_probe/__init__.py +0 -0
  34. msprobe/core/grad_probe/constant.py +71 -0
  35. msprobe/core/grad_probe/grad_compare.py +171 -0
  36. msprobe/core/grad_probe/utils.py +64 -0
  37. msprobe/docs/01.installation.md +89 -0
  38. msprobe/docs/02.config_introduction.md +165 -0
  39. msprobe/docs/03.config_examples.md +247 -0
  40. msprobe/docs/04.acl_config_examples.md +76 -0
  41. msprobe/docs/05.data_dump_PyTorch.md +198 -0
  42. msprobe/docs/06.data_dump_MindSpore.md +243 -0
  43. msprobe/docs/07.accuracy_checker_PyTorch.md +274 -0
  44. msprobe/docs/08.accuracy_checker_online_PyTorch.md +198 -0
  45. msprobe/docs/09.accuracy_checker_MindSpore.md +68 -0
  46. msprobe/docs/10.accuracy_compare_PyTorch.md +245 -0
  47. msprobe/docs/11.accuracy_compare_MindSpore.md +202 -0
  48. msprobe/docs/12.overflow_check_PyTorch.md +79 -0
  49. msprobe/docs/13.overflow_check_MindSpore.md +31 -0
  50. msprobe/{pytorch/doc/parse_tool.md → docs/14.data_parse_PyTorch.md} +283 -286
  51. msprobe/docs/15.free_benchmarking_PyTorch.md +164 -0
  52. msprobe/docs/17.grad_probe.md +207 -0
  53. msprobe/docs/FAQ_PyTorch.md +177 -0
  54. msprobe/docs/S02.report_free_benchmarking_validation_performance_baseline.md +146 -0
  55. msprobe/docs/img/free_benchmark_framework.png +0 -0
  56. msprobe/docs/img/grad_probe_image-1.png +0 -0
  57. msprobe/docs/img/grad_probe_image-2.png +0 -0
  58. msprobe/docs/img/grad_probe_image-3.png +0 -0
  59. msprobe/docs/img/grad_probe_image-4.png +0 -0
  60. msprobe/docs/img/grad_probe_image.png +0 -0
  61. msprobe/mindspore/__init__.py +1 -1
  62. msprobe/mindspore/api_accuracy_checker/__init__.py +0 -0
  63. msprobe/mindspore/api_accuracy_checker/api_accuracy_checker.py +255 -0
  64. msprobe/mindspore/api_accuracy_checker/api_info.py +69 -0
  65. msprobe/mindspore/api_accuracy_checker/api_runner.py +156 -0
  66. msprobe/mindspore/api_accuracy_checker/base_compare_algorithm.py +197 -0
  67. msprobe/mindspore/api_accuracy_checker/cmd_parser.py +6 -0
  68. msprobe/mindspore/api_accuracy_checker/compute_element.py +239 -0
  69. msprobe/mindspore/api_accuracy_checker/main.py +9 -0
  70. msprobe/mindspore/api_accuracy_checker/type_mapping.py +114 -0
  71. msprobe/mindspore/api_accuracy_checker/utils.py +80 -0
  72. msprobe/mindspore/cell_processor.py +34 -0
  73. msprobe/mindspore/common/const.py +106 -0
  74. msprobe/mindspore/common/log.py +38 -0
  75. msprobe/mindspore/common/utils.py +81 -0
  76. msprobe/mindspore/compare/distributed_compare.py +75 -0
  77. msprobe/mindspore/compare/ms_compare.py +219 -0
  78. msprobe/mindspore/compare/ms_graph_compare.py +348 -0
  79. msprobe/mindspore/compare/ms_to_pt_api.yaml +399 -0
  80. msprobe/mindspore/debugger/debugger_config.py +66 -51
  81. msprobe/mindspore/debugger/precision_debugger.py +126 -32
  82. msprobe/mindspore/dump/dump_tool_factory.py +35 -38
  83. msprobe/mindspore/dump/hook_cell/api_registry.py +118 -0
  84. msprobe/mindspore/dump/hook_cell/hook_cell.py +55 -0
  85. msprobe/mindspore/dump/hook_cell/support_wrap_ops.yaml +922 -0
  86. msprobe/mindspore/dump/hook_cell/wrap_api.py +113 -0
  87. msprobe/mindspore/dump/jit_dump.py +72 -0
  88. msprobe/mindspore/dump/kernel_graph_dump.py +59 -60
  89. msprobe/mindspore/dump/kernel_kbyk_dump.py +64 -0
  90. msprobe/mindspore/free_benchmark/__init__.py +0 -0
  91. msprobe/mindspore/free_benchmark/api_pynative_self_check.py +116 -0
  92. msprobe/mindspore/free_benchmark/common/__init__.py +0 -0
  93. msprobe/mindspore/free_benchmark/common/config.py +12 -0
  94. msprobe/mindspore/free_benchmark/common/handler_params.py +17 -0
  95. msprobe/mindspore/free_benchmark/common/utils.py +71 -0
  96. msprobe/mindspore/free_benchmark/data/support_wrap_ops.yaml +842 -0
  97. msprobe/mindspore/free_benchmark/decorator/__init__.py +0 -0
  98. msprobe/mindspore/free_benchmark/decorator/dec_forward.py +43 -0
  99. msprobe/mindspore/free_benchmark/decorator/decorator_factory.py +107 -0
  100. msprobe/mindspore/free_benchmark/handler/__init__.py +0 -0
  101. msprobe/mindspore/free_benchmark/handler/base_handler.py +90 -0
  102. msprobe/mindspore/free_benchmark/handler/check_handler.py +41 -0
  103. msprobe/mindspore/free_benchmark/handler/fix_handler.py +36 -0
  104. msprobe/mindspore/free_benchmark/handler/handler_factory.py +21 -0
  105. msprobe/mindspore/free_benchmark/perturbation/add_noise.py +67 -0
  106. msprobe/mindspore/free_benchmark/perturbation/base_perturbation.py +21 -0
  107. msprobe/mindspore/free_benchmark/perturbation/bit_noise.py +63 -0
  108. msprobe/mindspore/free_benchmark/perturbation/exchange_value.py +51 -0
  109. msprobe/mindspore/free_benchmark/perturbation/improve_precision.py +35 -0
  110. msprobe/mindspore/free_benchmark/perturbation/no_change.py +12 -0
  111. msprobe/mindspore/free_benchmark/perturbation/perturbation_factory.py +29 -0
  112. msprobe/mindspore/free_benchmark/self_check_tool_factory.py +33 -0
  113. msprobe/mindspore/grad_probe/__init__.py +0 -0
  114. msprobe/mindspore/grad_probe/global_context.py +90 -0
  115. msprobe/mindspore/grad_probe/grad_analyzer.py +231 -0
  116. msprobe/mindspore/grad_probe/grad_monitor.py +27 -0
  117. msprobe/mindspore/grad_probe/grad_stat_csv.py +132 -0
  118. msprobe/mindspore/grad_probe/hook.py +94 -0
  119. msprobe/mindspore/grad_probe/utils.py +30 -0
  120. msprobe/mindspore/ms_config.py +128 -78
  121. msprobe/mindspore/overflow_check/kernel_graph_overflow_check.py +44 -45
  122. msprobe/mindspore/overflow_check/overflow_check_tool_factory.py +34 -32
  123. msprobe/mindspore/runtime.py +4 -0
  124. msprobe/mindspore/service.py +378 -0
  125. msprobe/mindspore/task_handler_factory.py +24 -21
  126. msprobe/msprobe.py +105 -67
  127. msprobe/pytorch/__init__.py +4 -4
  128. msprobe/pytorch/api_accuracy_checker/common/config.py +53 -50
  129. msprobe/pytorch/api_accuracy_checker/common/utils.py +214 -224
  130. msprobe/pytorch/api_accuracy_checker/compare/algorithm.py +213 -216
  131. msprobe/pytorch/api_accuracy_checker/compare/api_precision_compare.py +606 -545
  132. msprobe/pytorch/api_accuracy_checker/compare/api_precision_standard.yaml +132 -132
  133. msprobe/pytorch/api_accuracy_checker/compare/api_precision_threshold.yaml +390 -390
  134. msprobe/pytorch/api_accuracy_checker/compare/compare.py +386 -345
  135. msprobe/pytorch/api_accuracy_checker/compare/compare_column.py +73 -73
  136. msprobe/pytorch/api_accuracy_checker/compare/compare_utils.py +245 -248
  137. msprobe/pytorch/api_accuracy_checker/config.yaml +10 -4
  138. msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +335 -328
  139. msprobe/pytorch/api_accuracy_checker/run_ut/multi_run_ut.py +200 -203
  140. msprobe/pytorch/api_accuracy_checker/run_ut/run_overflow_check.py +133 -127
  141. msprobe/pytorch/api_accuracy_checker/run_ut/run_ut.py +592 -493
  142. msprobe/pytorch/api_accuracy_checker/run_ut/run_ut_utils.py +70 -7
  143. msprobe/pytorch/api_accuracy_checker/run_ut/torch_ut_setting.json +7 -4
  144. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/__init__.py +0 -0
  145. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/attl.py +197 -0
  146. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/client.py +325 -0
  147. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/device_dispatch.py +204 -0
  148. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/server.py +219 -0
  149. msprobe/pytorch/api_accuracy_checker/tensor_transport_layer/ssl_config.py +10 -0
  150. msprobe/pytorch/bench_functions/__init__.py +15 -0
  151. msprobe/pytorch/bench_functions/apply_adam_w.py +28 -0
  152. msprobe/pytorch/bench_functions/confusion_transpose.py +19 -0
  153. msprobe/pytorch/bench_functions/fast_gelu.py +55 -0
  154. msprobe/pytorch/bench_functions/layer_norm_eval.py +6 -0
  155. msprobe/pytorch/bench_functions/linear.py +12 -0
  156. msprobe/pytorch/bench_functions/matmul_backward.py +48 -0
  157. msprobe/pytorch/bench_functions/npu_fusion_attention.py +509 -0
  158. msprobe/pytorch/bench_functions/rms_norm.py +15 -0
  159. msprobe/pytorch/bench_functions/rotary_mul.py +52 -0
  160. msprobe/pytorch/bench_functions/scaled_mask_softmax.py +26 -0
  161. msprobe/pytorch/bench_functions/swiglu.py +55 -0
  162. msprobe/pytorch/common/__init__.py +2 -2
  163. msprobe/pytorch/common/compare_script.template +14 -14
  164. msprobe/pytorch/common/log.py +20 -31
  165. msprobe/pytorch/common/parse_json.py +39 -37
  166. msprobe/pytorch/common/utils.py +305 -224
  167. msprobe/pytorch/compare/distributed_compare.py +66 -111
  168. msprobe/pytorch/compare/mapping.yaml +607 -607
  169. msprobe/pytorch/compare/match.py +34 -36
  170. msprobe/pytorch/compare/pt_compare.py +50 -0
  171. msprobe/pytorch/debugger/debugger_config.py +95 -86
  172. msprobe/pytorch/debugger/precision_debugger.py +125 -95
  173. msprobe/pytorch/free_benchmark/__init__.py +8 -8
  174. msprobe/pytorch/free_benchmark/common/constant.py +70 -67
  175. msprobe/pytorch/free_benchmark/common/counter.py +71 -71
  176. msprobe/pytorch/free_benchmark/common/enums.py +37 -37
  177. msprobe/pytorch/free_benchmark/common/params.py +129 -129
  178. msprobe/pytorch/free_benchmark/common/utils.py +102 -98
  179. msprobe/pytorch/free_benchmark/compare/grad_saver.py +179 -183
  180. msprobe/pytorch/free_benchmark/compare/single_benchmark.py +104 -104
  181. msprobe/pytorch/free_benchmark/main.py +105 -102
  182. msprobe/pytorch/free_benchmark/perturbed_layers/base_layer.py +13 -13
  183. msprobe/pytorch/free_benchmark/perturbed_layers/layer_factory.py +41 -41
  184. msprobe/pytorch/free_benchmark/perturbed_layers/npu/add_noise.py +90 -90
  185. msprobe/pytorch/free_benchmark/perturbed_layers/npu/bit_noise.py +104 -104
  186. msprobe/pytorch/free_benchmark/perturbed_layers/npu/change_value.py +63 -63
  187. msprobe/pytorch/free_benchmark/perturbed_layers/npu/improve_precision.py +68 -68
  188. msprobe/pytorch/free_benchmark/perturbed_layers/npu/no_change.py +28 -28
  189. msprobe/pytorch/free_benchmark/perturbed_layers/npu/npu_base_layser.py +45 -45
  190. msprobe/pytorch/free_benchmark/perturbed_layers/run_cpu.py +19 -19
  191. msprobe/pytorch/free_benchmark/result_handlers/base_handler.py +217 -203
  192. msprobe/pytorch/free_benchmark/result_handlers/check_handler.py +39 -39
  193. msprobe/pytorch/free_benchmark/result_handlers/fix_handler.py +23 -23
  194. msprobe/pytorch/free_benchmark/result_handlers/handler_factory.py +30 -31
  195. msprobe/pytorch/free_benchmark/result_handlers/preheat_handler.py +170 -170
  196. msprobe/pytorch/function_factory.py +76 -0
  197. msprobe/pytorch/functional/dump_module.py +39 -39
  198. msprobe/pytorch/grad_probe/__init__.py +0 -0
  199. msprobe/pytorch/grad_probe/grad_monitor.py +91 -0
  200. msprobe/pytorch/grad_probe/grad_stat_csv.py +129 -0
  201. msprobe/pytorch/hook_module/api_registry.py +161 -161
  202. msprobe/pytorch/hook_module/hook_module.py +120 -109
  203. msprobe/pytorch/hook_module/support_wrap_ops.yaml +1879 -1876
  204. msprobe/pytorch/hook_module/utils.py +30 -29
  205. msprobe/pytorch/hook_module/wrap_aten.py +110 -100
  206. msprobe/pytorch/hook_module/wrap_distributed.py +78 -75
  207. msprobe/pytorch/hook_module/wrap_functional.py +105 -108
  208. msprobe/pytorch/hook_module/wrap_npu_custom.py +93 -73
  209. msprobe/pytorch/hook_module/wrap_tensor.py +71 -72
  210. msprobe/pytorch/hook_module/wrap_torch.py +86 -88
  211. msprobe/pytorch/hook_module/wrap_vf.py +62 -64
  212. msprobe/pytorch/module_processer.py +138 -98
  213. msprobe/pytorch/online_dispatch/__init__.py +20 -20
  214. msprobe/pytorch/online_dispatch/compare.py +236 -236
  215. msprobe/pytorch/online_dispatch/dispatch.py +271 -273
  216. msprobe/pytorch/online_dispatch/dump_compare.py +155 -186
  217. msprobe/pytorch/online_dispatch/single_compare.py +391 -391
  218. msprobe/pytorch/online_dispatch/torch_ops_config.yaml +49 -49
  219. msprobe/pytorch/online_dispatch/utils.py +130 -187
  220. msprobe/pytorch/parse.py +4 -4
  221. msprobe/pytorch/parse_tool/cli.py +32 -32
  222. msprobe/pytorch/parse_tool/lib/compare.py +260 -259
  223. msprobe/pytorch/parse_tool/lib/config.py +52 -51
  224. msprobe/pytorch/parse_tool/lib/file_desc.py +31 -31
  225. msprobe/pytorch/parse_tool/lib/interactive_cli.py +102 -102
  226. msprobe/pytorch/parse_tool/lib/parse_exception.py +54 -54
  227. msprobe/pytorch/parse_tool/lib/parse_tool.py +158 -158
  228. msprobe/pytorch/parse_tool/lib/utils.py +316 -367
  229. msprobe/pytorch/parse_tool/lib/visualization.py +85 -90
  230. msprobe/pytorch/pt_config.py +188 -93
  231. msprobe/pytorch/service.py +246 -167
  232. mindstudio_probe-1.0.1.dist-info/RECORD +0 -228
  233. msprobe/config/README.md +0 -397
  234. msprobe/mindspore/doc/dump.md +0 -65
  235. msprobe/mindspore/dump/api_kbk_dump.py +0 -55
  236. msprobe/pytorch/compare/acc_compare.py +0 -1024
  237. msprobe/pytorch/compare/highlight.py +0 -100
  238. msprobe/pytorch/doc/FAQ.md +0 -193
  239. msprobe/pytorch/doc/api_accuracy_checker.md +0 -269
  240. msprobe/pytorch/doc/atat/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/205/342/225/226/320/265/321/205/320/225/342/225/226/321/206/320/245/342/226/221/321/206/320/235/320/276dump/321/206/320/260/320/227/321/205/320/227/320/226/321/206/320/220/320/267/321/210/320/223/342/225/234/321/205/320/257/342/225/221/321/207/342/225/221/342/224/220/321/206/320/232/320/265/321/205/320/241/320/232.md +0 -182
  241. msprobe/pytorch/doc/dump.md +0 -207
  242. msprobe/pytorch/doc/ptdbg_ascend_compare.md +0 -176
  243. msprobe/pytorch/doc/ptdbg_ascend_overview.md +0 -68
  244. msprobe/pytorch/doc/ptdbg_ascend_quickstart.md +0 -381
  245. msprobe/pytorch/doc/run_overflow_check.md +0 -25
  246. msprobe/pytorch/doc//321/205/320/254/320/270/321/207/342/225/221/342/224/220/321/207/342/226/223/342/225/233/321/205/342/225/221/320/266/321/206/320/277/320/244/321/205/320/277/342/225/243.md +0 -90
  247. msprobe/test/core_ut/common/test_utils.py +0 -345
  248. msprobe/test/core_ut/data_dump/test_data_collector.py +0 -47
  249. msprobe/test/core_ut/data_dump/test_json_writer.py +0 -183
  250. msprobe/test/core_ut/data_dump/test_scope.py +0 -151
  251. msprobe/test/core_ut/test_common_config.py +0 -152
  252. msprobe/test/core_ut/test_file_check.py +0 -218
  253. msprobe/test/core_ut/test_log.py +0 -109
  254. msprobe/test/mindspore_ut/test_api_kbk_dump.py +0 -51
  255. msprobe/test/mindspore_ut/test_debugger_config.py +0 -42
  256. msprobe/test/mindspore_ut/test_dump_tool_factory.py +0 -51
  257. msprobe/test/mindspore_ut/test_kernel_graph_dump.py +0 -66
  258. msprobe/test/mindspore_ut/test_kernel_graph_overflow_check.py +0 -63
  259. msprobe/test/mindspore_ut/test_ms_config.py +0 -69
  260. msprobe/test/mindspore_ut/test_overflow_check_tool_factory.py +0 -51
  261. msprobe/test/mindspore_ut/test_precision_debugger.py +0 -56
  262. msprobe/test/mindspore_ut/test_task_handler_factory.py +0 -58
  263. msprobe/test/pytorch_ut/advisor/test_advisor.py +0 -83
  264. msprobe/test/pytorch_ut/api_accuracy_checker/common/test_common_utils.py +0 -108
  265. msprobe/test/pytorch_ut/api_accuracy_checker/common/test_config.py +0 -39
  266. msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_algorithm.py +0 -112
  267. msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_api_precision_compare.py +0 -77
  268. msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_compare.py +0 -125
  269. msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_compare_column.py +0 -10
  270. msprobe/test/pytorch_ut/api_accuracy_checker/compare/test_compare_utils.py +0 -43
  271. msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/dump.json +0 -179
  272. msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/forward.json +0 -63
  273. msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/test_data_generate.py +0 -99
  274. msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/test_multi_run_ut.py +0 -115
  275. msprobe/test/pytorch_ut/api_accuracy_checker/run_ut/test_run_ut.py +0 -72
  276. msprobe/test/pytorch_ut/compare/test_acc_compare.py +0 -17
  277. msprobe/test/pytorch_ut/free_benchmark/perturbed_layers/test_perturbed_layser.py +0 -105
  278. msprobe/test/pytorch_ut/free_benchmark/result_handlers/test_result_handler.py +0 -121
  279. msprobe/test/pytorch_ut/free_benchmark/test_main.py +0 -101
  280. msprobe/test/pytorch_ut/functional/test_dump_module.py +0 -15
  281. msprobe/test/pytorch_ut/hook_module/test_api_registry.py +0 -130
  282. msprobe/test/pytorch_ut/hook_module/test_hook_module.py +0 -42
  283. msprobe/test/pytorch_ut/hook_module/test_wrap_aten.py +0 -65
  284. msprobe/test/pytorch_ut/hook_module/test_wrap_distributed.py +0 -35
  285. msprobe/test/pytorch_ut/hook_module/test_wrap_functional.py +0 -20
  286. msprobe/test/pytorch_ut/hook_module/test_wrap_tensor.py +0 -35
  287. msprobe/test/pytorch_ut/hook_module/test_wrap_torch.py +0 -43
  288. msprobe/test/pytorch_ut/hook_module/test_wrap_vf.py +0 -11
  289. msprobe/test/pytorch_ut/test_pt_config.py +0 -69
  290. msprobe/test/pytorch_ut/test_service.py +0 -59
  291. msprobe/test/resources/advisor.txt +0 -3
  292. msprobe/test/resources/compare_result_20230703104808.csv +0 -9
  293. msprobe/test/resources/compare_result_without_accuracy.csv +0 -9
  294. msprobe/test/resources/config.yaml +0 -3
  295. msprobe/test/resources/npu_test.pkl +0 -8
  296. msprobe/test/run_test.sh +0 -30
  297. msprobe/test/run_ut.py +0 -58
  298. msprobe/test/test_module_processer.py +0 -64
  299. {mindstudio_probe-1.0.1.dist-info → mindstudio_probe-1.0.4.dist-info}/top_level.txt +0 -0
  300. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_1.png +0 -0
  301. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_2.png +0 -0
  302. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_3.png +0 -0
  303. /msprobe/{pytorch/doc → docs}/img/BLOOM-7B_4.png +0 -0
  304. /msprobe/{pytorch/doc → docs}/img/GPT-3_1.png +0 -0
  305. /msprobe/{pytorch/doc → docs}/img/GPT-3_2.png +0 -0
  306. /msprobe/{pytorch/doc → docs}/img/GPT-3_3.png +0 -0
  307. /msprobe/{pytorch/doc → docs}/img/GPT-3_4.png +0 -0
  308. /msprobe/{pytorch/doc → docs}/img/GPT-3_5.png +0 -0
  309. /msprobe/{pytorch/doc → docs}/img/GPT-3_6.png +0 -0
  310. /msprobe/{pytorch/doc → docs}/img/GPT-3_7.png +0 -0
  311. /msprobe/{pytorch/doc → docs}/img/GPT-3_8.png +0 -0
  312. /msprobe/{pytorch/doc → docs}/img/YOLOV5S_1.png +0 -0
  313. /msprobe/{pytorch/doc → docs}/img/YOLOV5S_2.png +0 -0
  314. /msprobe/{pytorch/doc → docs}/img/accuracy_checking_details.png +0 -0
  315. /msprobe/{pytorch/doc → docs}/img/accuracy_checking_result.png +0 -0
  316. /msprobe/{pytorch/doc → docs}/img/api_precision_compare_details.png +0 -0
  317. /msprobe/{pytorch/doc → docs}/img/api_precision_compare_result.png +0 -0
  318. /msprobe/{pytorch/doc → docs}/img/auto_analyze_log.png +0 -0
  319. /msprobe/{pytorch/doc → docs}/img/compare_result_pkl.png +0 -0
  320. /msprobe/{pytorch/doc → docs}/img/compare_result_pkl_md5.png.png +0 -0
  321. /msprobe/{pytorch/doc → docs}/img/cpu_info.png +0 -0
  322. /msprobe/{config → docs}/img/free_benchmark.png +0 -0
  323. /msprobe/{pytorch/doc → docs}/img/module_compare.png +0 -0
msprobe/pytorch/compare/mapping.yaml CHANGED
@@ -1,607 +1,607 @@
1
- __and__: __and__
2
- __iand__: __iand__
3
- __ilshift__: __ilshift__
4
- __ior__: __ior__
5
- __irshift__: __irshift__
6
- __ixor__: __ixor__
7
- __lshift__: __lshift__
8
- __or__: __or__
9
- __rshift__: __rshift__
10
- __xor__: __xor__
11
- _adaptive_avg_pool2d: adaptive_avg_pool2d
12
- _adaptive_avg_pool3d: adaptive_avg_pool3d
13
- _cdist_forward: cdist
14
- _cudnn_rnn: rnn
15
- _embedding_bag: embedding_bag
16
- _fft_c2c: fft
17
- _fft_c2r: rfft
18
- _foreach_add_: _foreach_add_
19
- _foreach_addcdiv: _foreach_addcdiv
20
- _foreach_copy_: _foreach_copy_
21
- _foreach_lerp_: _foreach_lerp_
22
- _foreach_maximum: _foreach_maximum
23
- _foreach_mul: _foreach_mul
24
- _foreach_neg_: _foreach_neg_
25
- _foreach_pow: _foreach_pow
26
- _foreach_reciprocal_: _foreach_reciprocal_
27
- _foreach_sign: _foreach_sign
28
- _foreach_sqrt: _foreach_sqrt
29
- _foreach_sqrt_: _foreach_sqrt_
30
- _foreach_sub: _foreach_sub
31
- _fused_adam: FusedAdam
32
- _linalg_det: det
33
- _linalg_eigh: eigh
34
- _linalg_slogdet: slogdet
35
- _linalg_svd: svd
36
- _list_to_tensor: as_tensor
37
- _log_softmax: log_softmax
38
- _native_batch_norm_legit: batch_norm
39
- _nested_tensor_from_tensor_list: _nested_tensor_from_tensor_list
40
- _pdist_forward: pdist
41
- _pin_memory: pin_memory
42
- _reshape_alias: reshape
43
- _resize_output_: resize_
44
- _softmax: softmax
45
- _to_copy: to
46
- abs: abs
47
- abs_: abs_
48
- absolute: abs
49
- absolute_: abs_
50
- acos: acos
51
- acos_: acos_
52
- acosh: acosh
53
- acosh_: acosh_
54
- adaptive_max_pool2d: adaptive_max_pool2d
55
- adaptive_max_pool3d: adaptive_max_pool3d
56
- add: add
57
- add_: add_
58
- addbmm: addbmm
59
- addbmm_: addbmm_
60
- addcdiv: addcdiv
61
- addcdiv_: addcdiv_
62
- addcmul: addcmul
63
- addcmul_: addcmul_
64
- addmm: addmm
65
- addmm_: addmm_
66
- addmv: addmv
67
- addmv_: addmv_
68
- addr: addr
69
- affine_grid_generator: affine_grid
70
- alias: alias
71
- all: all
72
- alpha_dropout: AlphaDropout
73
- amax: amax
74
- amin: amin
75
- aminmax: aminmax
76
- angle: angle
77
- any: any
78
- arange: arange
79
- arccos: acos
80
- arccos_: arccos_
81
- arccosh: arccosh
82
- arccosh_: arccosh_
83
- arcsin: asin
84
- arcsin_: arcsin_
85
- arcsinh: asinh
86
- arcsinh_: arcsinh_
87
- arctan: atan
88
- arctan2: atan2
89
- arctan2_: arctan2_
90
- arctan_: arctan_
91
- arctanh: arctanh
92
- arctanh_: arctanh_
93
- argmax: argmax
94
- argmin: argmin
95
- argsort: argsort
96
- as_strided: as_strided
97
- asin: asin
98
- asin_: asin_
99
- asinh: asinh
100
- asinh_: asinh_
101
- atan: atan
102
- atan2: atan2
103
- atan2_: atan2_
104
- atan_: atan_
105
- atanh: atanh
106
- atanh_: atanh_
107
- avg_pool2d: avg_pool2d
108
- avg_pool3d: avg_pool3d
109
- baddbmm: baddbmm
110
- baddbmm_: baddbmm_
111
- bernoulli: bernoulli
112
- bernoulli_: bernoulli_
113
- binary_cross_entropy: BCELoss
114
- binary_cross_entropy_with_logits: binary_cross_entropy_with_logits
115
- bitwise_and: bitwise_and
116
- bitwise_and_: bitwise_and_
117
- bitwise_left_shift: __lshift__
118
- bitwise_left_shift_: bitwise_left_shift_
119
- bitwise_not: bitwise_not
120
- bitwise_not_: bitwise_not_
121
- bitwise_or: bitwise_or
122
- bitwise_or_: bitwise_or_
123
- bitwise_right_shift: __rshift__
124
- bitwise_right_shift_: bitwise_right_shift_
125
- bitwise_xor: bitwise_xor
126
- bitwise_xor_: bitwise_xor_
127
- bmm: bmm
128
- broadcast_tensors: broadcast_tensors
129
- bucketize: bucketize
130
- cat: cat
131
- cauchy: Cauchy
132
- cauchy_: cauchy_
133
- ceil: ceil
134
- ceil_: ceil_
135
- celu: celu
136
- celu_: celu_
137
- cholesky: cholesky
138
- cholesky_inverse: cholesky_inverse
139
- cholesky_solve: cholesky_solve
140
- clamp: clamp
141
- clamp_: clamp_
142
- clamp_max: clamp_max
143
- clamp_max_: clamp_max_
144
- clamp_min: clamp_min
145
- clamp_min_: clamp_min_
146
- clip: clip
147
- clip_: clip_
148
- clone: clone
149
- col2im: col2im
150
- complex: complex
151
- conj_physical: conj
152
- conj_physical_: conj_
153
- constant_pad_nd: pad
154
- convolution: Conv2d
155
- copy: copy_
156
- copy_: copy_
157
- copysign: copysign
158
- copysign_: copysign_
159
- cos: cos
160
- cos_: cos_
161
- cosh: cosh
162
- cosh_: cosh_
163
- count_nonzero: count_nonzero
164
- cudnn_batch_norm: BatchNorm2d
165
- cummax: cummax
166
- cummin: cummin
167
- cumprod: cumprod
168
- cumprod_: cumprod_
169
- cumsum: cumsum
170
- cumsum_: cumsum_
171
- deg2rad: deg2rad
172
- deg2rad_: deg2rad_
173
- detach: detach
174
- diag: diag
175
- diag_embed: diag_embed
176
- diagonal: diagonal
177
- diagonal_copy: diagonal
178
- diagonal_scatter: diagonal
179
- digamma: digamma
180
- digamma_: digamma_
181
- dist: dist
182
- div: div
183
- div_: div_
184
- divide: div
185
- divide_: divide_
186
- dot: dot
187
- dropout: dropout
188
- elu: ELU
189
- elu_: elu_
190
- embedding: embedding
191
- empty_like: empty_like
192
- empty_strided: empty_strided
193
- eq: eq
194
- eq_: eq_
195
- erf: erf
196
- erf_: erf_
197
- erfc: erfc
198
- erfc_: erfc_
199
- erfinv: erfinv
200
- erfinv_: erfinv_
201
- exp: exp
202
- exp2: exp2
203
- exp2_: exp2_
204
- exp_: exp_
205
- expand: expand
206
- expm1: expm1
207
- expm1_: expm1_
208
- exponential: Exponential
209
- exponential_: exponential_
210
- eye: eye
211
- fft_fft: fft
212
- fft_fft2: fft2
213
- fft_fftn: fftn
214
- fft_fftshift: fftshift
215
- fft_hfft: hfft
216
- fft_hfft2: hfft2
217
- fft_hfftn: hfftn
218
- fft_ifft: ifft
219
- fft_ifft2: ifft2
220
- fft_ifftn: ifftn
221
- fft_ifftshift: ifftshift
222
- fft_ihfft: ihfft
223
- fft_ihfft2: ihfft2
224
- fft_ihfftn: ifftn
225
- fft_irfft: irfft
226
- fft_irfft2: irfft2
227
- fft_irfftn: irfftn
228
- fft_rfft: rfft
229
- fft_rfft2: rfft2
230
- fft_rfftn: rfftn
231
- fill: fill_
232
- fill_: fill_
233
- fix: fix
234
- fix_: fix_
235
- flip: flip
236
- float_power_: float_power_
237
- floor: floor
238
- floor_: floor_
239
- floor_divide: floor_divide
240
- floor_divide_: floor_divide_
241
- fmax: fmax
242
- fmin: fmin
243
- fmod: fmod
244
- fmod_: fmod_
245
- frac: frac
246
- frac_: frac_
247
- full: full
248
- full_like: full_like
249
- gather: gather
250
- gcd: gcd
251
- gcd_: gcd_
252
- ge: ge
253
- ge_: ge_
254
- gelu: GELU
255
- gelu_: gelu_
256
- geometric: Geometric
257
- geometric_: geometric_
258
- glu: glu
259
- greater: gt
260
- greater_: ge_
261
- greater_equal: ge
262
- greater_equal_: ge_
263
- grid_sampler_2d: grid_sample
264
- grid_sampler_3d: grid_sample
265
- gru: GRU
266
- gt: gt
267
- gt_: gt_
268
- hardshrink: Hardshrink
269
- hardsigmoid: hardsigmoid
270
- hardsigmoid_: hardsigmoid_
271
- hardswish: hardswish
272
- hardswish_: hardswish_
273
- hardtanh: hardtanh
274
- hardtanh_: hardtanh_
275
- heaviside: heaviside
276
- heaviside_: heaviside_
277
- hinge_embedding_loss: HingeEmbeddingLoss
278
- huber_loss: huber_loss
279
- hypot: hypot
280
- hypot_: hypot_
281
- i0: i0
282
- i0_: i0_
283
- igamma: igamma
284
- igamma_: igamma_
285
- igammac: igammac
286
- igammac_: igammac_
287
- index: __getitem__
288
- index_add: index_add
289
- index_add_: index_add_
290
- index_copy: index_copy_
291
- index_copy_: index_copy_
292
- index_fill: index_fill_
293
- index_fill_: index_fill_
294
- index_put: index_put_
295
- index_put_: index_put_
296
- index_reduce: index_select
297
- index_select: index_select
298
- is_pinned: is_pinned
299
- is_same_size: is_same_size
300
- isinf: isinf
301
- isnan: isnan
302
- isneginf: isneginf
303
- isposinf: isposinf
304
- istft: istft
305
- item: item
306
- lcm: lcm
307
- lcm_: lcm_
308
- le: le
309
- le_: le_
310
- leaky_relu: LeakyReLU
311
- leaky_relu_: leaky_relu_
312
- lerp: lerp
313
- lerp_: lerp_
314
- less: less
315
- less_: less_
316
- less_equal: le
317
- less_equal_: less_equal_
318
- lgamma: lgamma
319
- lgamma_: lgamma_
320
- linalg_cholesky_ex: cholesky
321
- linalg_cross: cross
322
- linalg_householder_product: householder_product
323
- linalg_inv_ex: inv
324
- linalg_ldl_factor_ex: ldl
325
- linalg_ldl_solve: ldl_solve
326
- linalg_lu: lu
327
- linalg_lu_factor_ex: lu_factor
328
- linalg_lu_solve: lu_solve
329
- linalg_matrix_exp: matrix_exp
330
- linalg_qr: qr
331
- linalg_solve_triangular: solve
332
- linalg_vector_norm: norm
333
- linspace: linspace
334
- log: log
335
- log10: log10
336
- log10_: log10_
337
- log1p: log1p
338
- log1p_: log1p_
339
- log2: log2
340
- log2_: log2_
341
- log_: log_
342
- log_normal: LogNormal
343
- log_sigmoid_forward: log_sigmoid
344
- logaddexp: logaddexp
345
- logaddexp2: logaddexp2
346
- _native_batch_norm_legit_functional: batch_norm
347
- logcumsumexp: logcumsumexp
348
- logical_and: logical_and
349
- logical_and_: logical_and_
350
- logical_not: logical_not
351
- logical_not_: logical_not_
352
- logical_or: logical_or
353
- logical_or_: logical_or_
354
- logical_xor: logical_xor
355
- logical_xor_: logical_xor_
356
- logit: logit
357
- logit_: logit_
358
- logspace: logspace
359
- logsumexp: logsumexp
360
- lstm: LSTM
361
- lt: lt
362
- lt_: lt_
363
- lu_unpack: lu_unpack
364
- margin_ranking_loss: margin_ranking_loss
365
- masked_fill: masked_fill
366
- masked_fill_: masked_fill_
367
- matmul: matmul
368
- max: max
369
- max_pool2d_with_indices: MaxPool2d
370
- max_pool3d_with_indices: MaxPool3d
371
- max_unpool2d: MaxUnpool2d
372
- max_unpool3d: max_unpool3d
373
- maximum: maximum
374
- mean: mean
375
- median: median
376
- meshgrid: meshgrid
377
- min: min
378
- minimum: minimum
379
- mish: Mish
380
- mish_: mish_
381
- mm: mm
382
- mode: mode
383
- mse_loss: mse_loss
384
- mul: mul
385
- mul_: mul_
386
- multi_margin_loss: MultiMarginLoss
387
- multilabel_margin_loss_forward: multilabel_margin_loss
388
- multinomial: multinomial
389
- multiply: multiply
390
- multiply_: mul_
391
- mv: mv
392
- mvlgamma: mvlgamma
393
- mvlgamma_: mvlgamma_
394
- name: name
395
- nan_to_num: nan_to_num
396
- nan_to_num_: nan_to_num_
397
- nanmedian: nanmedian
398
- nansum: nansum
399
- narrow_copy: narrow
400
- native_batch_norm: BatchNorm2d
401
- native_dropout: dropout
402
- native_group_norm: group_norm
403
- native_layer_norm: LayerNorm
404
- ne: ne
405
- ne_: ne_
406
- neg: neg
407
- neg_: neg_
408
- negative: neg
409
- negative_: neg_
410
- new_empty: new_empty
411
- new_empty_strided: new_empty_strided
412
- new_full: new_full
413
- new_ones: new_ones
414
- new_zeros: new_zeros
415
- nextafter: nextafter
416
- nextafter_: nextafter_
417
- nll_loss: nll_loss
418
- nll_loss2d_forward: NLLLoss2d
419
- nll_loss_forward: NLLLoss
420
- nonzero_static: nonzero
421
- norm: norm
422
- normal: normal
423
- normal_: normal_
424
- not_equal: ne
425
- not_equal_: ne_
426
- ones: ones
427
- ones_like: ones_like
428
- ormqr: ormqr
429
- pairwise_distance: pairwise_distance
430
- pdist: pdist
431
- permute: permute
432
- pin_memory: pin_memory
433
- pixel_shuffle: PixelShuffle
434
- polar: polar
435
- polygamma: polygamma
436
- positive: positive
437
- pow: pow
438
- pow_: pow_
439
- prelu: prelu
440
- prod: prod
441
- quantized_gru: GRU
442
- quantized_lstm: LSTM
443
- rad2deg: rad2deg
444
- rad2deg_: rad2deg_
445
- rand: rand
446
- rand_like: rand_like
447
- randint: randint
448
- randint_like: randint_like
449
- randn: randn
450
- randn_like: randn_like
451
- randperm: randperm
452
- reciprocal: reciprocal
453
- reciprocal_: reciprocal_
454
- reflection_pad1d: reflection_pad1d
455
- reflection_pad2d: reflection_pad2d
456
- reflection_pad3d: ReflectionPad3d
457
- relu: relu
458
- relu6: relu6
459
- relu_: relu_
460
- remainder: remainder
461
- remainder_: remainder_
462
- renorm: renorm
463
- renorm_: renorm_
464
- repeat: repeat
465
- repeat_interleave: repeat_interleave
466
- replication_pad1d: ReplicationPad1d
467
- replication_pad2d: replication_pad2d
468
- replication_pad3d: replication_pad3d
469
- resize_as_: resize_as_
470
- rnn_relu: RNN
471
- rnn_tanh: RNN
472
- roll: roll
473
- rot90: rot90
474
- round: round
475
- round_: round_
476
- rrelu_with_noise: RReLU
477
- rrelu_with_noise_: rrelu_with_noise
478
- rsqrt: rsqrt
479
- rsqrt_: rsqrt_
480
- rsub: rsub
481
- scalar_tensor: scalar_tensor
482
- scatter: scatter_
483
- scatter_: scatter_
484
- scatter_add: scatter_add
485
- scatter_add_: scatter_add_
486
- searchsorted: searchsorted
487
- select: select
488
- selu: selu
489
- selu_: selu_
490
- sgn: sgn
491
- sgn_: sgn_
492
- sigmoid: sigmoid
493
- sigmoid_: sigmoid_
494
- sign: sign
495
- sign_: sign_
496
- signbit: signbit
497
- silu: silu
498
- silu_: silu_
499
- sin: sin
500
- sin_: sin_
501
- sinc: sinc
502
- sinc_: sinc_
503
- sinh: sinh
504
- sinh_: sinh_
505
- slice: slice
506
- smooth_l1_loss: smooth_l1_loss
507
- soft_margin_loss: soft_margin_loss
508
- softplus: softplus
509
- softshrink: softshrink
510
- sort: sort
511
- special_airy_ai: airy_ai
512
- special_bessel_j0: j0
513
- special_bessel_j1: j1
514
- special_bessel_y0: y0
515
- special_bessel_y1: y1
516
- special_chebyshev_polynomial_t: chebyshev_t
517
- special_chebyshev_polynomial_u: chebyshev_u
518
- special_entr: entr
519
- special_erfcx: erfcx
520
- special_hermite_polynomial_h: hermite
521
- special_hermite_polynomial_he: he
522
- special_i0: i0
523
- special_i0e: i0e
524
- special_i1: i1
525
- special_i1e: i1e
526
- special_laguerre_polynomial_l: laguerre_l
527
- special_log_ndtr: log_ndtr
528
- special_modified_bessel_i0: i0
529
- special_modified_bessel_i1: i1
530
- special_modified_bessel_k0: k0
531
- special_modified_bessel_k1: i1
532
- special_ndtr: ndtr
533
- special_ndtri: ndtri
534
- special_scaled_modified_bessel_k0: i0e
535
- special_scaled_modified_bessel_k1: scaled_modified_bessel_k1
536
- special_spherical_bessel_j0: spherical_jn
537
- special_xlog1py: xlog1py
538
- special_zeta: zeta
539
- split: split
540
- split_with_sizes: split
541
- sqrt: sqrt
542
- sqrt_: sqrt_
543
- square: square
544
- square_: square_
545
- squeeze: squeeze
546
- stack: stack
547
- std: std
548
- std_mean: std_mean
549
- stft: stft
550
- sub: sub
551
- sub_: sub_
552
- subtract: sub
553
- subtract_: subtract_
554
- sum: sum
555
- t: t
556
- t_: t_
557
- take: take
558
- tan: tan
559
- tan_: tan_
560
- tanh: tanh
561
- tanh_: tanh_
562
- threshold: threshold
563
- threshold_: threshold_
564
- to: to
565
- topk: topk
566
- trace: trace
567
- transpose: transpose
568
- transpose_: transpose_
569
- triangular_solve: triangular_solve
570
- tril: tril
571
- tril_: tril_
572
- tril_indices: tril_indices
573
- triu: triu
574
- triu_: triu_
575
- triu_indices: triu_indices
576
- true_divide: true_divide
577
- true_divide_: true_divide_
578
- trunc: trunc
579
- trunc_: trunc_
580
- unbind: unbind
581
- unfold: unfold
582
- uniform: Uniform
583
- uniform_: uniform_
584
- unsafe_chunk: unsafe_chunk
585
- unsafe_split: split
586
- unsafe_split_with_sizes: split_with_sizes
587
- unsqueeze: unsqueeze
588
- unsqueeze_: unsqueeze_
589
- upsample_bicubic2d: interpolate
590
- upsample_bilinear2d: upsample_bilinear
591
- upsample_nearest1d: interpolate
592
- upsample_nearest2d: interpolate
593
- upsample_nearest3d: interpolate
594
- var: var
595
- var_mean: var_mean
596
- vdot: vdot
597
- view: view
598
- where: where
599
- xlogy: xlogy
600
- xlogy_: xlogy_
601
- zero: zeros
602
- zero_: zero_
603
- zeros: zeros
604
- zeros_like: zeros_like
605
-
606
-
607
-
1
+ __and__: __and__
2
+ __iand__: __iand__
3
+ __ilshift__: __ilshift__
4
+ __ior__: __ior__
5
+ __irshift__: __irshift__
6
+ __ixor__: __ixor__
7
+ __lshift__: __lshift__
8
+ __or__: __or__
9
+ __rshift__: __rshift__
10
+ __xor__: __xor__
11
+ _adaptive_avg_pool2d: adaptive_avg_pool2d
12
+ _adaptive_avg_pool3d: adaptive_avg_pool3d
13
+ _cdist_forward: cdist
14
+ _cudnn_rnn: rnn
15
+ _embedding_bag: embedding_bag
16
+ _fft_c2c: fft
17
+ _fft_c2r: rfft
18
+ _foreach_add_: _foreach_add_
19
+ _foreach_addcdiv: _foreach_addcdiv
20
+ _foreach_copy_: _foreach_copy_
21
+ _foreach_lerp_: _foreach_lerp_
22
+ _foreach_maximum: _foreach_maximum
23
+ _foreach_mul: _foreach_mul
24
+ _foreach_neg_: _foreach_neg_
25
+ _foreach_pow: _foreach_pow
26
+ _foreach_reciprocal_: _foreach_reciprocal_
27
+ _foreach_sign: _foreach_sign
28
+ _foreach_sqrt: _foreach_sqrt
29
+ _foreach_sqrt_: _foreach_sqrt_
30
+ _foreach_sub: _foreach_sub
31
+ _fused_adam: FusedAdam
32
+ _linalg_det: det
33
+ _linalg_eigh: eigh
34
+ _linalg_slogdet: slogdet
35
+ _linalg_svd: svd
36
+ _list_to_tensor: as_tensor
37
+ _log_softmax: log_softmax
38
+ _native_batch_norm_legit: batch_norm
39
+ _nested_tensor_from_tensor_list: _nested_tensor_from_tensor_list
40
+ _pdist_forward: pdist
41
+ _pin_memory: pin_memory
42
+ _reshape_alias: reshape
43
+ _resize_output_: resize_
44
+ _softmax: softmax
45
+ _to_copy: to
46
+ abs: abs
47
+ abs_: abs_
48
+ absolute: abs
49
+ absolute_: abs_
50
+ acos: acos
51
+ acos_: acos_
52
+ acosh: acosh
53
+ acosh_: acosh_
54
+ adaptive_max_pool2d: adaptive_max_pool2d
55
+ adaptive_max_pool3d: adaptive_max_pool3d
56
+ add: add
57
+ add_: add_
58
+ addbmm: addbmm
59
+ addbmm_: addbmm_
60
+ addcdiv: addcdiv
61
+ addcdiv_: addcdiv_
62
+ addcmul: addcmul
63
+ addcmul_: addcmul_
64
+ addmm: addmm
65
+ addmm_: addmm_
66
+ addmv: addmv
67
+ addmv_: addmv_
68
+ addr: addr
69
+ affine_grid_generator: affine_grid
70
+ alias: alias
71
+ all: all
72
+ alpha_dropout: AlphaDropout
73
+ amax: amax
74
+ amin: amin
75
+ aminmax: aminmax
76
+ angle: angle
77
+ any: any
78
+ arange: arange
79
+ arccos: acos
80
+ arccos_: arccos_
81
+ arccosh: arccosh
82
+ arccosh_: arccosh_
83
+ arcsin: asin
84
+ arcsin_: arcsin_
85
+ arcsinh: asinh
86
+ arcsinh_: arcsinh_
87
+ arctan: atan
88
+ arctan2: atan2
89
+ arctan2_: arctan2_
90
+ arctan_: arctan_
91
+ arctanh: arctanh
92
+ arctanh_: arctanh_
93
+ argmax: argmax
94
+ argmin: argmin
95
+ argsort: argsort
96
+ as_strided: as_strided
97
+ asin: asin
98
+ asin_: asin_
99
+ asinh: asinh
100
+ asinh_: asinh_
101
+ atan: atan
102
+ atan2: atan2
103
+ atan2_: atan2_
104
+ atan_: atan_
105
+ atanh: atanh
106
+ atanh_: atanh_
107
+ avg_pool2d: avg_pool2d
108
+ avg_pool3d: avg_pool3d
109
+ baddbmm: baddbmm
110
+ baddbmm_: baddbmm_
111
+ bernoulli: bernoulli
112
+ bernoulli_: bernoulli_
113
+ binary_cross_entropy: BCELoss
114
+ binary_cross_entropy_with_logits: binary_cross_entropy_with_logits
115
+ bitwise_and: bitwise_and
116
+ bitwise_and_: bitwise_and_
117
+ bitwise_left_shift: __lshift__
118
+ bitwise_left_shift_: bitwise_left_shift_
119
+ bitwise_not: bitwise_not
120
+ bitwise_not_: bitwise_not_
121
+ bitwise_or: bitwise_or
122
+ bitwise_or_: bitwise_or_
123
+ bitwise_right_shift: __rshift__
124
+ bitwise_right_shift_: bitwise_right_shift_
125
+ bitwise_xor: bitwise_xor
126
+ bitwise_xor_: bitwise_xor_
127
+ bmm: bmm
128
+ broadcast_tensors: broadcast_tensors
129
+ bucketize: bucketize
130
+ cat: cat
131
+ cauchy: Cauchy
132
+ cauchy_: cauchy_
133
+ ceil: ceil
134
+ ceil_: ceil_
135
+ celu: celu
136
+ celu_: celu_
137
+ cholesky: cholesky
138
+ cholesky_inverse: cholesky_inverse
139
+ cholesky_solve: cholesky_solve
140
+ clamp: clamp
141
+ clamp_: clamp_
142
+ clamp_max: clamp_max
143
+ clamp_max_: clamp_max_
144
+ clamp_min: clamp_min
145
+ clamp_min_: clamp_min_
146
+ clip: clip
147
+ clip_: clip_
148
+ clone: clone
149
+ col2im: col2im
150
+ complex: complex
151
+ conj_physical: conj
152
+ conj_physical_: conj_
153
+ constant_pad_nd: pad
154
+ convolution: Conv2d
155
+ copy: copy_
156
+ copy_: copy_
157
+ copysign: copysign
158
+ copysign_: copysign_
159
+ cos: cos
160
+ cos_: cos_
161
+ cosh: cosh
162
+ cosh_: cosh_
163
+ count_nonzero: count_nonzero
164
+ cudnn_batch_norm: BatchNorm2d
165
+ cummax: cummax
166
+ cummin: cummin
167
+ cumprod: cumprod
168
+ cumprod_: cumprod_
169
+ cumsum: cumsum
170
+ cumsum_: cumsum_
171
+ deg2rad: deg2rad
172
+ deg2rad_: deg2rad_
173
+ detach: detach
174
+ diag: diag
175
+ diag_embed: diag_embed
176
+ diagonal: diagonal
177
+ diagonal_copy: diagonal
178
+ diagonal_scatter: diagonal
179
+ digamma: digamma
180
+ digamma_: digamma_
181
+ dist: dist
182
+ div: div
183
+ div_: div_
184
+ divide: div
185
+ divide_: divide_
186
+ dot: dot
187
+ dropout: dropout
188
+ elu: ELU
189
+ elu_: elu_
190
+ embedding: embedding
191
+ empty_like: empty_like
192
+ empty_strided: empty_strided
193
+ eq: eq
194
+ eq_: eq_
195
+ erf: erf
196
+ erf_: erf_
197
+ erfc: erfc
198
+ erfc_: erfc_
199
+ erfinv: erfinv
200
+ erfinv_: erfinv_
201
+ exp: exp
202
+ exp2: exp2
203
+ exp2_: exp2_
204
+ exp_: exp_
205
+ expand: expand
206
+ expm1: expm1
207
+ expm1_: expm1_
208
+ exponential: Exponential
209
+ exponential_: exponential_
210
+ eye: eye
211
+ fft_fft: fft
212
+ fft_fft2: fft2
213
+ fft_fftn: fftn
214
+ fft_fftshift: fftshift
215
+ fft_hfft: hfft
216
+ fft_hfft2: hfft2
217
+ fft_hfftn: hfftn
218
+ fft_ifft: ifft
219
+ fft_ifft2: ifft2
220
+ fft_ifftn: ifftn
221
+ fft_ifftshift: ifftshift
222
+ fft_ihfft: ihfft
223
+ fft_ihfft2: ihfft2
224
+ fft_ihfftn: ifftn
225
+ fft_irfft: irfft
226
+ fft_irfft2: irfft2
227
+ fft_irfftn: irfftn
228
+ fft_rfft: rfft
229
+ fft_rfft2: rfft2
230
+ fft_rfftn: rfftn
231
+ fill: fill_
232
+ fill_: fill_
233
+ fix: fix
234
+ fix_: fix_
235
+ flip: flip
236
+ float_power_: float_power_
237
+ floor: floor
238
+ floor_: floor_
239
+ floor_divide: floor_divide
240
+ floor_divide_: floor_divide_
241
+ fmax: fmax
242
+ fmin: fmin
243
+ fmod: fmod
244
+ fmod_: fmod_
245
+ frac: frac
246
+ frac_: frac_
247
+ full: full
248
+ full_like: full_like
249
+ gather: gather
250
+ gcd: gcd
251
+ gcd_: gcd_
252
+ ge: ge
253
+ ge_: ge_
254
+ gelu: GELU
255
+ gelu_: gelu_
256
+ geometric: Geometric
257
+ geometric_: geometric_
258
+ glu: glu
259
+ greater: gt
260
+ greater_: ge_
261
+ greater_equal: ge
262
+ greater_equal_: ge_
263
+ grid_sampler_2d: grid_sample
264
+ grid_sampler_3d: grid_sample
265
+ gru: GRU
266
+ gt: gt
267
+ gt_: gt_
268
+ hardshrink: Hardshrink
269
+ hardsigmoid: hardsigmoid
270
+ hardsigmoid_: hardsigmoid_
271
+ hardswish: hardswish
272
+ hardswish_: hardswish_
273
+ hardtanh: hardtanh
274
+ hardtanh_: hardtanh_
275
+ heaviside: heaviside
276
+ heaviside_: heaviside_
277
+ hinge_embedding_loss: HingeEmbeddingLoss
278
+ huber_loss: huber_loss
279
+ hypot: hypot
280
+ hypot_: hypot_
281
+ i0: i0
282
+ i0_: i0_
283
+ igamma: igamma
284
+ igamma_: igamma_
285
+ igammac: igammac
286
+ igammac_: igammac_
287
+ index: __getitem__
288
+ index_add: index_add
289
+ index_add_: index_add_
290
+ index_copy: index_copy_
291
+ index_copy_: index_copy_
292
+ index_fill: index_fill_
293
+ index_fill_: index_fill_
294
+ index_put: index_put_
295
+ index_put_: index_put_
296
+ index_reduce: index_select
297
+ index_select: index_select
298
+ is_pinned: is_pinned
299
+ is_same_size: is_same_size
300
+ isinf: isinf
301
+ isnan: isnan
302
+ isneginf: isneginf
303
+ isposinf: isposinf
304
+ istft: istft
305
+ item: item
306
+ lcm: lcm
307
+ lcm_: lcm_
308
+ le: le
309
+ le_: le_
310
+ leaky_relu: LeakyReLU
311
+ leaky_relu_: leaky_relu_
312
+ lerp: lerp
313
+ lerp_: lerp_
314
+ less: less
315
+ less_: less_
316
+ less_equal: le
317
+ less_equal_: less_equal_
318
+ lgamma: lgamma
319
+ lgamma_: lgamma_
320
+ linalg_cholesky_ex: cholesky
321
+ linalg_cross: cross
322
+ linalg_householder_product: householder_product
323
+ linalg_inv_ex: inv
324
+ linalg_ldl_factor_ex: ldl
325
+ linalg_ldl_solve: ldl_solve
326
+ linalg_lu: lu
327
+ linalg_lu_factor_ex: lu_factor
328
+ linalg_lu_solve: lu_solve
329
+ linalg_matrix_exp: matrix_exp
330
+ linalg_qr: qr
331
+ linalg_solve_triangular: solve
332
+ linalg_vector_norm: norm
333
+ linspace: linspace
334
+ log: log
335
+ log10: log10
336
+ log10_: log10_
337
+ log1p: log1p
338
+ log1p_: log1p_
339
+ log2: log2
340
+ log2_: log2_
341
+ log_: log_
342
+ log_normal: LogNormal
343
+ log_sigmoid_forward: log_sigmoid
344
+ logaddexp: logaddexp
345
+ logaddexp2: logaddexp2
346
+ _native_batch_norm_legit_functional: batch_norm
347
+ logcumsumexp: logcumsumexp
348
+ logical_and: logical_and
349
+ logical_and_: logical_and_
350
+ logical_not: logical_not
351
+ logical_not_: logical_not_
352
+ logical_or: logical_or
353
+ logical_or_: logical_or_
354
+ logical_xor: logical_xor
355
+ logical_xor_: logical_xor_
356
+ logit: logit
357
+ logit_: logit_
358
+ logspace: logspace
359
+ logsumexp: logsumexp
360
+ lstm: LSTM
361
+ lt: lt
362
+ lt_: lt_
363
+ lu_unpack: lu_unpack
364
+ margin_ranking_loss: margin_ranking_loss
365
+ masked_fill: masked_fill
366
+ masked_fill_: masked_fill_
367
+ matmul: matmul
368
+ max: max
369
+ max_pool2d_with_indices: MaxPool2d
370
+ max_pool3d_with_indices: MaxPool3d
371
+ max_unpool2d: MaxUnpool2d
372
+ max_unpool3d: max_unpool3d
373
+ maximum: maximum
374
+ mean: mean
375
+ median: median
376
+ meshgrid: meshgrid
377
+ min: min
378
+ minimum: minimum
379
+ mish: Mish
380
+ mish_: mish_
381
+ mm: mm
382
+ mode: mode
383
+ mse_loss: mse_loss
384
+ mul: mul
385
+ mul_: mul_
386
+ multi_margin_loss: MultiMarginLoss
387
+ multilabel_margin_loss_forward: multilabel_margin_loss
388
+ multinomial: multinomial
389
+ multiply: multiply
390
+ multiply_: mul_
391
+ mv: mv
392
+ mvlgamma: mvlgamma
393
+ mvlgamma_: mvlgamma_
394
+ name: name
395
+ nan_to_num: nan_to_num
396
+ nan_to_num_: nan_to_num_
397
+ nanmedian: nanmedian
398
+ nansum: nansum
399
+ narrow_copy: narrow
400
+ native_batch_norm: BatchNorm2d
401
+ native_dropout: dropout
402
+ native_group_norm: group_norm
403
+ native_layer_norm: LayerNorm
404
+ ne: ne
405
+ ne_: ne_
406
+ neg: neg
407
+ neg_: neg_
408
+ negative: neg
409
+ negative_: neg_
410
+ new_empty: new_empty
411
+ new_empty_strided: new_empty_strided
412
+ new_full: new_full
413
+ new_ones: new_ones
414
+ new_zeros: new_zeros
415
+ nextafter: nextafter
416
+ nextafter_: nextafter_
417
+ nll_loss: nll_loss
418
+ nll_loss2d_forward: NLLLoss2d
419
+ nll_loss_forward: NLLLoss
420
+ nonzero_static: nonzero
421
+ norm: norm
422
+ normal: normal
423
+ normal_: normal_
424
+ not_equal: ne
425
+ not_equal_: ne_
426
+ ones: ones
427
+ ones_like: ones_like
428
+ ormqr: ormqr
429
+ pairwise_distance: pairwise_distance
430
+ pdist: pdist
431
+ permute: permute
432
+ pin_memory: pin_memory
433
+ pixel_shuffle: PixelShuffle
434
+ polar: polar
435
+ polygamma: polygamma
436
+ positive: positive
437
+ pow: pow
438
+ pow_: pow_
439
+ prelu: prelu
440
+ prod: prod
441
+ quantized_gru: GRU
442
+ quantized_lstm: LSTM
443
+ rad2deg: rad2deg
444
+ rad2deg_: rad2deg_
445
+ rand: rand
446
+ rand_like: rand_like
447
+ randint: randint
448
+ randint_like: randint_like
449
+ randn: randn
450
+ randn_like: randn_like
451
+ randperm: randperm
452
+ reciprocal: reciprocal
453
+ reciprocal_: reciprocal_
454
+ reflection_pad1d: reflection_pad1d
455
+ reflection_pad2d: reflection_pad2d
456
+ reflection_pad3d: ReflectionPad3d
457
+ relu: relu
458
+ relu6: relu6
459
+ relu_: relu_
460
+ remainder: remainder
461
+ remainder_: remainder_
462
+ renorm: renorm
463
+ renorm_: renorm_
464
+ repeat: repeat
465
+ repeat_interleave: repeat_interleave
466
+ replication_pad1d: ReplicationPad1d
467
+ replication_pad2d: replication_pad2d
468
+ replication_pad3d: replication_pad3d
469
+ resize_as_: resize_as_
470
+ rnn_relu: RNN
471
+ rnn_tanh: RNN
472
+ roll: roll
473
+ rot90: rot90
474
+ round: round
475
+ round_: round_
476
+ rrelu_with_noise: RReLU
477
+ rrelu_with_noise_: rrelu_with_noise
478
+ rsqrt: rsqrt
479
+ rsqrt_: rsqrt_
480
+ rsub: rsub
481
+ scalar_tensor: scalar_tensor
482
+ scatter: scatter_
483
+ scatter_: scatter_
484
+ scatter_add: scatter_add
485
+ scatter_add_: scatter_add_
486
+ searchsorted: searchsorted
487
+ select: select
488
+ selu: selu
489
+ selu_: selu_
490
+ sgn: sgn
491
+ sgn_: sgn_
492
+ sigmoid: sigmoid
493
+ sigmoid_: sigmoid_
494
+ sign: sign
495
+ sign_: sign_
496
+ signbit: signbit
497
+ silu: silu
498
+ silu_: silu_
499
+ sin: sin
500
+ sin_: sin_
501
+ sinc: sinc
502
+ sinc_: sinc_
503
+ sinh: sinh
504
+ sinh_: sinh_
505
+ slice: slice
506
+ smooth_l1_loss: smooth_l1_loss
507
+ soft_margin_loss: soft_margin_loss
508
+ softplus: softplus
509
+ softshrink: softshrink
510
+ sort: sort
511
+ special_airy_ai: airy_ai
512
+ special_bessel_j0: j0
513
+ special_bessel_j1: j1
514
+ special_bessel_y0: y0
515
+ special_bessel_y1: y1
516
+ special_chebyshev_polynomial_t: chebyshev_t
517
+ special_chebyshev_polynomial_u: chebyshev_u
518
+ special_entr: entr
519
+ special_erfcx: erfcx
520
+ special_hermite_polynomial_h: hermite
521
+ special_hermite_polynomial_he: he
522
+ special_i0: i0
523
+ special_i0e: i0e
524
+ special_i1: i1
525
+ special_i1e: i1e
526
+ special_laguerre_polynomial_l: laguerre_l
527
+ special_log_ndtr: log_ndtr
528
+ special_modified_bessel_i0: i0
529
+ special_modified_bessel_i1: i1
530
+ special_modified_bessel_k0: k0
531
+ special_modified_bessel_k1: i1
532
+ special_ndtr: ndtr
533
+ special_ndtri: ndtri
534
+ special_scaled_modified_bessel_k0: i0e
535
+ special_scaled_modified_bessel_k1: scaled_modified_bessel_k1
536
+ special_spherical_bessel_j0: spherical_jn
537
+ special_xlog1py: xlog1py
538
+ special_zeta: zeta
539
+ split: split
540
+ split_with_sizes: split
541
+ sqrt: sqrt
542
+ sqrt_: sqrt_
543
+ square: square
544
+ square_: square_
545
+ squeeze: squeeze
546
+ stack: stack
547
+ std: std
548
+ std_mean: std_mean
549
+ stft: stft
550
+ sub: sub
551
+ sub_: sub_
552
+ subtract: sub
553
+ subtract_: subtract_
554
+ sum: sum
555
+ t: t
556
+ t_: t_
557
+ take: take
558
+ tan: tan
559
+ tan_: tan_
560
+ tanh: tanh
561
+ tanh_: tanh_
562
+ threshold: threshold
563
+ threshold_: threshold_
564
+ to: to
565
+ topk: topk
566
+ trace: trace
567
+ transpose: transpose
568
+ transpose_: transpose_
569
+ triangular_solve: triangular_solve
570
+ tril: tril
571
+ tril_: tril_
572
+ tril_indices: tril_indices
573
+ triu: triu
574
+ triu_: triu_
575
+ triu_indices: triu_indices
576
+ true_divide: true_divide
577
+ true_divide_: true_divide_
578
+ trunc: trunc
579
+ trunc_: trunc_
580
+ unbind: unbind
581
+ unfold: unfold
582
+ uniform: Uniform
583
+ uniform_: uniform_
584
+ unsafe_chunk: unsafe_chunk
585
+ unsafe_split: split
586
+ unsafe_split_with_sizes: split_with_sizes
587
+ unsqueeze: unsqueeze
588
+ unsqueeze_: unsqueeze_
589
+ upsample_bicubic2d: interpolate
590
+ upsample_bilinear2d: upsample_bilinear
591
+ upsample_nearest1d: interpolate
592
+ upsample_nearest2d: interpolate
593
+ upsample_nearest3d: interpolate
594
+ var: var
595
+ var_mean: var_mean
596
+ vdot: vdot
597
+ view: view
598
+ where: where
599
+ xlogy: xlogy
600
+ xlogy_: xlogy_
601
+ zero: zeros
602
+ zero_: zero_
603
+ zeros: zeros
604
+ zeros_like: zeros_like
605
+
606
+
607
+