|
|
@@ -37,16 +37,16 @@ class Profiler: |
|
|
""" |
|
|
""" |
|
|
Performance profiling API. |
|
|
Performance profiling API. |
|
|
|
|
|
|
|
|
Enable MindSpore users to profile the neural network. |
|
|
|
|
|
|
|
|
Enable MindSpore users to profile the performance of neural network. |
|
|
|
|
|
|
|
|
Args: |
|
|
Args: |
|
|
subgraph (str): Defines which subgraph to monitor and analyse, can be 'all', 'Default', 'Gradients'. |
|
|
|
|
|
|
|
|
subgraph (str): Define which subgraph to monitor and analyse, can be 'all', 'Default', 'Gradients'. |
|
|
is_detail (bool): Whether to show profiling data for op_instance level, only show optype level if False. |
|
|
is_detail (bool): Whether to show profiling data for op_instance level, only show optype level if False. |
|
|
is_show_op_path (bool): Whether to save the full path for each op instance. |
|
|
is_show_op_path (bool): Whether to save the full path for each op instance. |
|
|
output_path (str): Output data path. |
|
|
output_path (str): Output data path. |
|
|
optypes_to_deal (list): Op type names, the data of which optype should be collected and analysed, |
|
|
|
|
|
will deal with all op if null. |
|
|
|
|
|
optypes_not_deal (list): Op type names, the data of which optype will not be collected and analysed. |
|
|
|
|
|
|
|
|
optypes_to_deal (list[str]): Op type names, the data of which optype should be collected and analysed, |
|
|
|
|
|
will deal with all op if null. |
|
|
|
|
|
optypes_not_deal (list[str]): Op type names, the data of which optype will not be collected and analysed. |
|
|
|
|
|
|
|
|
Examples: |
|
|
Examples: |
|
|
>>> from mindinsight.profiler import Profiler |
|
|
>>> from mindinsight.profiler import Profiler |
|
|
|