Lines Matching refs:num_threads
735 def _batch(tensors, batch_size, keep_input, num_threads=1, capacity=32,
756 _enqueue(queue, tensor_list, num_threads, enqueue_many, keep_input)
768 # TODO(josh11b): Add a thread_multiplier or num_threads (that has to be
809 keep_input, num_threads=1, seed=None, enqueue_many=False,
833 _enqueue(queue, tensor_list, num_threads, enqueue_many, keep_input)
898 def batch(tensors, batch_size, num_threads=1, capacity=32,
950 num_threads: The number of threads enqueuing `tensors`. The batching will
951 be nondeterministic if `num_threads > 1`.
982 num_threads=num_threads,
993 def maybe_batch(tensors, keep_input, batch_size, num_threads=1, capacity=32,
1009 num_threads: The number of threads enqueuing `tensors`. The batching will
1010 be nondeterministic if `num_threads > 1`.
1035 num_threads=num_threads,
1206 num_threads=1, seed=None, enqueue_many=False, shapes=None,
1243 num_threads=4,
1266 num_threads: The number of threads enqueuing `tensor_list`.
1295 num_threads=num_threads,
1306 keep_input, num_threads=1, seed=None,
1326 num_threads: The number of threads enqueuing `tensor_list`.
1355 num_threads=num_threads,
