def _testWriteCloseTensorArray(self, use_gpu):
with self.test_session(use_gpu=use_gpu) as sess:
h = gen_data_flow_ops._tensor_array(
dtype=tf.float32, tensor_array_name="foo", size=3)
writes = [
gen_data_flow_ops._tensor_array_write(h, 0, [[4.0, 5.0]]),
gen_data_flow_ops._tensor_array_write(h, 1, [3.0]),
gen_data_flow_ops._tensor_array_write(h, 2, -1.0)]
with tf.control_dependencies(writes):
close = gen_data_flow_ops._tensor_array_close(h)
sess.run(close)
def close(self, name=None):
"""Close the current TensorArray."""
return gen_data_flow_ops._tensor_array_close(handle=self._handle,
name=name)
def close(self, name=None):
"""Close the current TensorArray."""
with ops.colocate_with(self._handle):
return gen_data_flow_ops._tensor_array_close(handle=self._handle, name=name)
def close(self): """Close the current TensorArray.""" return gen_data_flow_ops._tensor_array_close(handle=self._handle)
def _testCloseTensorArray(self, use_gpu):
with self.test_session(use_gpu=use_gpu) as sess:
h = gen_data_flow_ops._tensor_array(
dtype=tf.float32, tensor_array_name="foo", size=3)
c1 = gen_data_flow_ops._tensor_array_close(h)
sess.run(c1)
