| 88 | } |
| 89 | |
| 90 | Status SetupFlowControlInputs(OpKernelContext* ctx, bool set_output) { |
| 91 | const Tensor* flow_control; |
| 92 | TF_RETURN_IF_ERROR(ctx->input("flow_in", &flow_control)); |
| 93 | if (set_output) { |
| 94 | TF_RETURN_IF_ERROR(ctx->set_output("flow_out", *flow_control)); |
| 95 | } |
| 96 | return Status::OK(); |
| 97 | } |
| 98 | |
| 99 | // CREATION ******************************************************************* |
| 100 |