Lines Matching full:zeros

63         expected = torch.zeros(128).uniform_().to(dtype=dtype)
66             tensors.append(torch.zeros(128, dtype=dtype, device=device))
75             tensors.append(torch.zeros(128, dtype=dtype, device=device))
89             torch.zeros(128).uniform_().to(dtype=dtype) for i in range(nGPUs)
91         expected = torch.zeros(128, dtype=dtype)
115             torch.zeros(128).uniform_().to(dtype=dtype) for i in range(nGPUs)
117         expected = torch.zeros(128, dtype=dtype)
179         cpu_inputs = [torch.zeros(128).uniform_().to(dtype=dtype) for i in range(nGPUs)]
184             torch.zeros(128 * nGPUs, device=i, dtype=dtype) for i in range(nGPUs)
194             torch.zeros(128 * nGPUs, device=i, dtype=dtype) for i in range(nGPUs)
212             torch.zeros(in_size).uniform_().to(dtype=dtype) for i in range(nGPUs)
214         expected = torch.zeros(in_size, dtype=dtype)
220         outputs = [torch.zeros(out_size, device=i, dtype=dtype) for i in range(nGPUs)]
228         outputs = [torch.zeros(out_size, device=i, dtype=dtype) for i in range(nGPUs)]