Searched defs:bfloat16_support (Results 1 – 6 of 6) sorted by relevance
36 explicit BFloat16ConversionFolding(const BFloat16Support* bfloat16_support) in BFloat16ConversionFolding()
32 HloComputation* computation, const BFloat16Support* bfloat16_support) in BFloat16ConversionFoldingVisitor()41 const BFloat16Support* bfloat16_support) { in Run()
30 explicit BFloat16Normalization(const BFloat16Support* bfloat16_support) in BFloat16Normalization()
33 const BFloat16Support* bfloat16_support) in BFloat16NormalizationVisitor()39 const BFloat16Support* bfloat16_support) { in Run()
34 const BFloat16Support* bfloat16_support) in BFloat16Propagation()
67 TestBFloat16Support bfloat16_support; in PropagatePrecision() local