experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Expand an error message with a known workaround; add flag to enable said workaround.

Browse Source 
PiperOrigin-RevId: 254105522
This commit is contained in:
A. Unique TensorFlower 2019-06-19 17:31:45 -07:00 committed by TensorFlower Gardener
parent 401bbfc336
commit 223d17868c
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/client/session.py
View File

@ -1359,6 +1359,11 @@ class BaseSession(SessionInterface):
except KeyError: except KeyError:
pass pass
message = error_interpolation.interpolate(message, self._graph) message = error_interpolation.interpolate(message, self._graph)
if 'only supports NHWC tensor format' in message:
message += ('\nA possible workaround: Try disabling Grappler optimizer'
'\nby modifying the config for creating the session eg.'
'\nsession_config.graph_options.rewrite_options.'
'disable_meta_optimizer = True')
raise type(e)(node_def, op, message) raise type(e)(node_def, op, message)
def _extend_graph(self): def _extend_graph(self):