Home
last modified time | relevance | path

Searched refs:GpuDeviceHandle (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h71 static port::StatusOr<GpuDeviceHandle> DeviceFromContext(GpuContext* context);
150 static port::Status GetDevice(int device_ordinal, GpuDeviceHandle* device);
154 static bool GetDeviceName(GpuDeviceHandle device, string* device_name);
163 static port::Status CreateContext(int device_ordinal, GpuDeviceHandle device,
382 static port::StatusOr<GpuDeviceHandle> GetPointerDevice(GpuDevicePtr pointer);
399 GpuDeviceHandle device);
403 static port::Status GetGpuISAVersion(int* version, GpuDeviceHandle device);
407 static port::StatusOr<int> GetMultiprocessorCount(GpuDeviceHandle device);
412 GpuDeviceHandle device);
416 static port::StatusOr<int64> GetMaxThreadsPerBlock(GpuDeviceHandle device);
[all …]
Dgpu_types.h51 using GpuDeviceHandle = hipDevice_t; variable
69 using GpuDeviceHandle = CUdevice;
Dgpu_executor.h320 GpuDeviceHandle device_;