tensorflow::NTSessionFactory Class Reference

Inheritance diagram for tensorflow::NTSessionFactory:

Public Member Functions
bool	AcceptsOptions (const SessionOptions &options) override
void	Deregister (const NTSession *session)
Session *	NewSession (const SessionOptions &options) override
	NTSessionFactory ()
Status	Reset (const SessionOptions &options, const std::vector< string > &containers) override

Private Member Functions
std::vector< NTSession * > sessions_	GUARDED_BY (sessions_lock_)

Private Attributes
mutex	sessions_lock_

Detailed Description

Definition at line 103 of file NTSession.cc.

Constructor & Destructor Documentation

tensorflow::NTSessionFactory::NTSessionFactory ( )

inline

Definition at line 105 of file NTSession.cc.

Referenced by tensorflow::NTSessionRegistrar::NTSessionRegistrar().

{}

Member Function Documentation

bool tensorflow::NTSessionFactory::AcceptsOptions ( const SessionOptions &  options )

inlineoverride

Definition at line 107 of file NTSession.cc.

                                                              {
    return options.target == "no_threads";
  }

void tensorflow::NTSessionFactory::Deregister ( const NTSession *  session )

inline

Definition at line 155 of file NTSession.cc.

References checklumidiff::l, MatrixUtil::remove(), dataDML::session, and sessions_lock_.

Referenced by tensorflow::NTSession::Close().

                                            {
    mutex_lock l(sessions_lock_);
    sessions_.erase(std::remove(sessions_.begin(), sessions_.end(), session),
                    sessions_.end());
  }

std::vector<NTSession*> sessions_ tensorflow::NTSessionFactory::GUARDED_BY ( sessions_lock_  )

private

Session* tensorflow::NTSessionFactory::NewSession ( const SessionOptions &  options )

inlineoverride

Definition at line 111 of file NTSession.cc.

References dqm::qstatus::ERROR, checklumidiff::l, LOG, alignCSCRings::s, dataDML::session, sessions_lock_, and btagGenBb_cfi::Status.

                                                              {
    // Must do this before the CPU allocator is created.
    if (options.config.graph_options().build_cost_model() > 0) {
      EnableCPUAllocatorFullStats(true);
    }
    std::vector<Device*> devices;
    const Status s = DeviceFactory::AddDevices(
        options, "/job:localhost/replica:0/task:0", &devices);
    if (!s.ok()) {
      LOG(ERROR) << s;
      return nullptr;
    }

    NTSession* session =
        new NTSession(options, new DeviceMgr(devices), this);
    {
      mutex_lock l(sessions_lock_);
      sessions_.push_back(session);
    }
    return session;
  }

Status tensorflow::NTSessionFactory::Reset ( const SessionOptions &  options, const std::vector< string > &  containers  )

inlineoverride

Definition at line 133 of file NTSession.cc.

References checklumidiff::l, alignCSCRings::s, dataDML::session, sessions_lock_, btagGenBb_cfi::Status, and std::swap().

                                                             {
    std::vector<NTSession*> sessions_to_reset;
    {
      mutex_lock l(sessions_lock_);
      // We create a copy to ensure that we don't have a deadlock when
      // session->Close calls the NTSessionFactory.Deregister, which
      // acquires sessions_lock_.
      std::swap(sessions_to_reset, sessions_);
    }
    Status s;
    for (auto session : sessions_to_reset) {
      s.Update(session->Reset(containers));
    }
    // TODO(suharshs): Change the Reset behavior of all SessionFactories so that
    // it doesn't close the sessions?
    for (auto session : sessions_to_reset) {
      s.Update(session->Close());
    }
    return s;
  }

Member Data Documentation

mutex tensorflow::NTSessionFactory::sessions_lock_

private

Definition at line 162 of file NTSession.cc.

Referenced by Deregister(), NewSession(), and Reset().