Searched refs:HloTriangularSolveInstruction (Results 1 – 3 of 3) sorted by relevance
281 HloTriangularSolveInstruction::HloTriangularSolveInstruction( in HloTriangularSolveInstruction() function in xla::HloTriangularSolveInstruction290 HloInstructionProto HloTriangularSolveInstruction::ToProto() const { in ToProto()296 std::vector<string> HloTriangularSolveInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()301 bool HloTriangularSolveInstruction::IdenticalSlowPath( in IdenticalSlowPath()306 static_cast<const HloTriangularSolveInstruction&>(other); in IdenticalSlowPath()317 HloTriangularSolveInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()321 return absl::make_unique<HloTriangularSolveInstruction>( in CloneWithNewOperandsImpl()
156 class HloTriangularSolveInstruction : public HloInstruction {158 explicit HloTriangularSolveInstruction(const Shape& shape, HloInstruction* a,
807 return absl::make_unique<HloTriangularSolveInstruction>(shape, a, b, options); in CreateTriangularSolve()3546 return Cast<HloTriangularSolveInstruction>(this)->triangular_solve_options(); in triangular_solve_options()