Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_support.h35 virtual bool SupportsBF16Output(const HloInstruction& hlo) const;
Dbfloat16_normalization.h80 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function
Dbfloat16_support.cc43 bool BFloat16Support::SupportsBF16Output(const HloInstruction& hlo) const { in SupportsBF16Output() function in xla::BFloat16Support
Dbfloat16_normalization.cc175 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in HandleMultipleOutputs()
321 !bfloat16_support_->SupportsBF16Output(*hlo)) { in HandleInstruction()
Dbfloat16_conversion_folding.cc131 bfloat16_support_->SupportsBF16Output(*hlo); in TryFoldBF16Conversions()
Dbfloat16_conversion_folding_test.cc47 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::TestBFloat16Support
Dbfloat16_normalization_test.cc53 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::TestBFloat16Support
Dbfloat16_propagation.cc372 if (!bfloat16_support_->SupportsBF16Output(*hlo)) { in DetermineInstructionPrecision()
Dbfloat16_propagation_test.cc44 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::TestBFloat16Support