Home
last modified time | relevance | path

Searched defs:SupportsBF16Output (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_support.cc44 bool BFloat16Support::SupportsBF16Output(const HloInstruction& hlo) const { in SupportsBF16Output() function in xla::BFloat16Support
Dbfloat16_normalization.h87 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function
Dbfloat16_conversion_folding_test.cc47 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::TestBFloat16Support
Dbfloat16_normalization_test.cc54 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::TestBFloat16Support
Dbfloat16_propagation_test.cc45 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::TestBFloat16Support
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc213 bool SupportsBF16Output(const HloInstruction& hlo) const override { in SupportsBF16Output() function in xla::gpu::__anon4f55bdfe0111::GpuBfloat16Support