Searched refs:dot_result_index (Results 1 – 2 of 2) sorted by relevance
241 const llvm_ir::IrArray::Index& dot_result_index);
2173 const llvm_ir::IrArray::Index& dot_result_index) { in EmitElementalDot() argument2186 llvm::Type* index_type = dot_result_index.GetType(); in EmitElementalDot()2213 lhs_multi_index.push_back(dot_result_index[i]); in EmitElementalDot()2223 dot_result_index[dim_numbers.rhs_batch_dimensions(i)]); in EmitElementalDot()2226 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()2476 &operand_to_generator](const IrArray::Index& dot_result_index) in MakeElementGenerator()2478 return EmitElementalDot(hlo, operand_to_generator, dot_result_index); in MakeElementGenerator()