Searched defs:device_manager (Results 1 – 7 of 7) sorted by relevance
37 se::Platform* device_manager, PlatformDeviceId platform_device_id) { in ExecutorForPlatformDeviceId()41 const DeviceType& type, se::Platform* device_manager, in ExecutorForTfDeviceId()52 se::Platform* device_manager, in CheckValidTfDeviceId()
67 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()107 const tensorflow::DeviceMgr* device_manager, int64_t step_id, in KernelFallbackCompatRequestState()
94 const tensorflow::DeviceMgr& device_manager() const { in device_manager() function
93 const tensorflow::DeviceMgr* device_manager, in SetUpKernelFallbackCompatRequestContext()
53 const DeviceMgr &device_manager() const { return device_manager_; } in device_manager() function
87 const DeviceMgr* device_manager; in LiftVariablesFromSession() local
288 const auto* device_manager = &src_fallback_request_state->device_manager(); in SetUpKernelFallbackCompatRequestContextForBatch() local