Searched refs:bodies_ (Results 1 – 1 of 1) sorted by relevance
375 std::array<std::unique_ptr<Graph>, 2> bodies_; member in tensorflow::functionalize_cond::Conditional450 .Finalize(bodies_[branch_index].get(), in BuildArgumentNodes()470 bodies_[branch_index]->AddEdge(src_copy, 0, dst_copy, dst_input); in BuildArgumentNodes()532 bodies_[static_cast<int>(b)] = in ExtractBodies()568 auto output = bodies_[branch_index].get(); in ExtractBodies()723 auto output = bodies_[branch_index].get(); in ExtractBodies()735 auto output = bodies_[branch_index].get(); in ExtractBodies()766 *bodies_[branch_index], nullptr); in BuildIfNode()769 TF_RETURN_IF_ERROR(GraphToFunctionDef(*bodies_[branch_index], in BuildIfNode()935 auto output = bodies_[branch_index].get(); in BuildAndReplace()