Searched defs:ThreadPoolDevice (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDeviceThreadPool.h | 107 struct ThreadPoolDevice { struct
109 …ThreadPoolDevice(ThreadPoolInterface* pool, int num_cores) : pool_(pool), num_threads_(num_cores) … in ThreadPoolDevice() argument
111 EIGEN_STRONG_INLINE void* allocate(size_t num_bytes) const { in allocate()
115 EIGEN_STRONG_INLINE void deallocate(void* buffer) const { in deallocate()
119 EIGEN_STRONG_INLINE void memcpy(void* dst, const void* src, size_t n) const { in memcpy()
122 EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n) const { in memcpyHostToDevice()
125 EIGEN_STRONG_INLINE void memcpyDeviceToHost(void* dst, const void* src, size_t n) const { in memcpyDeviceToHost()
129 EIGEN_STRONG_INLINE void memset(void* buffer, int c, size_t n) const { in memset()
133 EIGEN_STRONG_INLINE int numThreads() const { in numThreads()
137 EIGEN_STRONG_INLINE size_t firstLevelCacheSize() const { in firstLevelCacheSize()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device.cc | 44 ThreadPoolDevice::ThreadPoolDevice(const SessionOptions& options, in ThreadPoolDevice() function in tensorflow::ThreadPoolDevice
|