Home
last modified time | relevance | path

Searched refs:MasterSession (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc65 class MasterSession::ReffedClientGraph : public core::RefCounted {
331 Status MasterSession::ReffedClientGraph::RegisterPartitions( in RegisterPartitions()
382 void MasterSession::ReffedClientGraph::TrackFeedsAndFetches( in TrackFeedsAndFetches()
420 Status MasterSession::ReffedClientGraph::DoBuildPartitions( in DoBuildPartitions()
436 Status MasterSession::ReffedClientGraph::DoRegisterPartitions( in DoRegisterPartitions()
591 Status MasterSession::ReffedClientGraph::RunPartitionsHelper( in RunPartitionsHelper()
750 Status MasterSession::ReffedClientGraph::RunPartitions( in RunPartitions()
775 Status MasterSession::ReffedClientGraph::RunPartitions( in RunPartitions()
869 void MasterSession::ReffedClientGraph::CleanupPartitionsAsync( in CleanupPartitionsAsync()
883 void MasterSession::ReffedClientGraph::ProcessStats(int64 step_id, in ProcessStats()
[all …]
Dmaster_session.h42 class MasterSession : public core::RefCounted {
50 MasterSession(
214 virtual ~MasterSession();
230 void FillPerStepState(MasterSession::ReffedClientGraph* rcg,
242 Status PostRunCleanup(MasterSession::ReffedClientGraph* rcg, uint64 step_id,
258 TF_DISALLOW_COPY_AND_ASSIGN(MasterSession);
Dmaster_env.h33 class MasterSession; variable
83 std::function<MasterSession*(
Dmaster.h87 std::unordered_map<string, MasterSession*> sessions_ GUARDED_BY(mu_);
110 MasterSession* FindMasterSession(const string& handle);
Dmaster.cc121 MasterSession* Master::FindMasterSession(const string& handle) { in FindMasterSession()
122 MasterSession* session = nullptr; in FindMasterSession()
472 MasterSession* session = env_->master_session_factory( in CreateSession()
561 MasterSession* session = nullptr; in CloseSession()
655 std::vector<MasterSession*> sessions_to_close; in Reset()
670 for (MasterSession* session : sessions_to_close) { in Reset()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc264 return new MasterSession(options, env, std::move(remote_devs), in Init()
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto263 // EXPERIMENTAL: This currently has no effect in MasterSession.