Exemplo n.º 1
0
 def testDestroyTemporaryVariableTwice(self):
   with test_util.use_gpu():
     var = gen_state_ops.temporary_variable([1, 2], dtypes.float32)
     val1 = gen_state_ops.destroy_temporary_variable(var, var_name="dup")
     val2 = gen_state_ops.destroy_temporary_variable(var, var_name="dup")
     final = val1 + val2
     with self.assertRaises(errors.NotFoundError):
       self.evaluate(final)
Exemplo n.º 2
0
 def testDestroyTemporaryVariableTwice(self):
   with test_util.use_gpu():
     var = gen_state_ops.temporary_variable([1, 2], dtypes.float32)
     val1 = gen_state_ops.destroy_temporary_variable(var, var_name="dup")
     val2 = gen_state_ops.destroy_temporary_variable(var, var_name="dup")
     final = val1 + val2
     with self.assertRaises(errors.NotFoundError):
       self.evaluate(final)
Exemplo n.º 3
0
 def testDestroyNonexistentTemporaryVariable(self):
     with test_util.use_gpu():
         var = gen_state_ops.temporary_variable([1, 2], dtypes.float32)
         final = gen_state_ops.destroy_temporary_variable(var,
                                                          var_name="bad")
         with self.assertRaises(errors.NotFoundError):
             self.evaluate(final)
 def testDestroyNonexistentTemporaryVariable(self):
     with self.test_session(use_gpu=True):
         var = gen_state_ops.temporary_variable([1, 2], dtypes.float32)
         final = gen_state_ops.destroy_temporary_variable(var,
                                                          var_name="bad")
         with self.assertRaises(errors.NotFoundError):
             final.eval()
Exemplo n.º 5
0
 def testTemporaryVariable(self):
   with test_util.use_gpu():
     var = gen_state_ops.temporary_variable(
         [1, 2], dtypes.float32, var_name="foo")
     var = state_ops.assign(var, [[4.0, 5.0]])
     var = state_ops.assign_add(var, [[6.0, 7.0]])
     final = gen_state_ops.destroy_temporary_variable(var, var_name="foo")
     self.assertAllClose([[10.0, 12.0]], self.evaluate(final))
Exemplo n.º 6
0
 def testTemporaryVariable(self):
   with self.test_session(use_gpu=True):
     var = gen_state_ops.temporary_variable(
         [1, 2], dtypes.float32, var_name="foo")
     var = state_ops.assign(var, [[4.0, 5.0]])
     var = state_ops.assign_add(var, [[6.0, 7.0]])
     final = gen_state_ops.destroy_temporary_variable(var, var_name="foo")
     self.assertAllClose([[10.0, 12.0]], final.eval())
Exemplo n.º 7
0
 def testTemporaryVariable(self):
   with test_util.use_gpu():
     var = gen_state_ops.temporary_variable(
         [1, 2], dtypes.float32, var_name="foo")
     var = state_ops.assign(var, [[4.0, 5.0]])
     var = state_ops.assign_add(var, [[6.0, 7.0]])
     final = gen_state_ops.destroy_temporary_variable(var, var_name="foo")
     self.assertAllClose([[10.0, 12.0]], self.evaluate(final))
Exemplo n.º 8
0
 def testTemporaryVariable(self):
   with self.test_session(use_gpu=True):
     var = gen_state_ops.temporary_variable(
         [1, 2], dtypes.float32, var_name="foo")
     var = state_ops.assign(var, [[4.0, 5.0]])
     var = state_ops.assign_add(var, [[6.0, 7.0]])
     final = gen_state_ops.destroy_temporary_variable(var, var_name="foo")
     self.assertAllClose([[10.0, 12.0]], final.eval())
 def _AccumulateNTemplate(self, inputs, init, shape, validate_shape):
   var = gen_state_ops.temporary_variable(
       shape=shape, dtype=inputs[0].dtype.base_dtype)
   ref = state_ops.assign(var, init, validate_shape=validate_shape)
   update_ops = [
       state_ops.assign_add(
           ref, tensor, use_locking=True).op for tensor in inputs
   ]
   with ops.control_dependencies(update_ops):
     return gen_state_ops.destroy_temporary_variable(ref, var_name=var.op.name)
Exemplo n.º 10
0
 def _AccumulateNTemplate(self, inputs, init, shape, validate_shape):
   var = gen_state_ops.temporary_variable(
       shape=shape, dtype=inputs[0].dtype.base_dtype)
   ref = state_ops.assign(var, init, validate_shape=validate_shape)
   update_ops = [
       state_ops.assign_add(
           ref, tensor, use_locking=True).op for tensor in inputs
   ]
   with ops.control_dependencies(update_ops):
     return gen_state_ops.destroy_temporary_variable(ref, var_name=var.op.name)
Exemplo n.º 11
0
 def testDestroyNonexistentTemporaryVariable(self):
   with test_util.use_gpu():
     var = gen_state_ops.temporary_variable([1, 2], dtypes.float32)
     final = gen_state_ops.destroy_temporary_variable(var, var_name="bad")
     with self.assertRaises(errors.NotFoundError):
       self.evaluate(final)
Exemplo n.º 12
0
 def testDestroyNonexistentTemporaryVariable(self):
   with self.test_session(use_gpu=True):
     var = gen_state_ops.temporary_variable([1, 2], dtypes.float32)
     final = gen_state_ops.destroy_temporary_variable(var, var_name="bad")
     with self.assertRaises(errors.NotFoundError):
       final.eval()