()
| 57 | with self.session() as session, self.test_scope(): |
| 58 | |
| 59 | def fn(): |
| 60 | ta = tensor_array_ops.TensorArray( |
| 61 | dtype=dtypes.float32, tensor_array_name="foo", size=3) |
| 62 | |
| 63 | w0 = ta.write(0, [[4.0, 5.0]]) |
| 64 | w1 = w0.write(1, [[1.0, 3.0]]) |
| 65 | w2 = w1.write(2, [[7.0, -8.5]]) |
| 66 | |
| 67 | r0 = w2.read(0) |
| 68 | r1 = w2.read(1) |
| 69 | r2 = w2.read(2) |
| 70 | flow = w2.flow |
| 71 | return [r0, r1, r2, flow] |
| 72 | |
| 73 | d0, d1, d2, flow_val = self.evaluate(xla.compile(fn)) |
| 74 | self.assertAllEqual([[4.0, 5.0]], d0) |