Home
last modified time | relevance | path

Searched refs:kTriangularSolve (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.cc55 case Thunk::kTriangularSolve: in ThunkKindToString()
Dthunk.h61 kTriangularSolve, enumerator
Dtriangular_solve_thunk.cc40 : Thunk(Kind::kTriangularSolve, hlo), in TriangularSolveThunk()
Dgpu_layout_assignment.cc243 } else if (instruction->opcode() == HloOpcode::kTriangularSolve) { in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h142 V(kTriangularSolve, "triangular-solve", 2) \
Dinstruction_fusion.cc159 case HloOpcode::kTriangularSolve: in IsExpensive()
Dtriangular_solve_expander.cc449 return instruction->opcode() == HloOpcode::kTriangularSolve; in InstructionMatchesPattern()
Dtuple_points_to_analysis.cc784 if (user->opcode() == HloOpcode::kTriangularSolve) { in CanShareOperandBufferWithUser()
Dhlo_instruction.cc178 case HloOpcode::kTriangularSolve: { in CreateFromProto()
1390 case HloOpcode::kTriangularSolve: in CloneWithNewOperands()
1840 case HloOpcode::kTriangularSolve: in IdenticalSlowPath()
2663 case HloOpcode::kTriangularSolve: in Visit()
Dhlo_graph_dumper.cc1021 case HloOpcode::kTriangularSolve: in GetInstructionColor()
Dlayout_assignment.cc2040 case HloOpcode::kTriangularSolve: in InstructionCanChangeLayout()
Dhlo_instructions.cc284 : HloInstruction(HloOpcode::kTriangularSolve, shape), in HloTriangularSolveInstruction()
Dhlo_parser.cc1120 case HloOpcode::kTriangularSolve: { in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3048 HloOpcode::kTriangularSolve, {a, b}); in TriangularSolve()