Searched defs:GpuDevice (Results 1 – 5 of 5) sorted by relevance
183 struct GpuDevice { in initializeDeviceProp() struct186 explicit GpuDevice(const StreamInterface* stream) : stream_(stream), max_blocks_(INT_MAX) { in initializeDeviceProp() function189 …explicit GpuDevice(const StreamInterface* stream, int num_blocks) : stream_(stream), max_blocks_(n… in initializeDeviceProp() function193 EIGEN_STRONG_INLINE const cudaStream_t& stream() const { in initializeDeviceProp()197 EIGEN_STRONG_INLINE void* allocate(size_t num_bytes) const { in initializeDeviceProp()201 EIGEN_STRONG_INLINE void deallocate(void* buffer) const { in initializeDeviceProp()205 EIGEN_STRONG_INLINE void* scratchpad() const { in initializeDeviceProp()209 EIGEN_STRONG_INLINE unsigned int* semaphore() const { in initializeDeviceProp()213 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memcpy(void* dst, const void* src, size_t n) const { in initializeDeviceProp()224 EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n) const { in initializeDeviceProp()[all …]
25 using GpuDevice = Eigen::GpuDevice; typedef
150 using GpuDevice = Eigen::GpuDevice; typedef
307 GpuDevice::GpuDevice(int id, in GpuDevice() function in xla::GpuDevice
40 constexpr char GpuDevice[] = "/device:GPU:0"; variable