Searched refs:HloElementTypeConverter (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_element_type_converter_test.cc | 43 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
59 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
78 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
106 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
130 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
157 HloElementTypeConverter type_converter(BF16, F32); in TEST_F()
188 HloElementTypeConverter converter(BF16, F32); in TEST_F()
|
| D | hlo_element_type_converter.h | 28 class HloElementTypeConverter : public HloModulePass {
32 HloElementTypeConverter(PrimitiveType eliminate_type,
|
| D | hlo_element_type_converter.cc | 108 HloElementTypeConverter::HloElementTypeConverter( in HloElementTypeConverter() function in xla::HloElementTypeConverter
114 StatusOr<bool> HloElementTypeConverter::Run(HloModule* module) { in Run()
|
| D | hlo_evaluator_test.cc | 65 HloElementTypeConverter(F32, BF16).Run(m_.get()).ValueOrDie();
76 HloElementTypeConverter(F32, BF16).Run(m_.get()).ValueOrDie();
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 319 pipeline.AddPass<HloElementTypeConverter>(BF16, F32); in RunHloPassesThroughLayoutAssn()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 194 pipeline.AddPass<HloElementTypeConverter>(BF16, F32); in OptimizeHloModule()
|