Home
last modified time | relevance | path

Searched refs:PluggableDeviceProcessState (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_process_state.cc45 /*static*/ PluggableDeviceProcessState* PluggableDeviceProcessState::singleton( in singleton()
48 std::unordered_map<string, PluggableDeviceProcessState*>; in singleton()
55 new PluggableDeviceProcessState(device_type, platform_name); in singleton()
59 PluggableDeviceProcessState::PluggableDeviceProcessState( in PluggableDeviceProcessState() function in tensorflow::PluggableDeviceProcessState
67 int PluggableDeviceProcessState::BusIdForPluggableDevice( in BusIdForPluggableDevice()
79 Allocator* PluggableDeviceProcessState::GetPluggableDeviceAllocator( in GetPluggableDeviceAllocator()
140 Allocator* PluggableDeviceProcessState::GetPluggableDeviceHostAllocator( in GetPluggableDeviceHostAllocator()
H A Dpluggable_device_process_state.h40 class PluggableDeviceProcessState {
44 static PluggableDeviceProcessState* singleton(const string& device_type,
90 PluggableDeviceProcessState(const string& device_type,
92 virtual ~PluggableDeviceProcessState() {} in ~PluggableDeviceProcessState()
99 static PluggableDeviceProcessState* instance_;
H A Dpluggable_device.cc156 PluggableDeviceProcessState::singleton(device_type, platform_name) in PluggableDevice()
247 PluggableDeviceProcessState* ps = in GetAllocator()
248 PluggableDeviceProcessState::singleton(device_type(), platform_name_); in GetAllocator()
H A Dpluggable_device_factory.cc247 PluggableDeviceProcessState* process_state = in CreatePluggableDevice()
248 PluggableDeviceProcessState::singleton(device_type_, platform_name_); in CreatePluggableDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_state.h84 friend class PluggableDeviceProcessState; variable