Home
last modified time | relevance | path

Searched refs:instr_index_groups (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5609 std::vector<std::vector<int>> instr_index_groups = in EmitReductionFromOrToContiguousDimensions() local
5612 VLOG(2) << StrCat("Generate in ", instr_index_groups.size(), " groups for ", in EmitReductionFromOrToContiguousDimensions()
5648 for (size_t i = 0; i < instr_index_groups.size(); ++i) { in EmitReductionFromOrToContiguousDimensions()
5658 EmitIRForReduction(unnested_hlo, instr_index_groups[i], fused_computation, in EmitReductionFromOrToContiguousDimensions()
5668 llvm_ir::AddRangeMetadata(0, instr_index_groups.size(), in EmitReductionFromOrToContiguousDimensions()
5682 /*y=*/static_cast<int64>(instr_index_groups.size()), in EmitReductionFromOrToContiguousDimensions()