Searched refs:index_nodes_ (Results 1 – 1 of 1) sorted by relevance
420 : index_nodes_(num_devices), sizes_(types.size()) { in TensorDevicePlacer()442 heap_.Push(&index_nodes_[i]); in TensorDevicePlacer()449 DeviceNode* node = &index_nodes_.at(device); in ReportDeviceAssigned()458 int64 device = node - index_nodes_.data(); in RetrieveAssignment()485 std::vector<DeviceNode> index_nodes_; member in tensorflow::__anond2c82b040111::TensorDevicePlacer