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
155 for (auto& associated_function : associated_functions) { in FunctionalizeControlFlowForNodeAssociatedFunctions() local
163 DCHECK(associated_function.type() != in FunctionalizeControlFlowForNodeAssociatedFunctions()
168 string func_name = associated_function.func_name(); in FunctionalizeControlFlowForNodeAssociatedFunctions()
170 Canonicalize(func_name, AttrSlice(&associated_function.attrs())); in FunctionalizeControlFlowForNodeAssociatedFunctions()
178 graph, n, fld, associated_function, new_func_name)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
185 GetNewFunctionName(func_name, n, associated_function.type(), fld); in FunctionalizeControlFlowForNodeAssociatedFunctions()
188 func_name, new_func_name, associated_function.attrs(), fld, flr, in FunctionalizeControlFlowForNodeAssociatedFunctions()
195 new_func_name, associated_function, graph, fld)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
197 graph, n, fld, associated_function, new_func_name)); in FunctionalizeControlFlowForNodeAssociatedFunctions()
Dtf2xla_util.cc86 for (const auto& associated_function : in CopyAssociatedFunctions() local
88 switch (associated_function.type()) { in CopyAssociatedFunctions()
91 lookup_fld->Find(associated_function.func_name()); in CopyAssociatedFunctions()
94 "Cannot find function ", associated_function.func_name(), in CopyAssociatedFunctions()
622 const AssociatedFunctionInfo& associated_function, in RewriteAssociatedFunction() argument
624 switch (associated_function.type()) { in RewriteAssociatedFunction()
676 GetNodeAttr(node->attrs(), associated_function.attr_name(), &func)); in RewriteAssociatedFunction()
677 node->ClearAttr(associated_function.attr_name()); in RewriteAssociatedFunction()
679 node->AddAttr(associated_function.attr_name(), func); in RewriteAssociatedFunction()
Dtf2xla_util.h143 const AssociatedFunctionInfo& associated_function,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1850 for (auto& associated_function : associated_functions) { in InferXlaShardingFromNeighbors() local
1853 associated_function.func_name(), in InferXlaShardingFromNeighbors()
1854 AttrSlice(&associated_function.attrs()), &handle)); in InferXlaShardingFromNeighbors()