• Home
  • Raw
  • Download

Lines Matching refs:unitriangular

4126         X = torch.linalg.solve_triangular(A, B, upper=upper, left=left, unitriangular=uni)
4135 torch.linalg.solve_triangular(A, B, upper=upper, left=left, unitriangular=uni, out=out)
4198 X = torch.linalg.solve_triangular(A, B, upper=upper, left=left, unitriangular=uni)
4206 def triangular_solve_test_helper(self, A_dims, b_dims, upper, unitriangular, argument
4214 if unitriangular:
4227 for k, n, (upper, unitriangular, transpose) in itertools.product(ks, ns,
4230 unitriangular, device, dtype)
4231 … x = torch.triangular_solve(b, A, upper=upper, unitriangular=unitriangular, transpose=transpose)[0]
4243 def triangular_solve_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4245 unitriangular, device, dtype)
4249 unitriangular=unitriangular,
4253 unitriangular=unitriangular,
4262 def triangular_solve_zero_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4264 unitriangular, device, dtype)
4266 unitriangular=unitriangular,
4270 for upper, unitriangular, transpose in itertools.product([True, False], repeat=3):
4273 upper, unitriangular, transpose)
4277 upper, unitriangular, transpose)
4279 upper, unitriangular, transpose)
4284 upper, unitriangular, transpose)
4294 for upper, transpose, unitriangular in itertools.product([True, False], repeat=3):
4297 upper, unitriangular, device, dtype)
4299 … upper=upper, transpose=transpose, unitriangular=unitriangular)
4310 upper, unitriangular, device, dtype)
4312 unitriangular=unitriangular)
4339 def run_test(A_dims, b_dims, device, upper, transpose, unitriangular): argument
4341 unitriangular, device, dtype)
4343 upper, transpose, unitriangular))
4344 … x = torch.triangular_solve(b, A, upper=upper, transpose=transpose, unitriangular=unitriangular)[0]
4348 for upper, transpose, unitriangular in itertools.product([True, False], repeat=3):
4350 …run_test((2, 1, 3, 4, 4), (2, 1, 3, 4, 6), device, upper, transpose, unitriangular) # no broadcas…
4351 … run_test((2, 1, 3, 4, 4), (4, 6), device, upper, transpose, unitriangular) # broadcasting b
4352 … run_test((4, 4), (2, 1, 3, 4, 2), device, upper, transpose, unitriangular) # broadcasting A
4353 …run_test((1, 3, 1, 4, 4), (2, 1, 3, 4, 5), device, upper, transpose, unitriangular) # broadcastin…