Searched defs:device_ptr (Results 1 – 8 of 8) sorted by relevance
32 explicit HostOrDeviceScalar(DeviceMemory<T> device_ptr) in HostOrDeviceScalar()
219 void* device_ptr, const Shape& shape, PjRtDevice* device, in CreateViewOfDeviceBuffer()
464 void* device_ptr, const Shape& shape, PjRtDevice* device, in CreateViewOfDeviceBuffer()
1075 void* device_ptr, const Shape& shape, PjRtDevice* device, in CreateViewOfDeviceBuffer()
640 CFX_DefaultRenderDevice* device_ptr = device.get(); in FPDFTextObj_GetRenderedBitmap() local
261 let device_ptr = self._device.upgrade().ok_or(DeviceDestroyedError)?; in device_fd() localVariable
949 se::DeviceMemoryBase device_ptr( in CopyTensorToHost() local
1615 Device* device_ptr = device_.get(); in CreateContext() local