Searched refs:workers_pool (Results 1 – 6 of 6) sorted by relevance
87 auto workers_pool = context->workers_pool(); in MultiThreadTransform1D() local92 workers_pool->Execute(tasks); in MultiThreadTransform1D()
133 auto workers_pool = context->workers_pool(); in MultiThreadGemm() local138 workers_pool->Execute(tasks); in MultiThreadGemm()
38 WorkersPool* workers_pool() { return pool_; } in workers_pool() function
263 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads, in MetaGemmQuantized8Bit()269 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads, in MetaGemmQuantized8Bit()287 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads, in MetaGemmFloat()292 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads, in MetaGemmFloat()
540 WorkersPool* workers_pool() { return &workers_pool_; } in workers_pool() function646 auto* workers_pool = context->workers_pool(); in MultiThreadGemm() local685 workers_pool->Execute(tasks); in MultiThreadGemm()
949 TensorflowGemmlowpWorkersPool* workers_pool() { return &workers_pool_; } in workers_pool() function