Searched refs:lhs_multi_index (Results 1 – 3 of 3) sorted by relevance
531 std::vector<llvm::Value*> lhs_multi_index = in HandleDot() local542 rhs_multi_index[i] = lhs_multi_index[i]; in HandleDot()553 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()579 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_array.GetShape(), in HandleDot()
432 std::vector<llvm::Value*> lhs_multi_index = in EmitNaiveLlvmIrGemm() local457 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()458 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_shape, in EmitNaiveLlvmIrGemm()
2152 std::vector<llvm::Value*> lhs_multi_index, rhs_multi_index; in EmitElementalDot() local2154 lhs_multi_index.push_back(dot_result_index[i]); in EmitElementalDot()2156 lhs_multi_index.insert(lhs_multi_index.begin() + lhs_contracting_dim, in EmitElementalDot()2158 IrArray::Index lhs_index(lhs_multi_index, hlo->operand(0)->shape(), in EmitElementalDot()