Method VirtualDeviceCount

tensorflow/stream_executor/cuda/cuda_platform.cc:133–139

Platform::Id CudaPlatform::id() const { return cuda::kCudaPlatformId; }

int CudaPlatform::VirtualDeviceCount() const {
  if (virtual_device_count_ < 0) {
    return VisibleDeviceCount();
  } else {
    return virtual_device_count_;
  }
}

port::Status CudaPlatform::SetVirtualDeviceCount(int count) {
  virtual_device_count_ = count;