Searched refs:FusedIrEmitter (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 43 Status FusedIrEmitter::DefaultAction(HloInstruction* hlo) { in DefaultAction()
82 Status FusedIrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant()
104 Status FusedIrEmitter::HandleGetTupleElement( in HandleGetTupleElement()
144 Status FusedIrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter()
169 Status FusedIrEmitter::HandleTuple(HloInstruction* tuple) { in HandleTuple()
190 Status FusedIrEmitter::FinishVisit(HloInstruction* root) { in FinishVisit()
195 FusedIrEmitter::IndexedGenerator FusedIrEmitter::GetRootGenerator() const { in GetRootGenerator()
201 FusedIrEmitter::IndexedGenerator FusedIrEmitter::GetGenerator( in GetGenerator()
206 bool FusedIrEmitter::IsFusedIrEmitterInefficient( in IsFusedIrEmitterInefficient()
|
| D | fused_ir_emitter.h | 55 class FusedIrEmitter : public DfsHloVisitorWithDefault {
62 FusedIrEmitter(GeneratorForOperandIrArrays operand_arrays_generator, in FusedIrEmitter() function
|
| D | dynamic_update_slice_util.cc | 169 FusedIrEmitter fused_emitter(std::move(operand_arrays_generator), in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 125 if (FusedIrEmitter::IsFusedIrEmitterInefficient(consumer, producer)) { in ShouldFuse()
|
| D | ir_emitter.cc | 2163 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(fusion), in HandleFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_merger.cc | 259 return FusedIrEmitter::IsFusedIrEmitterInefficient(/*consumer=*/user, in HandleFusion()
|
| D | instruction_fusion.cc | 271 return !FusedIrEmitter::IsFusedIrEmitterInefficient(consumer, producer); in ShouldFuse()
|
| D | ir_emitter_unnested.cc | 592 FusedIrEmitter operand_fused_emitter( in HandleFusion()
613 FusedIrEmitter scatter_fused_emitter( in HandleFusion()
2003 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(hlo), in BuildInitializerThunk()
2494 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(hlo), in EmitTileElementForFusion()
2677 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(unnested_hlo), in EmitPrologueForOneReduction()
2885 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(unnested_hlo), in EmitTileElementForReduction()
|
| D | ir_emitter.cc | 728 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(fusion), in HandleFusion()
|