Searched defs:number_of_threads (Results 1 – 5 of 5) sorted by relevance
34 int number_of_threads, std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuTransferAsyncOpKernelBase()97 int number_of_threads, std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuTransferAsyncOpKernel()116 int number_of_threads, std::unique_ptr<TpuTransferOpInterface> transfer_op) in TpuTransferAsyncDynamicOrdinalOpKernel()
21 def __init__(self, number_of_threads): argument
41 int number_of_threads = 4; member
69 pid_t SetupChildProcess(int number_of_threads) { in SetupChildProcess()
378 uint32_t number_of_threads; member