diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/conditional_accumulator_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/conditional_accumulator_test.py | 38 |
1 files changed, 19 insertions, 19 deletions
diff --git a/tensorflow/python/kernel_tests/conditional_accumulator_test.py b/tensorflow/python/kernel_tests/conditional_accumulator_test.py index 86802664d1..262352a9af 100644 --- a/tensorflow/python/kernel_tests/conditional_accumulator_test.py +++ b/tensorflow/python/kernel_tests/conditional_accumulator_test.py @@ -80,26 +80,26 @@ class ConditionalAccumulatorTest(test.TestCase): """, q.accumulator_ref.op.node_def) def testAccumulatorSizeEmpty(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator(dtypes_lib.float32, name="Q") self.assertEqual(q.num_accumulated().eval(), 0) def testAccumulatorSetGlobalStep(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) set_global_step_op = q.set_global_step(1) set_global_step_op.run() def testAccumulatorApplyGradFloat32(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) accum_op = q.apply_grad((10.0,)) accum_op.run() def testDtypes(self): - with self.test_session() as sess: + with self.cached_session() as sess: dtypes = [dtypes_lib.float16, dtypes_lib.float32, dtypes_lib.float64] for i in range(len(dtypes)): @@ -116,7 +116,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertEqual(sum(elems) / len(elems), result) def testAccumulatorMultipleAccumulators(self): - with self.test_session(): + with self.cached_session(): q_f32_0 = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) q_f32_1 = data_flow_ops.ConditionalAccumulator( @@ -135,7 +135,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertEqual(result, i + 10.0) def testAccumulatorApplyAndTakeGradWithShape(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=(3, 2)) elems = [[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], @@ -166,7 +166,7 @@ class ConditionalAccumulatorTest(test.TestCase): q.apply_grad([[1.0], [2.0], [3.0]]) def testAccumulatorDynamicShape(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=None) @@ -191,7 +191,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertTrue(is_all_equal) def testAccumulatorWrongDynamicShape(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=None) @@ -209,7 +209,7 @@ class ConditionalAccumulatorTest(test.TestCase): sess.run(accum_op, feed_dict={x: [[1.0], [2.0], [3.0]]}) def testAccumulatorSizeAfterApplyGrad(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) accum_op = q.apply_grad((10.0,)) @@ -220,7 +220,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertEqual(q.num_accumulated().eval(), 2) def testAccumulatorSizeAfterApplyGradAndTakeGrad(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) accum_op = q.apply_grad((10.0,)) @@ -248,7 +248,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertEqual(q.num_accumulated().eval(), 0) def testAccumulatorTakeGradMean(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) elems = [10.0, 20.0] @@ -307,7 +307,7 @@ class ConditionalAccumulatorTest(test.TestCase): reduction_type="Invalid") def testAccumulatorInvalidTakeGrad(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) elems = [10.0, 20.0] @@ -322,7 +322,7 @@ class ConditionalAccumulatorTest(test.TestCase): takeg_t.eval() def testAccumulatorRepeatedTakeGradMean(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) @@ -379,7 +379,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertEqual(elems_sum, val) def testAccumulatorIncrementGlobalStep(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) @@ -395,7 +395,7 @@ class ConditionalAccumulatorTest(test.TestCase): inc_global_step.eval() def testAccumulatorSetGlobalStepPreventsAccumulation(self): - with self.test_session(): + with self.cached_session(): q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) @@ -416,7 +416,7 @@ class ConditionalAccumulatorTest(test.TestCase): if x >= ls), val) def testParallelApplyGrad(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) elems = [10.0, 20.0, 30.0, 40.0, 50.0, 60.0, 70.0, 80.0, 90.0, 100.0] @@ -441,7 +441,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertEqual(val, sum(elems) / len(elems)) def testParallelTakeGrad(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) elems = [e for e in range(10)] @@ -473,7 +473,7 @@ class ConditionalAccumulatorTest(test.TestCase): self.assertItemsEqual(elems, results) def testAccumulatorApplyAndBlockingTake(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) @@ -506,7 +506,7 @@ class ConditionalAccumulatorTest(test.TestCase): sess.run(takeg_op) def testAccumulatorCancel(self): - with self.test_session() as sess: + with self.cached_session() as sess: q = data_flow_ops.ConditionalAccumulator( dtypes_lib.float32, name="Q", shape=tensor_shape.TensorShape([1])) takeg_t = q.take_grad(1) |