Home
last modified time | relevance | path

Searched refs:PluginEnum (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/train/summary/
D_writer_pool.py25 from mindspore.train.summary.enums import PluginEnum, WriterPluginEnum
44 if plugin == PluginEnum.GRAPH.value:
46 elif plugin in (PluginEnum.TRAIN_LINEAGE.value, PluginEnum.EVAL_LINEAGE.value,
47 PluginEnum.CUSTOM_LINEAGE_DATA.value, PluginEnum.DATASET_GRAPH.value):
49 … elif plugin in (PluginEnum.SCALAR.value, PluginEnum.TENSOR.value, PluginEnum.HISTOGRAM.value,
50 PluginEnum.IMAGE.value):
53 elif plugin == PluginEnum.EXPLAINER.value:
Dwriter.py23 from mindspore.train.summary.enums import PluginEnum, WriterPluginEnum
101 if plugin in (WriterPluginEnum.SUMMARY.value, PluginEnum.GRAPH.value):
110 …if plugin in (PluginEnum.DATASET_GRAPH.value, PluginEnum.TRAIN_LINEAGE.value, PluginEnum.EVAL_LINE…
111 PluginEnum.CUSTOM_LINEAGE_DATA.value):
Denums.py28 class PluginEnum(BaseEnum): class
/third_party/mindspore/mindspore/train/callback/
D_summary_collector.py32 from mindspore.train.summary.enums import PluginEnum, ModeEnum
394 … self._collect_at_step_end(cb_params, lambda plugin: plugin == PluginEnum.TENSOR.value)
396 … self._collect_at_step_end(cb_params, lambda plugin: plugin != PluginEnum.TENSOR.value)
484 self._record.add_value(PluginEnum.IMAGE.value, 'input_data/auto', input_data)
543 self._record.add_value(PluginEnum.GRAPH.value, 'train_network/auto', graph_proto)
555 self._record.add_value(PluginEnum.SCALAR.value, 'loss/auto', loss)
665 … self._record.add_value(PluginEnum.HISTOGRAM.value, parameter.name+'/auto', parameter.data)
672 … self._record.add_value(PluginEnum.HISTOGRAM.value, parameter.name+'/auto', parameter.data)
733 … self._record.add_value(PluginEnum.TRAIN_LINEAGE.value, 'train_lineage', train_lineage_message)
918 self._record.add_value(PluginEnum.EVAL_LINEAGE.value, 'eval_lineage', eval_lineage_message)
/third_party/mindspore/tests/ut/python/train/summary/
Dtest_summary_collector.py29 from mindspore.train.summary.enums import ModeEnum, PluginEnum
452 assert PluginEnum.HISTOGRAM.value == result[0][0]