Searched refs:elemental_emitter (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.h | 63 ElementalIrEmitter* elemental_emitter) in FusedIrEmitter() argument
67 elemental_emitter_(elemental_emitter), in FusedIrEmitter()
68 b_(elemental_emitter->b()), in FusedIrEmitter()
69 module_(elemental_emitter->module()) {} in FusedIrEmitter()
|
| D | dynamic_update_slice_util.cc | 140 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl() argument
170 elemental_emitter); in EmitFusedDynamicUpdateSliceInPlaceImpl()
188 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlace() argument
192 elemental_emitter, in EmitFusedDynamicUpdateSliceInPlace()
199 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace() argument
203 elemental_emitter, &launch_dimensions, b); in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | dynamic_update_slice_util.h | 81 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter,
89 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 726 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, module_, &b_, in HandleFusion() local
729 &elemental_emitter); in HandleFusion()
|
| D | ir_emitter_unnested.h | 281 GpuElementalIrEmitter* elemental_emitter,
|
| D | ir_emitter_unnested.cc | 653 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in HandleFusion() local
672 &elemental_emitter, launch_dimensions, &b_); in HandleFusion()
1999 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in BuildInitializerThunk() local
2004 &elemental_emitter); in BuildInitializerThunk()
2644 KernelCodegenInfo* kernel_info, GpuElementalIrEmitter* elemental_emitter, in EmitPrologueForOneReduction() argument
2678 elemental_emitter); in EmitPrologueForOneReduction()
2708 GpuElementalIrEmitter elemental_emitter(hlo_module_config_, in EmitPrologueForReduction() local
2728 &elemental_emitter, in EmitPrologueForReduction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2153 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2158 &elemental_emitter, &b_); in HandleFusion()
2161 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleFusion() local
2164 &elemental_emitter); in HandleFusion()
2660 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in HandleRng() local
2662 rng, elemental_emitter.MakeElementGenerator(rng, operand_to_generator))); in HandleRng()
3034 CpuElementalIrEmitter elemental_emitter(hlo_module_config_, this, module_); in DefaultAction() local
3036 hlo, elemental_emitter.MakeElementGenerator(hlo, operand_to_generator)); in DefaultAction()
|