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 | 575 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion() local
|
| D | ir_emitter_unnested.cc | 1745 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitLoopFusion() local
1965 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in EmitFusion() local
3411 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildFusedInitializerThunk() local
5126 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in EmitUnnestedReduction() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2072 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2082 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
3255 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
|