Home
last modified time | relevance | path

Searched refs:_convert_image_format (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/train/summary/
D_summary_adapter.py369 tensor = _convert_image_format(np_value, input_format)
413 def _convert_image_format(np_tensor, input_format, out_format='HWC'): function
/third_party/mindspore/mindspore/explainer/
D_image_classification_runner.py33 from mindspore.train.summary._summary_adapter import _convert_image_format
570 data_np = _convert_image_format(np.expand_dims(inp.asnumpy(), 0), 'NCHW')