Home
last modified time | relevance | path

Searched refs:HloCholeskyInstruction (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc325 HloCholeskyInstruction::HloCholeskyInstruction(const Shape& shape, in HloCholeskyInstruction() function in xla::HloCholeskyInstruction
332 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()
Dhlo_instructions.h184 class HloCholeskyInstruction : public HloInstruction {
186 explicit HloCholeskyInstruction(const Shape& shape, HloInstruction* a,
Dhlo_instruction.cc812 return absl::make_unique<HloCholeskyInstruction>(shape, a, options); in CreateCholesky()
3550 return Cast<HloCholeskyInstruction>(this)->cholesky_options(); in cholesky_options()