Home
last modified time | relevance | path

Searched refs:SupportsMixedPrecisions (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding.cc133 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in TryFoldBF16Conversions()
170 !bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in DefaultAction()
188 if (!bfloat16_support_->SupportsMixedPrecisions(*crs)) { in HandleAllReduce()
Dbfloat16_support.h40 virtual bool SupportsMixedPrecisions(const HloInstruction& hlo) const;
Dbfloat16_normalization.cc192 if (bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in HandleMultipleOutputs()
208 (bfloat16_support_->SupportsMixedPrecisions(*hlo) || f32_count == 0 || in HandleMultipleOutputs()
331 if (bfloat16_support_->SupportsMixedPrecisions(*hlo) || bf16_count == 0 || in HandleInstruction()
Dbfloat16_normalization.h84 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override { in SupportsMixedPrecisions() function
Dbfloat16_support.cc61 bool BFloat16Support::SupportsMixedPrecisions(const HloInstruction& hlo) const { in SupportsMixedPrecisions() function in xla::BFloat16Support
Dbfloat16_propagation.cc40 if (!bfloat16_support_->SupportsMixedPrecisions(*fusion)) { in DetermineFusionComputationPrecision()
390 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo) && in InstructionIsCandidateForBF16Output()
Dbfloat16_conversion_folding_test.cc58 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override { in SupportsMixedPrecisions() function in xla::TestBFloat16Support
Dbfloat16_normalization_test.cc63 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override { in SupportsMixedPrecisions() function in xla::TestBFloat16Support
Dbfloat16_propagation_test.cc48 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override { in SupportsMixedPrecisions() function in xla::TestBFloat16Support