Searched refs:LocalDeviceManager (Results 1 – 6 of 6) sorted by relevance
225 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in GetPeakMemoryUsage()374 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ResetSession()454 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ClearAllocatorStats()
194 virtual Status LocalDeviceManager(const DeviceMgr** output) { in LocalDeviceManager() function
193 TF_ASSERT_OK(session->LocalDeviceManager(&device_mgr)); in TEST()
102 ::tensorflow::Status LocalDeviceManager(const DeviceMgr** output) override { in LocalDeviceManager() function
1863 TEST(DirectSessionTest, LocalDeviceManager) { in TEST() argument1868 TF_ASSERT_OK(session->LocalDeviceManager(&mgr)); in TEST()
356 status->status = sess->session->LocalDeviceManager(&device_mgr); in TFE_NewContextFromSession()