From 4b5ef9c84402d0ff6f6ec82c1f4883a0f51d1644 Mon Sep 17 00:00:00 2001 From: abhichou4 Date: Tue, 30 Jun 2020 00:39:02 +0530 Subject: [PATCH] remove context, change import --- tensorflow/python/eager/forwardprop_test.py | 3 --- tensorflow/python/ops/parallel_for/gradients.py | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/tensorflow/python/eager/forwardprop_test.py b/tensorflow/python/eager/forwardprop_test.py index be16195c63f..015dff37506 100644 --- a/tensorflow/python/eager/forwardprop_test.py +++ b/tensorflow/python/eager/forwardprop_test.py @@ -286,9 +286,6 @@ class ForwardpropTest(test.TestCase, parameterized.TestCase): ) def testJVPFunctionRaisesError(self): - context.ensure_initialized() - ctx = context.context() - sum_outputs = (constant_op.constant(6.),) with self.assertRaisesRegexp(ValueError, r".*was expected to be of shape*"): diff --git a/tensorflow/python/ops/parallel_for/gradients.py b/tensorflow/python/ops/parallel_for/gradients.py index 94ab49951aa..81a00d4ab69 100644 --- a/tensorflow/python/ops/parallel_for/gradients.py +++ b/tensorflow/python/ops/parallel_for/gradients.py @@ -20,7 +20,7 @@ from __future__ import print_function from tensorflow.python.framework import ops from tensorflow.python.ops import array_ops from tensorflow.python.ops import check_ops -from tensorflow.python.ops import gradients as gradient_ops +from tensorflow.python.ops import gradients_impl as gradient_ops from tensorflow.python.ops.parallel_for import control_flow_ops from tensorflow.python.util import nest