Home
last modified time | relevance | path

Searched refs:PerformHostTrainingLoopOptimization (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.h578 static Status PerformHostTrainingLoopOptimization(
H A Ddistributed_tpu_rewrite_pass.cc4879 DistributedTPURewritePass::PerformHostTrainingLoopOptimization( in PerformHostTrainingLoopOptimization() function in tensorflow::DistributedTPURewritePass
5050 PerformHostTrainingLoopOptimization(graph, options.flib_def, flr)); in InternalRun()