Home
last modified time | relevance | path

Searched refs:bfloat16_support_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding.cc37 bfloat16_support_(bfloat16_support), in BFloat16ConversionFoldingVisitor()
70 const BFloat16Support* bfloat16_support_; member in xla::BFloat16ConversionFoldingVisitor
127 bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in TryFoldBF16Conversions()
140 bfloat16_support_->SupportsBF16Output(*hlo); in TryFoldBF16Conversions()
142 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in TryFoldBF16Conversions()
181 !bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in DefaultAction()
199 if (!bfloat16_support_->SupportsMixedPrecisions(*crs)) { in HandleAllReduce()
261 if (BFloat16ConversionFoldingVisitor::Run(comp, bfloat16_support_, this)) { in Run()
Dbfloat16_normalization.cc42 bfloat16_support_(bfloat16_support), in BFloat16NormalizationVisitor()
90 const BFloat16Support* bfloat16_support_; member in xla::__anon96c306cc0111::BFloat16NormalizationVisitor
278 if (!bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleMultipleOutputs()
286 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in HandleMultipleOutputs()
300 if (!bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleMultipleOutputs()
303 if (bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in HandleMultipleOutputs()
320 (bfloat16_support_->SupportsMixedPrecisions(*hlo) || f32_count == 0 || in HandleMultipleOutputs()
426 !bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleInstruction()
435 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in HandleInstruction()
449 if (bfloat16_support_->SupportsMixedPrecisions(*hlo) || bf16_count == 0 || in HandleInstruction()
[all …]
Dbfloat16_conversion_folding.h37 : bfloat16_support_(bfloat16_support) {} in BFloat16ConversionFolding()
47 const BFloat16Support* bfloat16_support_;
Dbfloat16_normalization.h31 : bfloat16_support_(bfloat16_support) {} in BFloat16Normalization()
41 const BFloat16Support* bfloat16_support_;
Dbfloat16_propagation.cc37 : bfloat16_support_(bfloat16_support) {} in BFloat16Propagation()
42 if (!bfloat16_support_->SupportsMixedPrecisions(*fusion)) { in DetermineFusionComputationPrecision()
306 if (bfloat16_support_->EffectiveOperandPrecisionIsBF16( in AllUsersConsumeBF16()
313 if (bfloat16_support_->EffectiveOperandPrecisionIsOutputPrecision( in AllUsersConsumeBF16()
435 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in DetermineInstructionPrecision()
453 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo) && in InstructionIsCandidateForBF16Output()
459 if (!bfloat16_support_->EffectiveOperandPrecisionIsOutputPrecision(*hlo, in InstructionIsCandidateForBF16Output()
Dbfloat16_propagation.h211 const BFloat16Support* bfloat16_support_; variable
Dbfloat16_conversion_folding_test.cc75 TestBFloat16Support bfloat16_support_; in FoldConversions() local
76 BFloat16ConversionFolding fold(&bfloat16_support_); in FoldConversions()
Dbfloat16_normalization_test.cc81 TestBFloat16Support bfloat16_support_; in Normalize() local
82 BFloat16Normalization normalization(&bfloat16_support_); in Normalize()