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 | 50 explicit FusedIrEmitter(ElementalIrEmitter& elemental_emitter) in FusedIrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 562 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion() local
|
| D | ir_emitter_unnested.cc | 1753 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitLoopFusion() local
3278 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in BuildFusedInitializerThunk() local
4944 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitUnnestedReduction() local
5172 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in EmitDynamicUpdateSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2178 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2188 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
3422 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
|