| 324 | } |
| 325 | |
| 326 | StatusOr<XlaComputation> XlaBuilder::Build(bool remove_dynamic_dimensions) { |
| 327 | TF_RETURN_IF_ERROR(GetCurrentStatus()); |
| 328 | return Build(instructions_.back().id(), remove_dynamic_dimensions); |
| 329 | } |
| 330 | |
| 331 | StatusOr<XlaComputation> XlaBuilder::Build(XlaOp root, |
| 332 | bool remove_dynamic_dimensions) { |