Searched defs: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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 580 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion() local
|
| D | ir_emitter_unnested.cc | 1794 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitLoopFusionFromMlir() local
2031 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion() local
3727 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildFusedInitializerThunkForMlir() local
5615 absl::optional<GpuElementalIrEmitter> elemental_emitter; in EmitReductionFromOrToContiguousDimensions() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2067 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2077 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
3218 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
|