Lines Matching defs:queue
62 def _test_cuda_ipc_deadlock_actor(queue, iterations): argument
69 def _test_cuda_ipc_deadlock_learner(queue, iterations): argument
77 def simple_fill(queue, event): argument
88 def send_tensor(queue, event, device, dtype): argument
95 def send_and_delete_tensors(queue, event, device, dtype, count, size=5): argument
103 def receive_and_send_sum(queue, out_queue, event, device, dtype, count, size=5): argument
112 def receive_and_send(queue, out_queue, event, count): argument
144 def cuda_multiply_two(queue, ready, done): argument
155 def requires_grad_variable_sharing(queue, ready): argument
165 def autograd_sharing(queue, ready, master_modified, device, is_parameter): argument
186 def mixed_type_producer(queue, event): argument