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()
1336 NodeDef replace_def; in ExtractOutsideCompilationForNodesWithAssociatedFunctions() local1337 TF_RETURN_IF_ERROR(replace_builder.Finalize(&replace_def)); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()1338 TF_ASSIGN_OR_RETURN(Node * replace, ReplaceNode(g, n, replace_def)); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()