(i)
| 40 | x = random_ops.random_uniform([3, 3, 3]) |
| 41 | |
| 42 | def loop_fn(i): |
| 43 | outputs = [] |
| 44 | x_i = array_ops.gather(x, i) |
| 45 | for y in [x, x_i]: |
| 46 | axes = [0, 2, -1] if y is x else [0] |
| 47 | for axis in axes: |
| 48 | outputs.append(array_ops.gather(y, 2, axis=axis)) |
| 49 | outputs.append(array_ops.gather(y, i, axis=axis)) |
| 50 | outputs.append(array_ops.gather(y, [i], axis=axis)) |
| 51 | outputs.append(array_ops.gather(y, [i, 2], axis=axis)) |
| 52 | outputs.append(array_ops.gather(y, [[2, i], [i, 1]], axis=axis)) |
| 53 | return outputs |
| 54 | |
| 55 | self._test_loop_fn(loop_fn, 3, loop_fn_dtypes=[dtypes.float32] * 20) |
| 56 |
nothing calls this directly
no test coverage detected