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