aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/pywrap_tfe.i
diff options
context:
space:
mode:
authorGravatar Akshay Modi <nareshmodi@google.com>2018-04-03 10:04:47 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-03 10:09:05 -0700
commit655b2663e7d2609b0f578b9ef3c1401de22dc5c2 (patch)
tree61ac9426556a65f99edf99a23fa491d329d70016 /tensorflow/python/pywrap_tfe.i
parentcfc886ac6064a04c71dd6c52e8c21ebec91eae50 (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/pywrap_tfe.i')
-rw-r--r--tensorflow/python/pywrap_tfe.i2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i
index 39fabb9c1b..7acb8eeb1a 100644
--- a/tensorflow/python/pywrap_tfe.i
+++ b/tensorflow/python/pywrap_tfe.i
@@ -13,6 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
+%include "tensorflow/python/platform/base.i"
+
%ignore "";
%rename("%s") TFE_NewContext;