Home
last modified time | relevance | path

Searched refs:encapsulating_function_attrs (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.h43 absl::optional<AttrValueMap> encapsulating_function_attrs; member
H A Dhost_training_loop_optimization_util.cc263 host_training_loop_info.encapsulating_function_attrs = in FindTPUCompileNodes()
H A Ddistributed_tpu_rewrite_pass.cc4900 const auto& function_attr = host_loop.encapsulating_function_attrs; in PerformHostTrainingLoopOptimization()