Searched refs:DirectSession (Results 1 – 6 of 6) sorted by relevance
164 DirectSession* session = in NewSession()165 new DirectSession(options, new DeviceMgr(std::move(devices)), this); in NewSession()176 std::vector<DirectSession*> sessions_to_reset; in Reset()196 void Deregister(const DirectSession* session) { in Deregister()204 std::vector<DirectSession*> sessions_ GUARDED_BY(sessions_lock_);215 std::atomic_int_fast64_t DirectSession::step_id_counter_(1);232 void DirectSession::SchedClosure(thread::ThreadPool* pool, in SchedClosure()258 bool DirectSession::ShouldUseRunHandlerPool( in ShouldUseRunHandlerPool()274 DirectSession::DirectSession(const SessionOptions& options, in DirectSession() function in tensorflow::DirectSession332 DirectSession::~DirectSession() { in ~DirectSession()[all …]
96 DirectSession* ds = static_cast<DirectSession*>(session.get()); in TEST()158 DirectSession* ds = static_cast<DirectSession*>(session.get()); in TEST()223 DirectSession* ds = static_cast<DirectSession*>(session.get()); in TestHWAccelerator()
55 class DirectSession : public Session {63 DirectSession(const SessionOptions& options, const DeviceMgr* device_mgr,65 ~DirectSession() override;406 TF_DISALLOW_COPY_AND_ASSIGN(DirectSession);
2267 DirectSession* direct_session = static_cast<DirectSession*>(session.get()); in RunGraphWithCollectiveFunctions()
1647 # This is a link-only library to provide a DirectSession
1244 * Improved DirectSession::Run() overhead and error checking. Feeding a value of the wrong type will…