(self, fn, *args)
| 1370 | return self._do_call(_prun_fn, handle, feeds, fetches) |
| 1371 | |
| 1372 | def _do_call(self, fn, *args): |
| 1373 | try: |
| 1374 | return fn(*args) |
| 1375 | except errors.OpError as e: |
| 1376 | message = compat.as_text(e.message) |
| 1377 | m = BaseSession._NODEDEF_NAME_RE.search(message) |
| 1378 | node_def = None |
| 1379 | op = None |
| 1380 | if m is not None: |
| 1381 | node_name = m.group(3) |
| 1382 | try: |
| 1383 | op = self._graph.get_operation_by_name(node_name) |
| 1384 | node_def = op.node_def |
| 1385 | except KeyError: |
| 1386 | pass |
| 1387 | message = error_interpolation.interpolate(message, self._graph) |
| 1388 | if 'only supports NHWC tensor format' in message: |
| 1389 | message += ('\nA possible workaround: Try disabling Grappler optimizer' |
| 1390 | '\nby modifying the config for creating the session eg.' |
| 1391 | '\nsession_config.graph_options.rewrite_options.' |
| 1392 | 'disable_meta_optimizer = True') |
| 1393 | raise type(e)(node_def, op, message) |
| 1394 | |
| 1395 | def _extend_graph(self): |
| 1396 | with self._graph._session_run_lock(): # pylint: disable=protected-access |
no test coverage detected