Searched refs:LocalDeviceManager (Results 1 – 13 of 13) sorted by relevance
224 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in GetPeakMemoryUsage()373 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ResetSession()453 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ClearAllocatorStats()
40 ::tensorflow::Status LocalDeviceManager(
101 Status FakeSession::LocalDeviceManager( in LocalDeviceManager() function in mlir::TF::test_util::FakeSession
57 auto status = session->LocalDeviceManager(&mgr); in MarkInitializedVariablesInFunction()
61 Status LocalDeviceManager( in LocalDeviceManager() function
107 auto status = session->LocalDeviceManager(&mgr); in InitializeVariablesInSessionInitializer()
89 if (!(session->LocalDeviceManager(&device_manager).ok())) { in LiftVariablesFromSession()
310 auto status = session->LocalDeviceManager(&mgr); in FreezeVariables()
229 virtual Status LocalDeviceManager(const DeviceMgr** output) { in LocalDeviceManager() function
109 TF_ASSERT_OK(session->LocalDeviceManager(&device_mgr)); in TEST()
112 ::tensorflow::Status LocalDeviceManager(const DeviceMgr** output) override { in LocalDeviceManager() function
2319 TEST(DirectSessionTest, LocalDeviceManager) { in TEST() argument2324 TF_ASSERT_OK(session->LocalDeviceManager(&mgr)); in TEST()
257 TF_RETURN_IF_ERROR(session->LocalDeviceManager(&device_mgr)); in GetResourceManager()