Method XrtContext

tensorflow/compiler/xrt/client/xrt_client.cc:498–509

}

XrtContext::XrtContext(std::shared_ptr<XrtTfContext> tf_context,
                       string device_type)
    : tf_context_(std::move(tf_context)), device_type_(std::move(device_type)) {
  for (int i = 0; i < tf_context_->devices().size(); ++i) {
    const DeviceAttributes& device = tf_context_->devices()[i];
    VLOG(2) << "Device: " << i << ": " << device.DebugString();
    if (device.device_type() == device_type_) {
      tf_device_ids_.push_back(i);
      VLOG(1) << "Accelerator device " << i << ": " << device.name();
    }
  }
}

int XrtContext::device_count() const { return tf_device_ids_.size(); }