Searched defs:device_ptr (Results 1 – 10 of 10) sorted by relevance
32 explicit HostOrDeviceScalar(DeviceMemory<T> device_ptr) in HostOrDeviceScalar()
132 hw_device_t** device_ptr) { in testing_nvram_open()
28 using device_ptr = thrust::device_ptr<int64_t>; in index_put_with_sort_kernel_thrust_helper() typedef
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
263 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