Searched refs:lhs_multi_index (Results 1 – 2 of 2) sorted by relevance
575 std::vector<llvm::Value*> lhs_multi_index = in EmitNaiveLlvmIrGemm() local600 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()601 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_shape, in EmitNaiveLlvmIrGemm()
2211 std::vector<llvm::Value*> lhs_multi_index, rhs_multi_index; in EmitElementalDot() local2213 lhs_multi_index.push_back(dot_result_index[i]); in EmitElementalDot()2215 lhs_multi_index.insert(lhs_multi_index.begin() + lhs_contracting_dim, in EmitElementalDot()2217 IrArray::Index lhs_index(lhs_multi_index, hlo->operand(0)->shape(), in EmitElementalDot()