Home
last modified time | relevance | path

Searched refs:thread_pool_ (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/
Dexample_partitioner_test.cc28 : thread_pool_(tensorflow::Env::Default(), "test_pool", 2), in ExamplePartitionerTest()
35 thread::ThreadPool thread_pool_; member in tensorflow::boosted_trees::learner::__anon313ecfb70111::ExamplePartitionerTest
44 &thread_pool_, in TEST_F()
67 &thread_pool_, in TEST_F()
90 &thread_pool_, in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc33 : thread_pool_(env, thread_options, name, num_threads, low_latency_hint), in ThreadPoolResource()
39 thread_pool_.Schedule(std::move(fn)); in Schedule()
41 thread_pool_.Schedule(std::bind( in Schedule()
52 int32 NumThreads() { return thread_pool_.NumThreads(); } in NumThreads()
57 thread::ThreadPool thread_pool_; member in tensorflow::data::__anon808a52080111::ThreadPoolResource
355 thread_pool_ = absl::make_unique<thread::ThreadPool>( in Dataset()
408 thread::ThreadPool* pool = dataset()->thread_pool_.get(); in GetNextInternal()
431 std::unique_ptr<thread::ThreadPool> thread_pool_; member in tensorflow::data::__anon808a52080111::PrivateThreadPoolDatasetOp::Dataset
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc45 thread_pool_.reset(new thread::ThreadPool( in SingleMachine()
68 thread_pool_.reset(); in ~SingleMachine()
266 timeout_s * 1000, thread_pool_.get()); in RunWithTimeout()
277 if (!session_ || !thread_pool_) { in CloseSession()
309 use_timeout ? timeout_s_ * 1000 : -1, thread_pool_.get()); in CloseSession()
332 thread_pool_.reset(); in ShutdownSession()
362 thread_pool_.reset(new thread::ThreadPool( in ResetSession()
Dsingle_machine.h77 std::unique_ptr<thread::ThreadPool> thread_pool_; variable
/external/tensorflow/tensorflow/cc/training/
Dcoordinator_test.cc61 thread_pool_.reset(new thread::ThreadPool(Env::Default(), "test-pool", 10)); in MockQueueRunner()
72 thread_pool_->Schedule( in StartCounting()
78 thread_pool_->Schedule(std::bind(&MockQueueRunner::SetStatusThread, this, in StartSettingStatus()
86 thread_pool_.reset(); in Join()
113 std::unique_ptr<thread::ThreadPool> thread_pool_; member in tensorflow::__anon7bc84bb50111::MockQueueRunner
Dqueue_runner.cc76 thread_pool_.reset(new thread::ThreadPool( in Init()
99 thread_pool_->Schedule( in Start()
103 thread_pool_->Schedule(std::bind(&QueueRunner::Stop, this, sess)); in Start()
140 thread_pool_.reset(); in Join()
Dqueue_runner.h119 std::unique_ptr<thread::ThreadPool> thread_pool_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dunbounded_thread_pool.cc86 thread_pool_.clear(); in ~UnboundedThreadPool()
96 return thread_pool_.size(); in size()
123 thread_pool_.emplace_back(new_thread); in RunOnPooledThread()
Ddataset_test_base.cc133 thread_pool_ = absl::make_unique<thread::ThreadPool>( in InitThreadPool()
160 opts, thread_pool_.get(), nullptr /* cluster_flr */); in InitFunctionLibraryRuntime()
162 if (thread_pool_ == nullptr) { in InitFunctionLibraryRuntime()
166 thread_pool_->Schedule(std::move(fn)); in InitFunctionLibraryRuntime()
Dunbounded_thread_pool.h71 std::vector<std::unique_ptr<Thread>> thread_pool_ GUARDED_BY(thread_pool_mu_);
Dparallel_interleave_dataset_op.cc210 thread_pool_(absl::make_unique<thread::ThreadPool>( in ParallelInterleaveIterator()
498 thread_pool_->Schedule( in CurrentElementsManager()
613 thread_pool_->Schedule( in FutureElementsManager()
911 std::unique_ptr<thread::ThreadPool> thread_pool_; member in tensorflow::data::__anon6594f9940111::ParallelInterleaveDatasetOp::Dataset::ParallelInterleaveIterator
/external/v8/src/libplatform/
Ddefault-worker-threads-task-runner.cc16 thread_pool_.push_back(base::make_unique<WorkerThread>(&queue_)); in DefaultWorkerThreadsTaskRunner()
30 thread_pool_.clear(); in Terminate()
Ddefault-worker-threads-task-runner.h40 std::vector<std::unique_ptr<WorkerThread>> thread_pool_; in NON_EXPORTED_BASE() local
/external/grpc-grpc/src/cpp/server/
Dsecure_server_credentials.h42 : thread_pool_(CreateDefaultThreadPool()), processor_(processor) {} in AuthMetadataProcessorAyncWrapper()
48 std::unique_ptr<ThreadPoolInterface> thread_pool_; variable
/external/tensorflow/tensorflow/core/kernels/
Dmutex_ops.cc40 thread_pool_(new thread::ThreadPool( in Mutex()
131 thread_pool_->Schedule(std::bind( in AcquireAsync()
162 std::unique_ptr<thread::ThreadPool> thread_pool_; member in tensorflow::__anon07f294b10111::Mutex
Dreader_ops.cc51 thread_pool_(new thread::ThreadPool( in ReaderVerbAsyncOpKernel()
61 thread_pool_->Schedule([this, context, reader, done]() { in ComputeAsync()
73 std::unique_ptr<thread::ThreadPool> thread_pool_; member in tensorflow::ReaderVerbAsyncOpKernel
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc61 thread_pool_(NewThreadPoolFromSessionOptions(opts)), in EagerContext()
64 opts.config.graph_options().optimizer_options(), thread_pool_.get())), in EagerContext()
81 this->thread_pool_->Schedule(std::move(closure)); in EagerContext()
446 {}, thread_pool_.get())); in StoreCollectiveOpsServer()
479 {}, thread_pool_.get())); in InitializeRemote()
/external/tensorflow/tensorflow/core/grappler/costs/
Dmeasuring_cost_estimator.cc40 thread_pool_.reset(new thread::ThreadPool( in MeasuringCostEstimator()
128 thread_pool_->Schedule([i, &measurement_fn]() { measurement_fn(i); }); in PredictCosts()
Dmeasuring_cost_estimator.h70 std::unique_ptr<thread::ThreadPool> thread_pool_; variable
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_kernels.cc183 thread_pool_(new thread::ThreadPool( in ToBigtableOp()
192 thread_pool_->Schedule([this, ctx, done]() { in ComputeAsync()
352 std::unique_ptr<thread::ThreadPool> thread_pool_; member in tensorflow::data::__anon24f989ab0311::ToBigtableOp
/external/grpc-grpc/src/cpp/client/
Dsecure_credentials.h78 std::unique_ptr<ThreadPoolInterface> thread_pool_; variable
Dsecure_credentials.cc221 w->thread_pool_->Add( in GetMetadata()
295 : thread_pool_(CreateDefaultThreadPool()), plugin_(std::move(plugin)) {} in MetadataCredentialsPluginWrapper()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.h114 thread::ThreadPool* thread_pool_; variable
Dxla_device.cc209 thread_pool_.reset(new thread::ThreadPool(session_options.env, "xla_device", in XlaDevice()
327 client(), shape_representation_fn_, thread_pool_.get()); in GetDeviceContextLocked()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor_test.cc48 thread_pool_ = ComputePool(options);
73 runner_ = [this](std::function<void()> fn) { thread_pool_->Schedule(fn); }; in Create()
85 thread::ThreadPool* thread_pool_ = nullptr; member in tensorflow::ExecutorTest

12