remove context, change import

This commit is contained in:
abhichou4 2020-06-30 00:39:02 +05:30
parent 9c8db3c6b5
commit 4b5ef9c844
2 changed files with 1 additions and 4 deletions

View File

@ -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*"):

View File

@ -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