Searched refs:kCholesky (Results 1 – 16 of 16) sorted by relevance
134 if (instruction->opcode() != HloOpcode::kCholesky) { in RunOnInstruction()156 if (hlo->opcode() == HloOpcode::kCholesky) { in RunOnComputation()
35 case Thunk::kCholesky: in ThunkKindToString()
47 kCholesky, enumerator
40 : Thunk(Kind::kCholesky, thunk_info), in CholeskyThunk()
63 V(kCholesky, "cholesky", 1) \
219 return instruction->opcode() == HloOpcode::kCholesky; in InstructionMatchesPattern()
147 case HloOpcode::kCholesky: in IsExpensive()
205 case HloOpcode::kCholesky: { in CreateFromProto()1710 case HloOpcode::kCholesky: in CloneWithNewOperands()2246 case HloOpcode::kCholesky: in IdenticalSlowPath()3270 case HloOpcode::kCholesky: in Visit()
1045 case HloOpcode::kCholesky: in GetInstructionColor()
238 case HloOpcode::kCholesky: in PickRepresentativeOperand()
94 case HloOpcode::kCholesky: in CanInferShape()1722 case HloOpcode::kCholesky: { in ParseInstructionRhs()
2270 case HloOpcode::kCholesky: in InstructionCanChangeLayout()
384 : HloInstruction(HloOpcode::kCholesky, shape), cholesky_options_(options) { in HloCholeskyInstruction()
354 return xla::HloOpcode::kCholesky; in MhloToHloOpcode()
340 case HloOpcode::kCholesky: { in ImportInstructionImpl()
1660 return AddInstruction(std::move(instr), HloOpcode::kCholesky, {a}); in CholeskyInternal()