Searched defs:fused_emitter (Results 1 – 4 of 4) sorted by relevance
193 FusedIrEmitter* fused_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl()238 FusedIrEmitter* fused_emitter, in EmitFusedDynamicUpdateSliceInPlace()247 FusedIrEmitter* fused_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace()
582 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local686 FusedIrEmitter* fused_emitter) { in BindFusionArguments()
1796 FusedIrEmitter fused_emitter(&elemental_emitter); in EmitLoopFusionFromMlir() local2048 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local3731 FusedIrEmitter fused_emitter(&elemental_emitter); in BuildFusedInitializerThunkForMlir() local4155 FusedIrEmitter fused_emitter(&elem_emitter); in EmitTileElementForFusion() local4217 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitPrologueForReduction()4597 FusedIrEmitter* fused_emitter, in EmitTileElementForReduction()5397 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitIRForReduction()5617 FusedIrEmitter* fused_emitter = nullptr; in EmitReductionFromOrToContiguousDimensions() local5740 FusedIrEmitter fused_emitter(&elem_emitter); in EmitElementForInputFusibleSlices() local
2068 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local2078 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local3323 FusedIrEmitter* fused_emitter) { in BindFusionArguments()