Searched defs:ThreadPoolDevice (Results 1 – 4 of 4) sorted by relevance
53 struct ThreadPoolDevice { struct56 : pool_(pool), num_threads_(num_cores), allocator_(allocator) { } in pool_()58 EIGEN_STRONG_INLINE void* allocate(size_t num_bytes) const { in allocate()63 EIGEN_STRONG_INLINE void deallocate(void* buffer) const { in deallocate()71 EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const { in allocate_temp()75 EIGEN_STRONG_INLINE void deallocate_temp(void* buffer) const { in deallocate_temp()80 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Type get(Type data) const { in get()84 EIGEN_STRONG_INLINE void memcpy(void* dst, const void* src, size_t n) const { in memcpy()114 EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n) const { in memcpyHostToDevice()117 EIGEN_STRONG_INLINE void memcpyDeviceToHost(void* dst, const void* src, size_t n) const { in memcpyDeviceToHost()[all …]
31 class ThreadPoolDevice; variable
76 llvm::Optional<Eigen::ThreadPoolDevice>) { in EigenSoftmax()
56 ThreadPoolDevice::ThreadPoolDevice(const SessionOptions& options, in ThreadPoolDevice() function in tensorflow::ThreadPoolDevice