Searched refs:BFloat16NormalizationVisitor (Results 1 – 1 of 1) sorted by relevance
30 class BFloat16NormalizationVisitor : public DfsHloVisitorWithDefault { class32 explicit BFloat16NormalizationVisitor(HloComputation* computation, in BFloat16NormalizationVisitor() function in xla::BFloat16NormalizationVisitor40 BFloat16NormalizationVisitor visitor(computation, bfloat16_support); in Run()79 Status BFloat16NormalizationVisitor::InsertConvertAfterOutput( in InsertConvertAfterOutput()102 Status BFloat16NormalizationVisitor::ChangeOutputTypeThenInsertConvertBack( in ChangeOutputTypeThenInsertConvertBack()109 Status BFloat16NormalizationVisitor::InsertConvertBeforeOperand( in InsertConvertBeforeOperand()120 Status BFloat16NormalizationVisitor::ConvertCalledComputations( in ConvertCalledComputations()152 Status BFloat16NormalizationVisitor::HandleMultipleOutputs( in HandleMultipleOutputs()269 Status BFloat16NormalizationVisitor::HandleInstruction(HloInstruction* hlo) { in HandleInstruction()374 Status BFloat16NormalizationVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()[all …]