Searched refs:bfloat16_support_ (Results 1 – 8 of 8) sorted by relevance
34 : computation_(computation), bfloat16_support_(bfloat16_support) {} in BFloat16NormalizationVisitor()75 const BFloat16Support* bfloat16_support_; member in xla::BFloat16NormalizationVisitor167 if (!bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleMultipleOutputs()175 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in HandleMultipleOutputs()189 if (!bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleMultipleOutputs()192 if (bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in HandleMultipleOutputs()208 (bfloat16_support_->SupportsMixedPrecisions(*hlo) || f32_count == 0 || in HandleMultipleOutputs()312 !bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleInstruction()321 !bfloat16_support_->SupportsBF16Output(*hlo)) { in HandleInstruction()331 if (bfloat16_support_->SupportsMixedPrecisions(*hlo) || bf16_count == 0 || in HandleInstruction()[all …]
33 : computation_(computation), bfloat16_support_(bfloat16_support) {} in BFloat16ConversionFoldingVisitor()63 const BFloat16Support* bfloat16_support_; member in xla::BFloat16ConversionFoldingVisitor118 bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in TryFoldBF16Conversions()131 bfloat16_support_->SupportsBF16Output(*hlo); in TryFoldBF16Conversions()133 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in TryFoldBF16Conversions()170 !bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in DefaultAction()188 if (!bfloat16_support_->SupportsMixedPrecisions(*crs)) { in HandleAllReduce()244 if (BFloat16ConversionFoldingVisitor::Run(comp, bfloat16_support_)) { in Run()
37 : bfloat16_support_(bfloat16_support) {} in BFloat16ConversionFolding()47 const BFloat16Support* bfloat16_support_;
31 : bfloat16_support_(bfloat16_support) {} in BFloat16Normalization()41 const BFloat16Support* bfloat16_support_;
35 : bfloat16_support_(bfloat16_support) {} in BFloat16Propagation()40 if (!bfloat16_support_->SupportsMixedPrecisions(*fusion)) { in DetermineFusionComputationPrecision()269 if (bfloat16_support_->EffectiveOperandPrecisionIsBF16( in AllUsersConsumeBF16()276 if (bfloat16_support_->EffectiveOperandPrecisionIsOutputPrecision( in AllUsersConsumeBF16()372 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in DetermineInstructionPrecision()390 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo) && in InstructionIsCandidateForBF16Output()396 if (!bfloat16_support_->EffectiveOperandPrecisionIsOutputPrecision(*hlo, in InstructionIsCandidateForBF16Output()
206 const BFloat16Support* bfloat16_support_; variable
75 TestBFloat16Support bfloat16_support_; in FoldConversions() local76 BFloat16ConversionFolding fold(&bfloat16_support_); in FoldConversions()
79 TestBFloat16Support bfloat16_support_; in Normalize() local80 BFloat16Normalization normalization(&bfloat16_support_); in Normalize()