Searched refs:number_of_threads (Results 1 – 9 of 9) sorted by relevance
34 int number_of_threads) in TpuTransferAsyncOpKernelBase() argument105 int number_of_threads) in TpuTransferAsyncOpKernel() argument106 : TpuTransferAsyncOpKernelBase(ctx, transfer_type, number_of_threads) { in TpuTransferAsyncOpKernel()123 int number_of_threads) in TpuTransferAsyncDynamicOrdinalOpKernel() argument124 : TpuTransferAsyncOpKernelBase(ctx, transfer_type, number_of_threads) {} in TpuTransferAsyncDynamicOrdinalOpKernel()
32 int number_of_threads);61 int number_of_threads);77 int number_of_threads);
21 def __init__(self, number_of_threads): argument28 assert number_of_threads > 029 self._number_of_threads = number_of_threads
41 int number_of_threads = 4; member
167 auto delegate = evaluation::CreateXNNPACKDelegate(s->number_of_threads); in GetDelegates()277 if (settings->number_of_threads != -1) { in RunInference()278 interpreter->SetNumThreads(settings->number_of_threads); in RunInference()543 s.number_of_threads = strtol( // NOLINT(runtime/deprecated_fn) in Main()
69 pid_t SetupChildProcess(int number_of_threads) { in SetupChildProcess() argument71 sprintf(kNumberOfThreadsArgument, "%d", number_of_threads); in SetupChildProcess()104 for (int threads = 0; threads < number_of_threads; threads++) { in SetupChildProcess()
378 uint32_t number_of_threads; member
433 list.get()->number_of_threads = lwp_count - 1; in WriteLwpListStream()
1010 list.get()->number_of_threads = non_generator_thread_count; in WriteThreadListStream()