Searched refs:triangular_solve_op (Results 1 – 1 of 1) sorted by relevance
1599 auto triangular_solve_op = in EmitTriangularSolveFromMlir() local1606 TF_RET_CHECK(has_fortran_layout(triangular_solve_op.layout_a())); in EmitTriangularSolveFromMlir()1607 TF_RET_CHECK(has_fortran_layout(triangular_solve_op.layout_b())); in EmitTriangularSolveFromMlir()1608 TF_RET_CHECK(has_fortran_layout(triangular_solve_op.layout_output())); in EmitTriangularSolveFromMlir()1610 const Shape b_shape = TypeToShape(triangular_solve_op.b().getType()); in EmitTriangularSolveFromMlir()1613 TypeToShape(triangular_solve_op.output().getType()); in EmitTriangularSolveFromMlir()1616 GetAllocationSliceForMlir(triangular_solve_op.a())); in EmitTriangularSolveFromMlir()1618 GetAllocationSliceForMlir(triangular_solve_op.b())); in EmitTriangularSolveFromMlir()1620 GetAllocationSliceForMlir(triangular_solve_op.output())); in EmitTriangularSolveFromMlir()1622 ConvertTranspose(triangular_solve_op.transpose_a())); in EmitTriangularSolveFromMlir()[all …]