Searched refs:host_training_loop_info (Results 1 – 1 of 1) sorted by relevance
243 HostTrainingLoopInfo host_training_loop_info; in FindTPUCompileNodes() local244 host_training_loop_info.compile_node_name = n->name(); in FindTPUCompileNodes()245 host_training_loop_info.loop_cond_node_name = frame->loop_cond->name(); in FindTPUCompileNodes()246 host_training_loop_info.while_loop_name = frame->name; in FindTPUCompileNodes()253 host_training_loop_info.loop_arguments.push_back(std::move(arg_info)); in FindTPUCompileNodes()255 host_training_loop_info.loop_nodes = frame->nodes; in FindTPUCompileNodes()258 host_training_loop_info.encapsulating_function_name = in FindTPUCompileNodes()262 host_training_loop_info.encapsulating_function_attrs = in FindTPUCompileNodes()267 std::move(host_training_loop_info)); in FindTPUCompileNodes()