Searched refs:LocalDeviceManager (Results 1 – 12 of 12) sorted by relevance
223 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in GetPeakMemoryUsage()372 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ResetSession()452 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ClearAllocatorStats()
57 auto status = session->LocalDeviceManager(&mgr); in MarkInitializedVariablesInFunction()
61 Status LocalDeviceManager( in LocalDeviceManager() function
88 if (!(session->LocalDeviceManager(&device_manager).ok())) { in LiftVariablesFromSession()
131 auto status = session_->LocalDeviceManager(&mgr); in runOnOperation()
321 auto status = session_->LocalDeviceManager(&mgr); in runOnOperation()
40 ::tensorflow::Status LocalDeviceManager(
101 Status FakeSession::LocalDeviceManager( in LocalDeviceManager() function in mlir::TF::test_util::FakeSession
229 virtual Status LocalDeviceManager(const DeviceMgr** output) { in LocalDeviceManager() function
120 TF_ASSERT_OK(session->LocalDeviceManager(&device_mgr)); in TEST()
112 ::tensorflow::Status LocalDeviceManager(const DeviceMgr** output) override { in LocalDeviceManager() function
2293 TEST(DirectSessionTest, LocalDeviceManager) { in TEST() argument2298 TF_ASSERT_OK(session->LocalDeviceManager(&mgr)); in TEST()