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)