Searched defs:device_ptr (Results 1 – 9 of 9) sorted by relevance
32 explicit HostOrDeviceScalar(DeviceMemory<T> device_ptr) in HostOrDeviceScalar()
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()
263 let device_ptr = self._device.upgrade().ok_or(DeviceDestroyedError)?; in device_fd() localVariable
949 se::DeviceMemoryBase device_ptr( in CopyTensorToHost() local
779 CFX_DefaultRenderDevice* device_ptr = device.get(); in FPDFTextObj_GetRenderedBitmap() local
1615 Device* device_ptr = device_.get(); in CreateContext() local