Searched refs:FoldOutputConversions (Results 1 – 1 of 1) sorted by relevance
55 Status FoldOutputConversions(HloInstruction* hlo);67 Status BFloat16ConversionFoldingVisitor::FoldOutputConversions( in FoldOutputConversions() function in xla::BFloat16ConversionFoldingVisitor143 TF_RETURN_IF_ERROR(FoldOutputConversions(hlo)); in TryFoldBF16Conversions()232 TF_RETURN_IF_ERROR(FoldOutputConversions(gte)); in HandleAllReduce()