Searched refs:build_sparse_tensor (Results 1 – 2 of 2) sorted by relevance
83 def build_sparse_tensor(coalesce=False, requires_grad=True, dtype=torch.float32): function281 t1 = build_sparse_tensor()282 t2 = build_sparse_tensor()341 t1 = build_sparse_tensor(requires_grad=True)342 t2 = build_sparse_tensor(requires_grad=True)424 t1 = build_sparse_tensor(requires_grad=True)425 t2 = build_sparse_tensor(requires_grad=True)496 t1 = build_sparse_tensor(requires_grad=False)497 t2 = build_sparse_tensor(requires_grad=False)539 tensor = build_sparse_tensor(requires_grad=(i % 2 == 0))[all …]
222 def build_sparse_tensor(coalesce=False): function304 args=(build_sparse_tensor(), build_sparse_tensor())334 args=(build_sparse_tensor(), build_sparse_tensor())339 args=(build_sparse_tensor(), build_sparse_tensor())349 rref = rpc.remote(dst, torch.add, args=(build_sparse_tensor(), build_sparse_tensor()))812 x = build_sparse_tensor() * n813 y = build_sparse_tensor() * n1099 tensor = build_sparse_tensor().to("cuda:0")1100 add_tensor = build_sparse_tensor().to("cuda:0")2697 return (build_sparse_tensor(), build_sparse_tensor())[all …]