Searched refs:replace_def (Results 1 – 2 of 2) sorted by relevance
149 NodeDef replace_def = usage_node->def(); in ReplaceArgUsageWithConstNode() local150 *replace_def.mutable_input(usages[i].dst_input) = const_node->name(); in ReplaceArgUsageWithConstNode()152 ReplaceNode(g, usage_node, replace_def)); in ReplaceArgUsageWithConstNode()
1871 auto replace_def = absl::make_unique<NodeDef>(); in ExtractOutsideCompilationForFuncCallNode() local1872 TF_RETURN_IF_ERROR(replace_builder->Finalize(replace_def.get())); in ExtractOutsideCompilationForFuncCallNode()1873 TF_ASSIGN_OR_RETURN(Node * replace, ReplaceNode(g, n, *replace_def)); in ExtractOutsideCompilationForFuncCallNode()