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_normalization.h | 30 explicit BFloat16Normalization(const BFloat16Support* bfloat16_support) in BFloat16Normalization() argument
31 : bfloat16_support_(bfloat16_support) {} in BFloat16Normalization()
|
| D | bfloat16_conversion_folding.cc | 34 HloComputation* computation, const BFloat16Support* bfloat16_support, in BFloat16ConversionFoldingVisitor() argument
37 bfloat16_support_(bfloat16_support), in BFloat16ConversionFoldingVisitor()
46 const BFloat16Support* bfloat16_support, in Run() argument
48 BFloat16ConversionFoldingVisitor visitor(computation, bfloat16_support, in Run()
|
| D | bfloat16_propagation.h | 65 explicit BFloat16Propagation(const BFloat16Support* bfloat16_support);
|
| D | bfloat16_normalization.cc | 39 const BFloat16Support* bfloat16_support, in BFloat16NormalizationVisitor() argument
42 bfloat16_support_(bfloat16_support), in BFloat16NormalizationVisitor()
|
| D | bfloat16_propagation.cc | 36 const BFloat16Support* bfloat16_support) in BFloat16Propagation() argument
37 : bfloat16_support_(bfloat16_support) {} in BFloat16Propagation()
|
| D | BUILD | 73 name = "bfloat16_support",
74 srcs = ["bfloat16_support.cc"],
75 hdrs = ["bfloat16_support.h"],
86 ":bfloat16_support",
102 ":bfloat16_support",
121 ":bfloat16_support",
139 ":bfloat16_support",
160 ":bfloat16_support",
182 ":bfloat16_support",
|
| D | bfloat16_propagation_test.cc | 67 TestBFloat16Support bfloat16_support; in PropagatePrecision() local
68 BFloat16Propagation propagation(&bfloat16_support); in PropagatePrecision()
|