Searched refs:ComputationKind (Results 1 – 2 of 2) sorted by relevance
66 enum class ComputationKind { enum82 TrackedInstruction(HloInstruction* instruction, ComputationKind kind,100 ComputationKind kind_ = ComputationKind::kInvalid;
39 case ComputationKind::kInvalid: in ToString()42 case ComputationKind::kWhileCondition: in ToString()45 case ComputationKind::kWhileBody: in ToString()48 case ComputationKind::kConditionalBranch: in ToString()51 case ComputationKind::kCallFunction: in ToString()303 TrackedInstruction(hlo, ComputationKind::kWhileCondition); in RecordInstructions()305 TrackedInstruction(hlo, ComputationKind::kWhileBody); in RecordInstructions()309 TrackedInstruction(hlo, ComputationKind::kConditionalBranch, b); in RecordInstructions()313 TrackedInstruction(hlo, ComputationKind::kCallFunction); in RecordInstructions()