Home
last modified time | relevance | path

Searched refs:callback_threadpool_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_remote_worker.cc52 callback_threadpool_(callback_threadpool), in GrpcRemoteWorker()
278 callback_threadpool_, MaxRetries(), fail_fast, &target_); in IssueRequest()
286 callback_threadpool_, MaxRetries(), in IssueRequest()
314 thread::ThreadPool* callback_threadpool_; member in tensorflow::GrpcRemoteWorker
/aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/
H A Drpc_ops.cc404 callback_threadpool_ = std::make_unique<thread::ThreadPool>( in RpcClient()
428 /*threadpool=*/callback_threadpool_.get(), in CallAsync()
443 /*threadpool=*/callback_threadpool_.get(), in ListAsync()
454 std::unique_ptr<thread::ThreadPool> callback_threadpool_; member in tensorflow::rpc::RpcClient