Searched refs:input_gens (Results 1 – 1 of 1) sorted by relevance
4606 InlinedVector<llvm_ir::ElementGenerator, 1> input_gens; in EmitTileElementForReduction() local4620 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()