Searched refs:host_config_def (Results 1 – 1 of 1) sorted by relevance
81 NodeDef host_config_def; in AddHostConfigNode() local82 host_config_def.set_name(graph->NewName("configure_tpu_host")); in AddHostConfigNode()83 host_config_def.set_op(kHostConfigureOp); in AddHostConfigNode()84 host_config_def.set_device(host_device_name); in AddHostConfigNode()86 &host_config_def); in AddHostConfigNode()87 MergeDebugInfo(NodeDebugInfo(configuration_node->def()), &host_config_def); in AddHostConfigNode()90 *host_configuration_node = graph->AddNode(host_config_def, &status); in AddHostConfigNode()