Searched defs:SupportsBF16Operand (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | bfloat16_support.cc | 23 bool BFloat16Support::SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::BFloat16Support |
| D | bfloat16_normalization.h | 82 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function |
| D | bfloat16_conversion_folding_test.cc | 35 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::TestBFloat16Support |
| D | bfloat16_normalization_test.cc | 37 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::TestBFloat16Support |
| D | bfloat16_propagation_test.cc | 40 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::TestBFloat16Support |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ | ||
| D | gpu_compiler.cc | 206 bool SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand() function in xla::gpu::__anon4f55bdfe0111::GpuBfloat16Support |