Searched defs:GpuDevice (Results 1 – 10 of 10) sorted by relevance
207 struct GpuDevice { struct210 explicit GpuDevice(const StreamInterface* stream) : stream_(stream), max_blocks_(INT_MAX) { in GpuDevice() argument213 …explicit GpuDevice(const StreamInterface* stream, int num_blocks) : stream_(stream), max_blocks_(n… in GpuDevice() function217 EIGEN_STRONG_INLINE const gpuStream_t& stream() const { in stream()221 EIGEN_STRONG_INLINE void* allocate(size_t num_bytes) const { in allocate()225 EIGEN_STRONG_INLINE void deallocate(void* buffer) const { in deallocate()229 EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const { in allocate_temp()233 EIGEN_STRONG_INLINE void deallocate_temp(void* buffer) const { in deallocate_temp()238 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Type get(Type data) const { in get()242 EIGEN_STRONG_INLINE void* scratchpad() const { in scratchpad()[all …]
25 using GpuDevice = Eigen::GpuDevice; typedef
150 using GpuDevice = Eigen::GpuDevice; typedef
37 using GpuDevice = ::Eigen::GpuDevice; typedef
404 GpuDevice::GpuDevice(int id, in GpuDevice() function in xla::GpuDevice
40 constexpr char GpuDevice[] = "/device:GPU:0"; variable
12168 "GpuDevice": [ array
12209 "GpuDevice": [ array
25738 "GpuDevice": [ array
74170 "GpuDevice": [ array