Searched refs:BFloat16ConversionFoldingVisitor (Results 1 – 1 of 1) sorted by relevance
29 class BFloat16ConversionFoldingVisitor : public DfsHloVisitorWithDefault { class31 explicit BFloat16ConversionFoldingVisitor( in BFloat16ConversionFoldingVisitor() function in xla::BFloat16ConversionFoldingVisitor42 BFloat16ConversionFoldingVisitor visitor(computation, bfloat16_support); in Run()67 Status BFloat16ConversionFoldingVisitor::FoldOutputConversions( in FoldOutputConversions()79 Status BFloat16ConversionFoldingVisitor::FoldOperandConversion( in FoldOperandConversion()109 Status BFloat16ConversionFoldingVisitor::TryFoldBF16Conversions( in TryFoldBF16Conversions()152 Status BFloat16ConversionFoldingVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction()179 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce()244 if (BFloat16ConversionFoldingVisitor::Run(comp, bfloat16_support_)) { in Run()