Home
last modified time | relevance | path

Searched defs:associated_function (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_control_flow.cc98 const string& func_name, const AssociatedFunctionInfo& associated_function, in AddFunctionDefToGraphLibrary()
154 for (auto& associated_function : associated_functions) { in FunctionalizeControlFlowForNodeAssociatedFunctions() local
H A Dtf2xla_util.cc90 for (const auto& associated_function : in CopyAssociatedFunctions() local
793 const AssociatedFunctionInfo& associated_function, in RewriteAssociatedFunction()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc2063 for (auto& associated_function : associated_functions) { in InferXlaShardingFromNeighbors() local