Searched refs:elemental_emitter (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.h | 57 explicit FusedIrEmitter(ElementalIrEmitter* elemental_emitter) in FusedIrEmitter() argument
58 : elemental_emitter_(elemental_emitter), in FusedIrEmitter()
59 b_(elemental_emitter->b()), in FusedIrEmitter()
60 module_(elemental_emitter->module()) {} in FusedIrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 580 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion() local
582 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
|
| D | ir_emitter_unnested.cc | 1794 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitLoopFusionFromMlir() local
1796 FusedIrEmitter fused_emitter(&elemental_emitter); in EmitLoopFusionFromMlir()
2031 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion() local
2048 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
3727 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildFusedInitializerThunkForMlir() local
3731 FusedIrEmitter fused_emitter(&elemental_emitter); in BuildFusedInitializerThunkForMlir()
5615 absl::optional<GpuElementalIrEmitter> elemental_emitter; in EmitReductionFromOrToContiguousDimensions() local
5622 elemental_emitter.emplace(hlo_module_config_, in EmitReductionFromOrToContiguousDimensions()
5625 optional_fused_emitter.emplace(&*elemental_emitter); in EmitReductionFromOrToContiguousDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2067 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2068 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
2077 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2078 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
3218 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
3220 hlo, elemental_emitter.MakeElementGenerator(hlo, operand_to_generator)); in DefaultAction()
|