Searched refs:associated_function (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 144 for (auto& associated_function : associated_functions) { in FunctionalizeControlFlowForFunction() local
145 string name = associated_function.func_name(); in FunctionalizeControlFlowForFunction()
147 Canonicalize(name, AttrSlice(&associated_function.attrs())); in FunctionalizeControlFlowForFunction()
160 if (associated_function.type() == in FunctionalizeControlFlowForFunction()
169 name, new_name, associated_function.attrs(), fld, flr, in FunctionalizeControlFlowForFunction()
191 g, n, fld, associated_function, new_name)); in FunctionalizeControlFlowForFunction()
|
| D | tf2xla_util.cc | 86 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()
623 const AssociatedFunctionInfo& associated_function, in RewriteAssociatedFunction() argument
625 switch (associated_function.type()) { in RewriteAssociatedFunction()
677 GetNodeAttr(node->attrs(), associated_function.attr_name(), &func)); in RewriteAssociatedFunction()
678 node->ClearAttr(associated_function.attr_name()); in RewriteAssociatedFunction()
680 node->AddAttr(associated_function.attr_name(), func); in RewriteAssociatedFunction()
|
| D | tf2xla_util.h | 142 const AssociatedFunctionInfo& associated_function,
|