Searched refs:kCholesky (Results 1 – 13 of 13) sorted by relevance
136 if (instruction->opcode() != HloOpcode::kCholesky) { in RunOnInstruction()161 if (hlo->opcode() == HloOpcode::kCholesky) { in RunOnComputation()
23 case Thunk::kCholesky: in ThunkKindToString()
45 kCholesky, enumerator
40 : Thunk(Kind::kCholesky, hlo), in CholeskyThunk()
66 V(kCholesky, "cholesky", 1) \
211 return instruction->opcode() == HloOpcode::kCholesky; in InstructionMatchesPattern()
124 case HloOpcode::kCholesky: in IsExpensive()
183 case HloOpcode::kCholesky: { in CreateFromProto()1391 case HloOpcode::kCholesky: in CloneWithNewOperands()1841 case HloOpcode::kCholesky: in IdenticalSlowPath()2665 case HloOpcode::kCholesky: in Visit()
1022 case HloOpcode::kCholesky: in GetInstructionColor()
2041 case HloOpcode::kCholesky: in InstructionCanChangeLayout()
328 : HloInstruction(HloOpcode::kCholesky, shape), cholesky_options_(options) { in HloCholeskyInstruction()
1144 case HloOpcode::kCholesky: { in ParseInstructionRhs()
3063 return builder->AddInstruction(std::move(instr), HloOpcode::kCholesky, {a}); in Cholesky()