Searched refs:compute_pool (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| H A D | grpc_eager_service_impl.h | 57 env_->compute_pool->Schedule([this, call]() { \
75 env_->compute_pool->Schedule([this, call]() { in EnqueueHandler()
92 env_->compute_pool->Schedule([this, call]() { in RunComponentFunctionHandler()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/coordination/ |
| H A D | grpc_coordination_service_impl.cc | 24 thread::ThreadPool* compute_pool, ::grpc::ServerBuilder* server_builder) in GrpcCoordinationServiceImpl() argument
25 : compute_pool_(*compute_pool), shutdown_(false) { in GrpcCoordinationServiceImpl()
|
| H A D | grpc_coordination_service_impl.h | 43 GrpcCoordinationServiceImpl(thread::ThreadPool* compute_pool,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_server_lib.cc | 261 thread::ThreadPool* compute_pool = ComputePool(sess_opts); in Init() local
263 new GrpcCoordinationServiceImpl(compute_pool, &builder); in Init()
310 worker_env_.compute_pool = compute_pool; in Init()
|
| H A D | grpc_worker_service.cc | 175 worker_->env()->compute_pool->Schedule(std::move(f)); in Schedule()
205 worker_->env()->compute_pool->Schedule(std::move(closure)); \
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | process_util.cc | 89 static thread::ThreadPool* compute_pool = InitComputePool(options); in ComputePool() local
90 return compute_pool; in ComputePool()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | worker_env.h | 66 thread::ThreadPool* compute_pool = nullptr; member
|
| H A D | graph_mgr.cc | 144 graph_options.optimizer_options(), worker_env_->compute_pool, cluster_flr, in InitItem()
550 thread::ThreadPool* pool = worker_env_->compute_pool; in StartParallelExecutors()
|