Home
last modified time | relevance | path

Searched refs:LocalDeviceManager (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc223 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()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_initialized_variables.cc57 auto status = session->LocalDeviceManager(&mgr); in MarkInitializedVariablesInFunction()
Dlift_variables_test_pass.h61 Status LocalDeviceManager( in LocalDeviceManager() function
Dlift_variables.cc88 if (!(session->LocalDeviceManager(&device_manager).ok())) { in LiftVariablesFromSession()
Dinitialize_variables_in_session_init.cc131 auto status = session_->LocalDeviceManager(&mgr); in runOnOperation()
Dtf_saved_model_freeze_variables.cc321 auto status = session_->LocalDeviceManager(&mgr); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dfake_session.h40 ::tensorflow::Status LocalDeviceManager(
Dfake_session.cc101 Status FakeSession::LocalDeviceManager( in LocalDeviceManager() function in mlir::TF::test_util::FakeSession
/external/tensorflow/tensorflow/core/public/
Dsession.h229 virtual Status LocalDeviceManager(const DeviceMgr** output) { in LocalDeviceManager() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc120 TF_ASSERT_OK(session->LocalDeviceManager(&device_mgr)); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.h112 ::tensorflow::Status LocalDeviceManager(const DeviceMgr** output) override { in LocalDeviceManager() function
Ddirect_session_test.cc2293 TEST(DirectSessionTest, LocalDeviceManager) { in TEST() argument
2298 TF_ASSERT_OK(session->LocalDeviceManager(&mgr)); in TEST()