Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4606 InlinedVector<llvm_ir::ElementGenerator, 1> input_gens; in EmitTileElementForReduction() local
4620 input_gens.push_back(*fused_emitter->GetGenerator(hlo->operand(0))); in EmitTileElementForReduction()
4627 input_gens.push_back([&](const IrArray::Index& index) { in EmitTileElementForReduction()
4651 input_gens[i](num_partial_results > 1 ? index_without_linear in EmitTileElementForReduction()