Searched defs:bfloat16_support (Results 1 – 6 of 6) sorted by relevance
33 HloComputation* computation, const BFloat16Support* bfloat16_support, in BFloat16ConversionFoldingVisitor()45 const BFloat16Support* bfloat16_support, in Run()
36 explicit BFloat16ConversionFolding(const BFloat16Support* bfloat16_support) in BFloat16ConversionFolding()
30 explicit BFloat16Normalization(const BFloat16Support* bfloat16_support) in BFloat16Normalization()
38 const BFloat16Support* bfloat16_support, in BFloat16NormalizationVisitor()
36 const BFloat16Support* bfloat16_support) in BFloat16Propagation()
68 TestBFloat16Support bfloat16_support; in PropagatePrecision() local