Searched refs:bodies_ (Results 1 – 1 of 1) sorted by relevance
362 std::array<std::unique_ptr<Graph>, 2> bodies_; member in tensorflow::functionalize_cond::Conditional434 .Finalize(bodies_[branch_index].get(), in BuildArgumentNodes()454 bodies_[branch_index]->AddEdge(src_copy, 0, dst_copy, dst_input); in BuildArgumentNodes()516 bodies_[static_cast<int>(b)] = in ExtractBodies()552 auto output = bodies_[branch_index].get(); in ExtractBodies()695 auto output = bodies_[branch_index].get(); in ExtractBodies()707 auto output = bodies_[branch_index].get(); in ExtractBodies()740 *bodies_[branch_index], nullptr); in BuildIfNode()743 TF_RETURN_IF_ERROR(GraphToFunctionDef(*bodies_[branch_index], in BuildIfNode()894 auto output = bodies_[branch_index].get(); in BuildAndReplace()