Searched refs:kTriangularSolve (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | thunk.cc | 79 case Thunk::kTriangularSolve: in ThunkKindToString()
|
| D | thunk.h | 69 kTriangularSolve, enumerator
|
| D | triangular_solve_thunk.cc | 40 : Thunk(Kind::kTriangularSolve, thunk_info), in TriangularSolveThunk()
|
| D | gpu_layout_assignment.cc | 276 } else if (instruction->opcode() == HloOpcode::kTriangularSolve) { in AddBackendConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode.h | 151 V(kTriangularSolve, "triangular-solve", 2) \
|
| D | instruction_fusion.cc | 187 case HloOpcode::kTriangularSolve: in IsExpensive()
|
| D | hlo_dataflow_analysis.cc | 1342 user->opcode() == HloOpcode::kTriangularSolve || in CanShareOperandBufferWithUser()
1347 int64 operand_no = user->opcode() == HloOpcode::kTriangularSolve ? 1 : 0; in CanShareOperandBufferWithUser()
|
| D | triangular_solve_expander.cc | 558 return instruction->opcode() == HloOpcode::kTriangularSolve; in InstructionMatchesPattern()
|
| D | hlo_instruction.cc | 200 case HloOpcode::kTriangularSolve: { in CreateFromProto()
1709 case HloOpcode::kTriangularSolve: in CloneWithNewOperands()
2245 case HloOpcode::kTriangularSolve: in IdenticalSlowPath()
3268 case HloOpcode::kTriangularSolve: in Visit()
|
| D | hlo_graph_dumper.cc | 1044 case HloOpcode::kTriangularSolve: in GetInstructionColor()
|
| D | sharding_propagation.cc | 276 case HloOpcode::kTriangularSolve: in PickRepresentativeOperand()
|
| D | hlo_parser.cc | 156 case HloOpcode::kTriangularSolve: in CanInferShape()
1680 case HloOpcode::kTriangularSolve: { in ParseInstructionRhs()
|
| D | layout_assignment.cc | 2269 case HloOpcode::kTriangularSolve: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 340 : HloInstruction(HloOpcode::kTriangularSolve, shape), in HloTriangularSolveInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 477 return xla::HloOpcode::kTriangularSolve; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 638 case HloOpcode::kTriangularSolve: { in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 384 case HloOpcode::kTriangularSolve: in EmitOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1650 return AddInstruction(std::move(instr), HloOpcode::kTriangularSolve, {a, b}); in TriangularSolveInternal()
|