Searched refs:BFloat16Support (Results 1 – 11 of 11) sorted by relevance
23 bool BFloat16Support::SupportsBF16Operand(const HloInstruction& hlo, in SupportsBF16Operand()43 bool BFloat16Support::SupportsBF16Output(const HloInstruction& hlo) const { in SupportsBF16Output()61 bool BFloat16Support::SupportsMixedPrecisions(const HloInstruction& hlo) const { in SupportsMixedPrecisions()78 bool BFloat16Support::EffectiveOperandPrecisionIsOutputPrecision( in EffectiveOperandPrecisionIsOutputPrecision()132 bool BFloat16Support::EffectiveOperandPrecisionIsBF16( in EffectiveOperandPrecisionIsBF16()
24 class BFloat16Support {26 BFloat16Support() {} in BFloat16Support() function27 virtual ~BFloat16Support() {} in ~BFloat16Support()
30 explicit BFloat16Normalization(const BFloat16Support* bfloat16_support) in BFloat16Normalization()41 const BFloat16Support* bfloat16_support_;69 class BFloat16SupportForMixedPrecisionRemoval : public BFloat16Support {
36 explicit BFloat16ConversionFolding(const BFloat16Support* bfloat16_support) in BFloat16ConversionFolding()47 const BFloat16Support* bfloat16_support_;
65 explicit BFloat16Propagation(const BFloat16Support* bfloat16_support);206 const BFloat16Support* bfloat16_support_;
32 HloComputation* computation, const BFloat16Support* bfloat16_support) in BFloat16ConversionFoldingVisitor()41 const BFloat16Support* bfloat16_support) { in Run()63 const BFloat16Support* bfloat16_support_;
33 const BFloat16Support* bfloat16_support) in BFloat16NormalizationVisitor()39 const BFloat16Support* bfloat16_support) { in Run()75 const BFloat16Support* bfloat16_support_;
30 class TestBFloat16Support : public BFloat16Support {
32 class TestBFloat16Support : public BFloat16Support {
34 const BFloat16Support* bfloat16_support) in BFloat16Propagation()
34 class TestBFloat16Support : public BFloat16Support {