diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-04-03 10:04:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-03 10:09:05 -0700 |
| commit | 655b2663e7d2609b0f578b9ef3c1401de22dc5c2 (patch) | |
| tree | 61ac9426556a65f99edf99a23fa491d329d70016 /tensorflow/python/grappler | |
| parent | cfc886ac6064a04c71dd6c52e8c21ebec91eae50 (diff) | |
Apply "Raise exception in SWIG on bad TF_Status" to base.i
Minor fixes to make this work.
PiperOrigin-RevId: 191457070
Diffstat (limited to 'tensorflow/python/grappler')
| -rw-r--r-- | tensorflow/python/grappler/item.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/python/grappler/item.py b/tensorflow/python/grappler/item.py index 4a083849bd..1748efdd13 100644 --- a/tensorflow/python/grappler/item.py +++ b/tensorflow/python/grappler/item.py @@ -51,9 +51,7 @@ class Item(object): self._BuildTFItem() def IdentifyImportantOps(self, sort_topologically=False): - with errors.raise_exception_on_not_ok_status() as status: - return tf_item.TF_IdentifyImportantOps(self.tf_item, sort_topologically, - status) + return tf_item.TF_IdentifyImportantOps(self.tf_item, sort_topologically) def GetOpProperties(self): ret_from_swig = tf_item.TF_GetOpProperties(self.tf_item) |