Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.h42 absl::optional<std::string> encapsulating_function_name; member
H A Dhost_training_loop_optimization_util.cc259 host_training_loop_info.encapsulating_function_name = in FindTPUCompileNodes()
H A Ddistributed_tpu_rewrite_pass.cc4894 const auto& function_name = host_loop.encapsulating_function_name; in PerformHostTrainingLoopOptimization()