Searched refs:export_options (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/train/summary/ |
| D | summary_record.py | 64 def process_export_options(export_options): argument
66 if export_options is None:
69 check_value_type('export_options', export_options, [dict, type(None)])
71 for export_option, export_format in export_options.items():
75 unexpected_params = set(export_options) - set(_DEFAULT_EXPORT_OPTIONS)
80 for export_option, export_format in export_options.items():
87 for item in set(export_options):
88 check_value_type(item, export_options.get(item), [str, type(None)])
90 return export_options
147 network=None, max_file_size=None, raise_exception=False, export_options=None): argument
[all …]
|
| /third_party/mindspore/tests/ut/python/train/summary/ |
| D | test_summary_collector.py | 159 def test_params_with_tensor_format_type_error(self, export_options): argument
163 SummaryCollector(summary_dir, export_options=export_options)
165 unexpected_format = {export_options.get("tensor_format")}
174 def test_params_with_export_options_type_error(self, export_options): argument
178 SummaryCollector(summary_dir, export_options=export_options)
256 SummaryCollector(summary_dir, export_options=data)
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _summary_collector.py | 197 export_options=None): argument
217 self._export_options = process_export_options(export_options)
238 export_options=self._export_options)
|
| /third_party/mindspore/tests/st/summary/ |
| D | test_summary_collector.py | 178 export_options={'tensor_format': 'npy'})
|