Home
last modified time | relevance | path

Searched refs:device_manager (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_id_utils.h35 se::Platform* device_manager, PlatformDeviceId platform_device_id) { in ExecutorForPlatformDeviceId() argument
36 return device_manager->ExecutorForDevice(platform_device_id.value()); in ExecutorForPlatformDeviceId()
39 const DeviceType& type, se::Platform* device_manager, in ExecutorForTfDeviceId() argument
44 return ExecutorForPlatformDeviceId(device_manager, platform_device_id); in ExecutorForTfDeviceId()
50 se::Platform* device_manager, in CheckValidTfDeviceId() argument
55 const int visible_device_count = device_manager->VisibleDeviceCount(); in CheckValidTfDeviceId()
/external/autotest/client/common_lib/cros/cfm/atrus/
Datrus_utils.py55 device_manager = usb_device_collector.UsbDeviceCollector()
57 lambda: len(device_manager.get_devices_by_spec(ATRUS)) >= 1,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables.cc88 const DeviceMgr* device_manager; in LiftVariablesFromSession() local
89 if (!(session->LocalDeviceManager(&device_manager).ok())) { in LiftVariablesFromSession()
106 if (!(device_manager->LookupDevice(resource_handle.device(), &device) in LiftVariablesFromSession()