Searched refs:bfloat16_support (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_conversion_folding.h | 36 explicit BFloat16ConversionFolding(const BFloat16Support* bfloat16_support) in BFloat16ConversionFolding() argument
37 : bfloat16_support_(bfloat16_support) {} in BFloat16ConversionFolding()
|
| D | bfloat16_conversion_folding.cc | 32 HloComputation* computation, const BFloat16Support* bfloat16_support) in BFloat16ConversionFoldingVisitor() argument
33 : computation_(computation), bfloat16_support_(bfloat16_support) {} in BFloat16ConversionFoldingVisitor()
41 const BFloat16Support* bfloat16_support) { in Run() argument
42 BFloat16ConversionFoldingVisitor visitor(computation, bfloat16_support); in Run()
|
| D | bfloat16_normalization.h | 30 explicit BFloat16Normalization(const BFloat16Support* bfloat16_support) in BFloat16Normalization() argument
31 : bfloat16_support_(bfloat16_support) {} in BFloat16Normalization()
|
| D | bfloat16_normalization.cc | 33 const BFloat16Support* bfloat16_support) in BFloat16NormalizationVisitor() argument
34 : computation_(computation), bfloat16_support_(bfloat16_support) {} in BFloat16NormalizationVisitor()
39 const BFloat16Support* bfloat16_support) { in Run() argument
40 BFloat16NormalizationVisitor visitor(computation, bfloat16_support); in Run()
|
| D | bfloat16_propagation.h | 65 explicit BFloat16Propagation(const BFloat16Support* bfloat16_support);
|
| D | bfloat16_propagation.cc | 34 const BFloat16Support* bfloat16_support) in BFloat16Propagation() argument
35 : bfloat16_support_(bfloat16_support) {} in BFloat16Propagation()
|
| D | BUILD | 52 name = "bfloat16_support",
53 srcs = ["bfloat16_support.cc"],
54 hdrs = ["bfloat16_support.h"],
65 ":bfloat16_support",
80 ":bfloat16_support",
99 ":bfloat16_support",
115 ":bfloat16_support",
136 ":bfloat16_support",
157 ":bfloat16_support",
|
| D | bfloat16_propagation_test.cc | 67 TestBFloat16Support bfloat16_support; in PropagatePrecision() local
68 BFloat16Propagation propagation(&bfloat16_support); in PropagatePrecision()
|