Searched refs:FusedIrEmitter (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 43 Status FusedIrEmitter::DefaultAction(const HloInstruction* hlo) { in DefaultAction()
82 Status FusedIrEmitter::HandleConstant(const HloInstruction* constant) { in HandleConstant()
111 Status FusedIrEmitter::HandleGetTupleElement( in HandleGetTupleElement()
116 Status FusedIrEmitter::HandleParameter(const HloInstruction* parameter) { in HandleParameter()
123 Status FusedIrEmitter::HandleTuple(const HloInstruction* tuple) { in HandleTuple()
144 bool FusedIrEmitter::IsFusedIrEmitterInefficient( in IsFusedIrEmitterInefficient()
160 StatusOr<FusedIrEmitter::IndexedGenerator> FusedIrEmitter::GetGenerator( in GetGenerator()
|
| D | dynamic_update_slice_util.h | 77 FusedIrEmitter* fused_emitter,
84 FusedIrEmitter* fused_emitter,
|
| D | fused_ir_emitter.h | 53 class FusedIrEmitter {
57 explicit FusedIrEmitter(ElementalIrEmitter* elemental_emitter) in FusedIrEmitter() function
|
| D | dynamic_update_slice_util.cc | 193 FusedIrEmitter* fused_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl()
238 FusedIrEmitter* fused_emitter, in EmitFusedDynamicUpdateSliceInPlace()
247 FusedIrEmitter* fused_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 589 HloComputation* fused_computation, FusedIrEmitter* fused_emitter,
602 HloComputation* fused_computation, FusedIrEmitter* fused_emitter,
621 FusedIrEmitter* fused_emitter,
|
| D | ir_emitter.h | 185 FusedIrEmitter* fused_emitter);
|
| D | fusion_merger.cc | 271 return FusedIrEmitter::IsFusedIrEmitterInefficient(/*consumer=*/user, in HandleFusion()
|
| D | multi_output_fusion.cc | 157 if (FusedIrEmitter::IsFusedIrEmitterInefficient(consumer, producer)) { in GetProducerConsumerMultiOutputFusionCandidates()
|
| D | ir_emitter.cc | 582 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
686 FusedIrEmitter* fused_emitter) { in BindFusionArguments()
|
| D | ir_emitter_unnested.cc | 1796 FusedIrEmitter fused_emitter(&elemental_emitter); in EmitLoopFusionFromMlir()
1901 FusedIrEmitter operand_fused_emitter(&operand_elemental_emitter); in HandleFusion()
1945 FusedIrEmitter scatter_fused_emitter(&scatter_elemental_emitter); in HandleFusion()
2048 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
3731 FusedIrEmitter fused_emitter(&elemental_emitter); in BuildFusedInitializerThunkForMlir()
4155 FusedIrEmitter fused_emitter(&elem_emitter); in EmitTileElementForFusion()
4217 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitPrologueForReduction()
4597 FusedIrEmitter* fused_emitter, in EmitTileElementForReduction()
5397 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitIRForReduction()
5616 absl::optional<FusedIrEmitter> optional_fused_emitter; in EmitReductionFromOrToContiguousDimensions()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 233 FusedIrEmitter* fused_emitter);
|
| D | ir_emitter.cc | 2068 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
2078 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion()
3323 FusedIrEmitter* fused_emitter) { in BindFusionArguments()
|