Home
last modified time | relevance | path

Searched defs:thread_pool (Results 1 – 25 of 36) sorted by relevance

12

/external/eigen/unsupported/test/
Dcxx11_tensor_notification.cpp37 ThreadPool thread_pool(1); in test_notification_single() local
61 ThreadPool thread_pool(1); in test_notification_multiple() local
Dcxx11_tensor_thread_pool.cpp275 ThreadPool thread_pool(num_threads); in test_multithreaded_reductions() local
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dthreadpool.py66 def __init__(self, input_dataset, thread_pool): argument
78 def override_threadpool(dataset, thread_pool): argument
/external/tensorflow/tensorflow/core/lib/core/
Dnotification_test.cc27 thread::ThreadPool* thread_pool = in TEST() local
56 thread::ThreadPool* thread_pool = in TEST() local
Dblocking_counter_test.cc38 thread::ThreadPool* thread_pool = in TEST() local
55 std::unique_ptr<thread::ThreadPool> thread_pool( in BM_BlockingCounter() local
/external/grpc-grpc/src/python/grpcio/grpc/beta/
Dimplementations.py144 metadata_transformer, thread_pool, thread_pool_size): argument
160 thread_pool=None, argument
243 response_serializers, thread_pool, thread_pool_size, argument
260 thread_pool=None, argument
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/
Dexample_partitioner.cc25 const int desired_parallelism, thread::ThreadPool* const thread_pool, in UpdatePartitions()
58 const int desired_parallelism, thread::ThreadPool* const thread_pool, in PartitionExamples()
/external/grpc-grpc/src/python/grpcio/grpc/
D_server.py523 def _handle_unary_unary(rpc_event, state, method_handler, thread_pool): argument
532 def _handle_unary_stream(rpc_event, state, method_handler, thread_pool): argument
541 def _handle_stream_unary(rpc_event, state, method_handler, thread_pool): argument
550 def _handle_stream_stream(rpc_event, state, method_handler, thread_pool): argument
592 def _handle_with_method_handler(rpc_event, method_handler, thread_pool): argument
615 def _handle_call(rpc_event, generic_handlers, interceptor_pipeline, thread_pool, argument
652 interceptor_pipeline, thread_pool, maximum_concurrent_rpcs): argument
805 def __init__(self, thread_pool, generic_handlers, interceptors, options, argument
835 def create_server(thread_pool, generic_rpc_handlers, interceptors, options, argument
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dparallel_for.cc23 thread::ThreadPool* thread_pool, in ParallelFor()
/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_cpu_device.cc34 static thread::ThreadPool* thread_pool = in GraphRunnerThreadPool() local
/external/tensorflow/tensorflow/core/kernels/
Dbincount_op.cc55 ThreadPool* thread_pool = in Compute() local
Dmatrix_set_diag_op.cc131 auto thread_pool = in Compute() local
Dmatrix_band_part_op.cc144 auto thread_pool = in operator ()() local
Dsparse_matmul_op.cc1142 const DeviceBase::CpuWorkerThreads* thread_pool) { in CreateSparseSlices()
1238 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer) { in ShuffleMatrix()
1306 const DeviceBase::CpuWorkerThreads* thread_pool, MatrixR* buffer, in CreateDenseSlices()
1362 void do_on_all_threads(const DeviceBase::CpuWorkerThreads* thread_pool, in do_on_all_threads()
1429 bool transpose_left, const DeviceBase::CpuWorkerThreads* thread_pool, in Compute()
1534 bool transpose_left, const DeviceBase::CpuWorkerThreads* thread_pool, in Compute()
Ddynamic_stitch_op.cc288 auto thread_pool = in Compute() local
/external/grpc-grpc/src/python/grpcio_tests/tests/unit/_cython/
D_cancel_many_calls_test.py94 def _serve(state, server, server_completion_queue, thread_pool): argument
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc78 thread::ThreadPool thread_pool(Env::Default(), "pool", 2); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.cc231 tensorflow::thread::ThreadPool thread_pool( in GetStreamExecutors() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc78 thread::ThreadPool* thread_pool) in XlaDeviceContext()
/external/tensorflow/tensorflow/contrib/coder/kernels/
Dpmf_to_cdf_op.cc69 thread::ThreadPool* thread_pool = in Compute() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc281 thread::ThreadPool thread_pool(Env::Default(), "extend_pool", 100); in TEST_F() local
353 thread::ThreadPool thread_pool(Env::Default(), "extend_pool", 3); in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h260 void set_tensorflow_device_thread_pool(thread::ThreadPool* thread_pool) { in set_tensorflow_device_thread_pool()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc366 tensorflow::thread::ThreadPool thread_pool( in RealMain() local
/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_io_utils_test.cc36 std::unique_ptr<thread::ThreadPool> thread_pool; member
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc182 std::unique_ptr<thread::ThreadPool> thread_pool( in TEST_F() local

12