Searched defs:elemental_emitter (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 140 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl()
188 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlace()
199 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | fused_ir_emitter.h | 63 ElementalIrEmitter* elemental_emitter) in FusedIrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2153 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2161 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2660 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleRng() local
3034 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 726 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion() local
|
| D | ir_emitter_unnested.cc | 653 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion() local
1999 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildInitializerThunk() local
2644 KernelCodegenInfo* kernel_info, GpuElementalIrEmitter* elemental_emitter, in EmitPrologueForOneReduction()
2708 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in EmitPrologueForReduction() local
|