remove context, change import
This commit is contained in:
parent
9c8db3c6b5
commit
4b5ef9c844
@ -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*"):
|
||||
|
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user