| 108 | } |
| 109 | |
| 110 | StatusOr<Node*> BuildArgNode(Graph* graph, DataType type, int index) { |
| 111 | const char* const kArgOp = "_Arg"; |
| 112 | NodeDef arg_def; |
| 113 | NodeDefBuilder builder(absl::StrCat(kArgOp, index), kArgOp); |
| 114 | builder.Attr("T", type); |
| 115 | builder.Attr("index", index); |
| 116 | TF_RETURN_IF_ERROR(builder.Finalize(&arg_def)); |
| 117 | return AddNodeDefToGraph(arg_def, graph); |
| 118 | } |
| 119 | |
| 120 | // Builds a graph for the loop condition. |
| 121 | Status BuildLoopCondition(const Graph& graph, WhileLoopFrame* frame, |
no test coverage detected