Home
last modified time | relevance | path

Searched defs:host_training_loops_info (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc225 std::vector<HostTrainingLoopInfo>* host_training_loops_info) { in FindTPUCompileNodes()
406 std::vector<HostTrainingLoopInfo>* host_training_loops_info) { in DetectHostTrainingLoop()
H A Ddistributed_tpu_rewrite_pass.cc4882 std::vector<tpu::HostTrainingLoopInfo> host_training_loops_info; in PerformHostTrainingLoopOptimization() local