Searched refs:HloTriangularSolveInstruction (Results 1 – 4 of 4) sorted by relevance
501 HloTriangularSolveInstruction::HloTriangularSolveInstruction( in HloTriangularSolveInstruction() function in xla::HloTriangularSolveInstruction510 HloInstructionProto HloTriangularSolveInstruction::ToProto() const { in ToProto()517 HloTriangularSolveInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()522 bool HloTriangularSolveInstruction::IdenticalSlowPath( in IdenticalSlowPath()527 static_cast<const HloTriangularSolveInstruction&>(other); in IdenticalSlowPath()538 HloTriangularSolveInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()542 return std::make_unique<HloTriangularSolveInstruction>( in CloneWithNewOperandsImpl()
329 class HloTriangularSolveInstruction : public HloInstruction {331 explicit HloTriangularSolveInstruction(const Shape& shape, HloInstruction* a,
1232 return std::make_unique<HloTriangularSolveInstruction>(shape, a, b, options); in CreateTriangularSolve()4900 return Cast<HloTriangularSolveInstruction>(this)->triangular_solve_options(); in triangular_solve_options()
1315 xla::Cast<xla::HloTriangularSolveInstruction>(instr); in EmitTriangularSolveOp()