(i, r)
| 331 | c = lambda i, r: math_ops.less(i, max_squarings) |
| 332 | |
| 333 | def b(i, r): |
| 334 | return i + 1, array_ops.where( |
| 335 | math_ops.less(i, squarings), math_ops.matmul(r, r), r) |
| 336 | |
| 337 | _, result = control_flow_ops.while_loop(c, b, [i, result]) |
| 338 | if not matrix.shape.is_fully_defined(): |