Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc147 auto associated_functions = GetAssociatedFunctions(*n, fld); in FunctionalizeControlFlowForNodeAssociatedFunctions() local
154 auto associated_functions = pair.second; in FunctionalizeControlFlowForNodeAssociatedFunctions() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc396 const auto associated_functions = GetAssociatedFunctions(*n, library); in DetectHostTrainingLoop() local
Ddistributed_tpu_rewrite_pass.cc1848 auto associated_functions = GetAssociatedFunctions( in InferXlaShardingFromNeighbors() local