Searched refs:HloCholeskyInstruction (Results 1 – 3 of 3) sorted by relevance
325 HloCholeskyInstruction::HloCholeskyInstruction(const Shape& shape, in HloCholeskyInstruction() function in xla::HloCholeskyInstruction332 HloInstructionProto HloCholeskyInstruction::ToProto() const { in ToProto()338 std::vector<string> HloCholeskyInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()343 bool HloCholeskyInstruction::IdenticalSlowPath( in IdenticalSlowPath()347 const auto& casted_other = static_cast<const HloCholeskyInstruction&>(other); in IdenticalSlowPath()355 HloCholeskyInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()359 return absl::make_unique<HloCholeskyInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
184 class HloCholeskyInstruction : public HloInstruction {186 explicit HloCholeskyInstruction(const Shape& shape, HloInstruction* a,
812 return absl::make_unique<HloCholeskyInstruction>(shape, a, options); in CreateCholesky()3550 return Cast<HloCholeskyInstruction>(this)->cholesky_options(); in cholesky_options()