Searched refs:dot_result_index (Results 1 – 2 of 2) sorted by relevance
203 const llvm_ir::IrArray::Index& dot_result_index);
2114 const llvm_ir::IrArray::Index& dot_result_index) { in EmitElementalDot() argument2127 llvm::Type* index_type = dot_result_index[0]->getType(); in EmitElementalDot()2154 lhs_multi_index.push_back(dot_result_index[i]); in EmitElementalDot()2164 dot_result_index[dim_numbers.rhs_batch_dimensions(i)]); in EmitElementalDot()2167 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()2435 &operand_to_generator](const IrArray::Index& dot_result_index) in MakeElementGenerator()2437 return EmitElementalDot(hlo, operand_to_generator, dot_result_index); in MakeElementGenerator()