Home
last modified time | relevance | path

Searched refs:UnboundedWorkQueue (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/platform/default/
Dunbounded_work_queue.cc25 UnboundedWorkQueue::UnboundedWorkQueue(Env* env, const string& thread_name, in UnboundedWorkQueue() function in tensorflow::UnboundedWorkQueue
29 UnboundedWorkQueue::~UnboundedWorkQueue() { in ~UnboundedWorkQueue()
55 void UnboundedWorkQueue::Schedule(WorkFunction fn) { in Schedule()
75 void UnboundedWorkQueue::PooledThreadFunc() { in PooledThreadFunc()
Dunbounded_work_queue.h37 class UnboundedWorkQueue {
39 UnboundedWorkQueue(Env* env, const string& thread_name,
41 ~UnboundedWorkQueue();
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.h31 std::shared_ptr<UnboundedWorkQueue> work_queue, in CollectiveRemoteAccessDistributed()
58 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dcollective_rma_distributed_test.cc186 std::make_shared<UnboundedWorkQueue>(Env::Default(), "test")) {} in CollRMADistTest()
288 std::shared_ptr<UnboundedWorkQueue> work_queue_;
/external/tensorflow/tensorflow/core/platform/
Dunbounded_work_queue_test.cc30 absl::make_unique<UnboundedWorkQueue>(Env::Default(), "test")) {} in UnboundedWorkQueueTest()
63 std::unique_ptr<UnboundedWorkQueue> work_queue_;
/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.h101 std::shared_ptr<UnboundedWorkQueue> work_queue) in BaseCollectiveExecutor()
145 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dcollective_executor_mgr.h78 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dcollective_executor_mgr.cc37 work_queue_(std::make_shared<UnboundedWorkQueue>(Env::Default(), in CollectiveExecutorMgr()
Dcollective_rma_local_test.cc42 work_queue_ = std::make_shared<UnboundedWorkQueue>(Env::Default(), "test"); in CollectiveRemoteAccessLocalTest()
60 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dpermuter_test.cc169 work_queue_ = std::make_shared<UnboundedWorkQueue>(Env::Default(), "test"); in Init()
434 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dring_gatherer_test.cc179 work_queue_ = std::make_shared<UnboundedWorkQueue>(Env::Default(), "test"); in Init()
528 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dring_reducer_test.cc202 work_queue_ = std::make_shared<UnboundedWorkQueue>(Env::Default(), "test"); in Init()
560 std::shared_ptr<UnboundedWorkQueue> work_queue_;
Dhierarchical_tree_broadcaster_test.cc262 work_queue_ = std::make_shared<UnboundedWorkQueue>(Env::Default(), "test"); in Init()
717 std::shared_ptr<UnboundedWorkQueue> work_queue_;
/external/tensorflow/tensorflow/core/kernels/data/
Dunbounded_thread_pool.h59 UnboundedWorkQueue unbounded_work_queue_;
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc89 work_queue_(std::make_shared<UnboundedWorkQueue>( in NcclTestBase()
437 std::shared_ptr<UnboundedWorkQueue> work_queue_;
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc76 work_queue_ = new UnboundedWorkQueue(Env::Default(), "nccl_manager_test"); in SetUpTestSuite()
432 static UnboundedWorkQueue* work_queue_;
460 UnboundedWorkQueue* NcclManagerTest<Scalar>::work_queue_ = nullptr;
Dnccl_manager.cc884 UnboundedWorkQueue queue(Env::Default(), "nccl_abort"); in StartAbort()