Searched refs:encapsulating_function_attrs (Results 1 – 3 of 3) sorted by relevance
43 absl::optional<AttrValueMap> encapsulating_function_attrs; member
262 host_training_loop_info.encapsulating_function_attrs = in FindTPUCompileNodes()
4395 const auto& function_attr = host_loop.encapsulating_function_attrs; in PerformHostTrainingLoopOptimization()