Searched refs:outside_compilation_attr (Results 1 – 5 of 5) sorted by relevance
128 auto outside_compilation_attr = in UpdateEmbeddingEnqueueOpInput() local130 if (outside_compilation_attr) in UpdateEmbeddingEnqueueOpInput()132 outside_compilation_attr); in UpdateEmbeddingEnqueueOpInput()
159 string outside_compilation_attr; in PreprocessDataEdgesBetweenOutsideCompilations() local162 &outside_compilation_attr)); in PreprocessDataEdgesBetweenOutsideCompilations()164 outside_compilation_attr); in PreprocessDataEdgesBetweenOutsideCompilations()
923 string outside_compilation_attr; in OutsideCompilationAttrToNode() local926 &outside_compilation_attr)) { in OutsideCompilationAttrToNode()929 outside_compilation_attr_to_node[outside_compilation_attr] = n; in OutsideCompilationAttrToNode()
689 string outside_compilation_attr; in MoveHeadOutsideCompilationToHost() local692 &outside_compilation_attr)) { in MoveHeadOutsideCompilationToHost()703 ph_builder.Attr(kOutsideCompilationAttr, outside_compilation_attr); in MoveHeadOutsideCompilationToHost()732 outside_compilation_attr); in MoveHeadOutsideCompilationToHost()1337 string outside_compilation_attr; in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local1339 &outside_compilation_attr)); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()1340 ph_builder.Attr(kOutsideCompilationAttr, outside_compilation_attr); in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
351 const AttrValue* outside_compilation_attr = in FindTaggedNodes() local354 if (outside_compilation_attr != nullptr) { in FindTaggedNodes()371 if (outside_compilation_attr == nullptr) { in FindTaggedNodes()376 const string& oc_cluster = outside_compilation_attr->s(); in FindTaggedNodes()3751 string outside_compilation_attr; in LowerOutsideCompilationFunctionalNodes() local3753 &outside_compilation_attr)); in LowerOutsideCompilationFunctionalNodes()3815 node->AddAttr(kOutsideCompilationAttr, outside_compilation_attr); in LowerOutsideCompilationFunctionalNodes()