Function GetDeviceNamePrefix

tensorflow/core/graph/stream_subgraph.cc:34–44

}

std::string GetDeviceNamePrefix(const std::string& device_name) {
  std::string prefix_token("device:GPU:");
  auto idx = device_name.find(prefix_token);
  if (idx == std::string::npos) {
    LOG(FATAL) << "Error device name, " << device_name;
  }
  std::string device_name_prefix =
      device_name.substr(0, idx + prefix_token.length());

  return device_name_prefix;
}

} // namesapce