tensorflow::TBBSessionFactory Class Reference

Inheritance diagram for tensorflow::TBBSessionFactory:

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

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

Private Attributes
mutex	sessions_lock_

Detailed Description

Definition at line 110 of file TBBSession.cc.

Constructor & Destructor Documentation

tensorflow::TBBSessionFactory::TBBSessionFactory ( )

inline

Definition at line 112 of file TBBSession.cc.

Referenced by tensorflow::TBBSessionRegistrar::TBBSessionRegistrar().

{}

Member Function Documentation

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

inlineoverride

Definition at line 114 of file TBBSession.cc.

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

void tensorflow::TBBSessionFactory::Deregister ( const TBBSession *  session )

inline

Definition at line 157 of file TBBSession.cc.

References cmsLHEtoEOSManager::l, MatrixUtil::remove(), and sessions_lock_.

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

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

std::vector<TBBSession*> sessions_ tensorflow::TBBSessionFactory::GUARDED_BY ( sessions_lock_  )

private

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

inlineoverride

Definition at line 116 of file TBBSession.cc.

References dqm::qstatus::ERROR, cmsLHEtoEOSManager::l, LOG, alignCSCRings::s, 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;
      }

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

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

inlineoverride

Definition at line 136 of file TBBSession.cc.

References cmsLHEtoEOSManager::l, alignCSCRings::s, sessions_lock_, btagGenBb_cfi::Status, and std::swap().

                                                                                              {
      std::vector<TBBSession*> 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 TBBSessionFactory.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::TBBSessionFactory::sessions_lock_

private

Definition at line 163 of file TBBSession.cc.

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