Home
last modified time | relevance | path

Searched refs:host_training_loops_info (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc224 std::vector<HostTrainingLoopInfo>* host_training_loops_info) { in FindTPUCompileNodes() argument
266 host_training_loops_info->emplace_back( in FindTPUCompileNodes()
393 std::vector<HostTrainingLoopInfo>* host_training_loops_info) { in DetectHostTrainingLoop() argument
422 host_training_loops_info)); in DetectHostTrainingLoop()
436 host_training_loops_info); in DetectHostTrainingLoop()
Dhost_training_loop_optimization_util.h70 std::vector<HostTrainingLoopInfo>* host_training_loops_info);
Ddistributed_tpu_rewrite_pass.cc4377 std::vector<tpu::HostTrainingLoopInfo> host_training_loops_info; in PerformHostTrainingLoopOptimization() local
4381 &host_training_loops_info); in PerformHostTrainingLoopOptimization()
4388 for (const auto& host_loop : host_training_loops_info) { in PerformHostTrainingLoopOptimization()