Searched defs:elemental_emitter (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ops.cc | 118 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl()
162 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlace()
172 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | fused_ir_emitter.h | 57 ElementalIrEmitter* elemental_emitter) in FusedIrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 692 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, in HandleFusion() local
|
| D | ir_emitter_unnested.cc | 513 GpuElementalIrEmitter elemental_emitter( in HandleFusion() local
575 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2103 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2113 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2939 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
|