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.cc142 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in TryFoldBF16Conversions()
181 !bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in DefaultAction()
199 if (!bfloat16_support_->SupportsMixedPrecisions(*crs)) { in HandleAllReduce()
Dbfloat16_support.h40 virtual bool SupportsMixedPrecisions(const HloInstruction& hlo) const;
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_normalization.cc303 if (bfloat16_support_->SupportsMixedPrecisions(*hlo)) { in HandleMultipleOutputs()
320 (bfloat16_support_->SupportsMixedPrecisions(*hlo) || f32_count == 0 || in HandleMultipleOutputs()
449 if (bfloat16_support_->SupportsMixedPrecisions(*hlo) || bf16_count == 0 || in HandleInstruction()
Dbfloat16_conversion_folding_test.cc58 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override { in SupportsMixedPrecisions() function in xla::TestBFloat16Support
Dbfloat16_propagation.cc42 if (!bfloat16_support_->SupportsMixedPrecisions(*fusion)) { in DetermineFusionComputationPrecision()
453 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo) && in InstructionIsCandidateForBF16Output()
Dbfloat16_normalization_test.cc65 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