Home
last modified time | relevance | path

Searched refs:num_inter_op_threads (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler_util_test.cc96 int num_inter_op_threads = 9; in TEST() local
97 std::vector<std::uint_fast32_t> start_vec(num_inter_op_threads); in TEST()
98 std::vector<std::uint_fast32_t> end_vec(num_inter_op_threads); in TEST()
104 ComputeInterOpStealingRanges(num_inter_op_threads, 6, &start_vec, &end_vec); in TEST()
107 for (int i = 0; i < num_inter_op_threads; ++i) { in TEST()
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()
1087 RunHandlerPool::RunHandlerPool(int num_inter_op_threads) in RunHandlerPool() argument
1088 : impl_(new Impl(num_inter_op_threads, 0)) {} in RunHandlerPool()
1090 RunHandlerPool::RunHandlerPool(int num_inter_op_threads, in RunHandlerPool() argument
1092 : impl_(new Impl(num_inter_op_threads, num_intra_op_threads)) {} in RunHandlerPool()
H A Drun_handler.h53 explicit RunHandlerPool(int num_inter_op_threads);
55 RunHandlerPool(int num_inter_op_threads, int num_intra_op_threads);
/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()
96 : inter_op_threadpool("default_work_queue_inter", num_inter_op_threads), in CreateDefaultTfThreadPoolWorkQueue()
115 num_inter_op_threads, num_intra_op_threads)); in CreateDefaultTfThreadPoolWorkQueue()
H A Druntime.cc149 std::unique_ptr<Runtime> Runtime::Create(int num_inter_op_threads, in Create() argument
155 num_intra_op_threads, num_inter_op_threads))); in Create()
H A Druntime.h44 static std::unique_ptr<Runtime> Create(int num_inter_op_threads,
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.cc48 options.num_inter_op_threads = num_threads; in TEST()
87 pool_options.num_inter_op_threads = num_threads; in TEST()
124 pool_options.num_inter_op_threads = num_threads; in TEST()
H A Drun_handler_concurrent_work_queue.cc41 pool_options.num_inter_op_threads = options.num_main_threads; in RunHandlerThreadWorkQueue()
H A Drun_handler.h67 int num_inter_op_threads = 1; member
H A Drun_handler.cc720 options.num_inter_op_threads, options.num_intra_op_threads, in Impl()