Searched refs:compile_node_name (Results 1 – 2 of 2) sorted by relevance
244 host_training_loop_info.compile_node_name = n->name(); in FindTPUCompileNodes()441 const auto& compile_node_name = host_loop_info.compile_node_name; in AddReshardOp() local443 const auto node_it = node_name_map.find(compile_node_name); in AddReshardOp()445 << "Unable to find compile node : " << compile_node_name; in AddReshardOp()
46 std::string compile_node_name; member