Home
last modified time | relevance | path

Searched refs:associated_function (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc98 const string& func_name, const AssociatedFunctionInfo& associated_function, in AddFunctionDefToGraphLibrary() argument
154 for (auto& associated_function : associated_functions) { in FunctionalizeControlFlowForNodeAssociatedFunctions() local
162 DCHECK(associated_function.type() != in FunctionalizeControlFlowForNodeAssociatedFunctions()
167 string func_name = associated_function.func_name(); in FunctionalizeControlFlowForNodeAssociatedFunctions()
169 Canonicalize(func_name, AttrSlice(&associated_function.attrs())); in FunctionalizeControlFlowForNodeAssociatedFunctions()
177 graph, n, fld, associated_function, new_func_name)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
184 GetNewFunctionName(func_name, n, associated_function.type(), fld); in FunctionalizeControlFlowForNodeAssociatedFunctions()
187 func_name, new_func_name, associated_function.attrs(), fld, flr, in FunctionalizeControlFlowForNodeAssociatedFunctions()
194 new_func_name, associated_function, graph, fld)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
196 graph, n, fld, associated_function, new_func_name)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
Dtf2xla_util.cc90 for (const auto& associated_function : in CopyAssociatedFunctions() local
92 switch (associated_function.type()) { in CopyAssociatedFunctions()
95 lookup_fld->Find(associated_function.func_name()); in CopyAssociatedFunctions()
98 "Cannot find function ", associated_function.func_name(), in CopyAssociatedFunctions()
793 const AssociatedFunctionInfo& associated_function, in RewriteAssociatedFunction() argument
795 switch (associated_function.type()) { in RewriteAssociatedFunction()
845 GetNodeAttr(node->attrs(), associated_function.attr_name(), &func)); in RewriteAssociatedFunction()
851 node->ClearAttr(associated_function.attr_name()); in RewriteAssociatedFunction()
853 node->AddAttr(associated_function.attr_name(), func); in RewriteAssociatedFunction()
Dtf2xla_util.h144 const AssociatedFunctionInfo& associated_function,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2063 for (auto& associated_function : associated_functions) { in InferXlaShardingFromNeighbors() local
2066 associated_function.func_name(), in InferXlaShardingFromNeighbors()
2067 AttrSlice(&associated_function.attrs()), &handle)); in InferXlaShardingFromNeighbors()