Home
last modified time | relevance | path

Searched defs:instr_index_group (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4216 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()