Please enable JavaScript.
Coggle requires JavaScript to display documents.
DeviceBase (functions (explicit DeviceBase(Env* env), virtual…
DeviceBase
functions
explicit DeviceBase(Env* env)
virtual ~DeviceBase()
Env* env() const
virtual bool RequiresRecordingAccessedTensors() const
void set_tensorflow_cpu_worker_threads(CpuWorkerThreads* t)
virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const
void set_tensorflow_gpu_device_info(GpuDeviceInfo* g)
virtual const GpuDeviceInfo* tensorflow_gpu_device_info() const
virtual thread::ThreadPool* tensorflow_device_thread_pool()
void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d)
void set_eigen_sycl_device(Eigen::SyclDevice* d)
virtual Allocator
GetAllocator(AllocatorAttributes /
attr*/)
Allocator
GetStepAllocator(AllocatorAttributes attr, ResourceMgr
)
Member Variables
Env* const env_
CpuWorkerThreads* cpu_worker
threads
GpuDeviceInfo* gpu_device
info
thread::ThreadPool* device_thread
pool
std::vector<Eigen::ThreadPoolDevice*> eigen_cpu
devices
Eigen::SyclDevice* eigen_sycl
device
Nested Structures
CpuWorkerThreads
int num_threads
thread::ThreadPool* workers
GpuDeviceInfo
stream_executor::Stream* stream
DeviceContext* default_context
EventMgr* event_mgr
int gpu_id
tensorflow\core\framework\device_base.h