Home
last modified time | relevance | path

Searched refs:num_intra_op_threads (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/
H A Dtf_threadpool_concurrent_work_queue.cc90 int num_inter_op_threads, int num_intra_op_threads) { in CreateDefaultTfThreadPoolWorkQueue() argument
95 ThreadPools(int num_inter_op_threads, int num_intra_op_threads) in CreateDefaultTfThreadPoolWorkQueue()
98 num_intra_op_threads) {} in CreateDefaultTfThreadPoolWorkQueue()
115 num_inter_op_threads, num_intra_op_threads)); in CreateDefaultTfThreadPoolWorkQueue()
H A Druntime.cc150 int num_intra_op_threads) { in Create() argument
151 if (num_intra_op_threads <= 0) in Create()
152 num_intra_op_threads = tensorflow::port::MaxParallelism(); in Create()
155 num_intra_op_threads, num_inter_op_threads))); in Create()
H A Druntime.h45 int num_intra_op_threads = 0);
H A Dtf_threadpool_concurrent_work_queue.h81 int num_inter_op_threads, int num_intra_op_threads);
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler_test.cc47 options.num_intra_op_threads = num_threads; in TEST()
86 pool_options.num_intra_op_threads = num_threads; in TEST()
123 pool_options.num_intra_op_threads = num_threads; in TEST()
H A Drun_handler_concurrent_work_queue.cc42 pool_options.num_intra_op_threads = options.num_complementary_threads; in RunHandlerThreadWorkQueue()
H A Drun_handler.h70 int num_intra_op_threads = 1; member
H A Drun_handler.cc720 options.num_inter_op_threads, options.num_intra_op_threads, in Impl()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler.cc779 explicit Impl(int num_inter_op_threads, int num_intra_op_threads) in Impl() argument
787 num_inter_op_threads, num_intra_op_threads, Env::Default(), in Impl()
1091 int num_intra_op_threads) in RunHandlerPool() argument
1092 : impl_(new Impl(num_inter_op_threads, num_intra_op_threads)) {} in RunHandlerPool()
H A Drun_handler.h55 RunHandlerPool(int num_inter_op_threads, int num_intra_op_threads);