Searched refs:DeviceNode (Results 1 – 1 of 1) sorted by relevance
449 DeviceNode* node = &index_nodes_.at(device); in ReportDeviceAssigned()457 DeviceNode* node = heap_.top(); in RetrieveAssignment()465 struct DeviceNode { struct in tensorflow::__anond2c82b040111::TensorDevicePlacer469 bool operator()(const DeviceNode* lhs, const DeviceNode* rhs) const { in operator ()()485 std::vector<DeviceNode> index_nodes_;486 IntrusiveHeap<DeviceNode, typename DeviceNode::Compare> heap_;