Searched defs:instr_index_group (Results 1 – 1 of 1) sorted by relevance
4216 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group, in EmitPrologueForReduction()4379 absl::Span<const int> instr_index_group, in EmitEpilogueForReduction()4596 absl::Span<const int> instr_index_group, HloComputation* fused_computation, in EmitTileElementForReduction()5396 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group, in EmitIRForReduction()