Searched refs:instr_index_group (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 588 absl::Span<const int> instr_index_group,
601 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group,
611 absl::Span<const int> instr_index_group,
619 absl::Span<const int> instr_index_group,
|
| D | ir_emitter_unnested.cc | 4216 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group, in EmitPrologueForReduction() argument
4223 for (int index : instr_index_group) { in EmitPrologueForReduction()
4379 absl::Span<const int> instr_index_group, in EmitEpilogueForReduction() argument
4418 for (int index : instr_index_group) { in EmitEpilogueForReduction()
4596 absl::Span<const int> instr_index_group, HloComputation* fused_computation, in EmitTileElementForReduction() argument
4612 for (int index : instr_index_group) { in EmitTileElementForReduction()
5396 mlir::Operation* unnested_hlo, absl::Span<const int> instr_index_group, in EmitIRForReduction() argument
5402 for (auto index : instr_index_group) { in EmitIRForReduction()
5432 EmitPrologueForReduction(unnested_hlo, instr_index_group, fused_computation, in EmitIRForReduction()
5440 unnested_hlo, input_shape, instr_index_group, fused_computation, in EmitIRForReduction()
[all …]
|