sparse_matmul_op.cc | 657 const auto* nr2 = right_ptrs[nindex.k2]; local 661 MulAdd3Way128(nl1, nl2, nl3, &nr1, &nr2, &nr3, &nout); 665 MulAdd3Way(nl1, nl2, nl3, &nr1, &nr2, &nr3, &nout); 676 ScalarMulAdd3Way(nsl1, nsl2, nsl3, &nr1, &nr2, &nr3, &nout); [all...] |