Home
last modified time | relevance | path

Searched refs:test_writer (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/ut/python/train/summary/
Dtest_histogram_summary.py56 with SummaryRecord(tmp_dir, file_suffix="_MS_HISTOGRAM") as test_writer:
59 test_writer.record(step=1)
60 file_name = os.path.realpath(test_writer.log_dir)
70 with SummaryRecord(tmp_dir, file_suffix="_MS_HISTOGRAM") as test_writer:
81 test_writer.record(step=i)
83 file_name = os.path.realpath(test_writer.log_dir)
94 with SummaryRecord(tmp_dir, file_suffix="_MS_HISTOGRAM") as test_writer:
97 test_writer.record(step=1)
99 file_name = os.path.realpath(test_writer.log_dir)
109 with SummaryRecord(tmp_dir, file_suffix="_MS_HISTOGRAM") as test_writer:
[all …]
Dtest_image_summary.py69 with SummaryRecord(SUMMARY_DIR, file_suffix="_MS_IMAGE") as test_writer:
74 test_writer.record(i)
75 test_writer.flush()
166 with SummaryRecord(SUMMARY_DIR, file_suffix="_MS_IMAGE") as test_writer:
168 callback = ImageSummaryCallback(test_writer)
187 with SummaryRecord(SUMMARY_DIR, file_suffix="_MS_IMAGE") as test_writer:
189 test_writer.record(1)
/third_party/mindspore/tests/st/summary/
Dtest_cpu_summary.py51 def train_summary_record(test_writer, steps): argument
58 test_writer.record(i)
75 with SummaryRecord(tmp_dir) as test_writer:
76 train_summary_record(test_writer, steps=steps)
78 file_name = os.path.realpath(test_writer.log_dir)
/third_party/mindspore/tests/st/auto_monad/
Dtest_effect_ops.py372 def train_summary_record(test_writer, steps): argument
379 test_writer.record(i)
392 with SummaryRecord(tmp_dir) as test_writer:
393 train_summary_record(test_writer, steps=steps)
395 file_name = os.path.realpath(test_writer.log_dir)