Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc167 if (!bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleMultipleOutputs()
189 if (!bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleMultipleOutputs()
312 !bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleInstruction()
346 bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in HandleInstruction()
Dbfloat16_support.h31 virtual bool SupportsBF16Operand(const HloInstruction& hlo,
Dbfloat16_normalization.h75 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function
Dbfloat16_support.cc23 bool BFloat16Support::SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::BFloat16Support
Dbfloat16_conversion_folding.cc118 bfloat16_support_->SupportsBF16Operand(*hlo, i)) { in TryFoldBF16Conversions()
Dbfloat16_conversion_folding_test.cc35 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::TestBFloat16Support
Dbfloat16_normalization_test.cc37 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::TestBFloat16Support
Dbfloat16_propagation_test.cc39 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::TestBFloat16Support