Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h279 HloInstruction* reduce_inst, int reduce_idx,
Dir_emitter_unnested.cc2643 HloInstruction* unnested_hlo, HloInstruction* reduce_inst, int reduce_idx, in EmitPrologueForOneReduction() argument
2670 "partial_reduction_result." + llvm::Twine(reduce_idx)); in EmitPrologueForOneReduction()