Lines Matching refs:B
101 float alpha, RsAllocation A, RsAllocation B, in nScriptIntrinsicBLAS_Single() argument
106 RsAllocation in_allocs[3] = {A, B, C}; in nScriptIntrinsicBLAS_Single()
115 double alpha, RsAllocation A, RsAllocation B, in nScriptIntrinsicBLAS_Double() argument
120 RsAllocation in_allocs[3] = {A, B, C}; in nScriptIntrinsicBLAS_Double()
128 float alphaX, float alphaY, RsAllocation A, RsAllocation B, in nScriptIntrinsicBLAS_Complex() argument
133 RsAllocation in_allocs[3] = {A, B, C}; in nScriptIntrinsicBLAS_Complex()
141 double alphaX, double alphaY, RsAllocation A, RsAllocation B, in nScriptIntrinsicBLAS_Z() argument
146 RsAllocation in_allocs[3] = {A, B, C}; in nScriptIntrinsicBLAS_Z()
154 RsAllocation A, int a_offset, RsAllocation B, int b_offset, in nScriptIntrinsicBLAS_BNNM() argument
167 RsAllocation in_allocs[3] = {A, B, C}; in nScriptIntrinsicBLAS_BNNM()
1195 sp<Allocation> A, sp<Allocation> B, sp<Allocation> C) { in validateL3() argument
1198 (B != nullptr && !B->getType()->getElement()->isCompatible(e)) || in validateL3()
1210 if ((A == nullptr && B != nullptr) || (A != nullptr && B == nullptr)) { in validateL3()
1213 if (B != nullptr) { in validateL3()
1218 aM = B->getType()->getY(); in validateL3()
1219 aN = B->getType()->getX(); in validateL3()
1231 if (B != nullptr) { in validateL3()
1233 bN = B->getType()->getY(); in validateL3()
1234 bM = B->getType()->getX(); in validateL3()
1236 bM = B->getType()->getY(); in validateL3()
1237 bN = B->getType()->getX(); in validateL3()
1241 if (A != nullptr && B != nullptr && C != nullptr) { in validateL3()
1253 } else if (A != nullptr && B != nullptr) { in validateL3()
1263 sp<Allocation> A, sp<Allocation> B, float beta, sp<Allocation> C) { in SGEMM() argument
1264 validateL3(mRS, Element::F32(mRS), TransA, TransB, 0, A, B, C); in SGEMM()
1275 N = B->getType()->getY(); in SGEMM()
1277 N = B->getType()->getX(); in SGEMM()
1281 alpha, A->getID(), B->getID(), in SGEMM()
1286 sp<Allocation> A, sp<Allocation> B, double beta, sp<Allocation> C) { in DGEMM() argument
1287 validateL3(mRS, Element::F64(mRS), TransA, TransB, 0, A, B, C); in DGEMM()
1297 N = B->getType()->getY(); in DGEMM()
1299 N = B->getType()->getX(); in DGEMM()
1303 alpha, A->getID(), B->getID(), in DGEMM()
1308 sp<Allocation> A, sp<Allocation> B, Float2 beta, sp<Allocation> C) { in CGEMM() argument
1309 validateL3(mRS, Element::F32_2(mRS), TransA, TransB, 0, A, B, C); in CGEMM()
1319 N = B->getType()->getY(); in CGEMM()
1321 N = B->getType()->getX(); in CGEMM()
1325 alpha.x, alpha.y, A->getID(), B->getID(), in CGEMM()
1330 … sp<Allocation> A, sp<Allocation> B, Double2 beta, sp<Allocation> C) { in ZGEMM() argument
1331 validateL3(mRS, Element::F64_2(mRS), TransA, TransB, 0, A, B, C); in ZGEMM()
1341 N = B->getType()->getY(); in ZGEMM()
1343 N = B->getType()->getX(); in ZGEMM()
1347 alpha.x, alpha.y, A->getID(), B->getID(), in ZGEMM()
1352 sp<Allocation> A, sp<Allocation> B, float beta, sp<Allocation> C) { in SSYMM() argument
1357 validateL3(mRS, Element::F32(mRS), 0, 0, Side, A, B, C); in SSYMM()
1360 alpha, A->getID(), B->getID(), in SSYMM()
1365 sp<Allocation> A, sp<Allocation> B, double beta, sp<Allocation> C) { in DSYMM() argument
1369 validateL3(mRS, Element::F64(mRS), 0, 0, Side, A, B, C); in DSYMM()
1372 alpha, A->getID(), B->getID(), in DSYMM()
1377 sp<Allocation> A, sp<Allocation> B, Float2 beta, sp<Allocation> C) { in CSYMM() argument
1381 validateL3(mRS, Element::F32_2(mRS), 0, 0, Side, A, B, C); in CSYMM()
1384 alpha.x, alpha.y, A->getID(), B->getID(), in CSYMM()
1389 … sp<Allocation> A, sp<Allocation> B, Double2 beta, sp<Allocation> C) { in ZSYMM() argument
1393 validateL3(mRS, Element::F64_2(mRS), 0, 0, Side, A, B, C); in ZSYMM()
1396 alpha.x, alpha.y, A->getID(), B->getID(), in ZSYMM()
1461 sp<Allocation> A, sp<Allocation> B, sp<Allocation> C) { in validateSYR2K() argument
1463 !B->getType()->getElement()->isCompatible(e) || in validateSYR2K()
1481 …if (A->getType()->getX() != B->getType()->getX() || A->getType()->getY() != B->getType()->getY()) { in validateSYR2K()
1487 sp<Allocation> A, sp<Allocation> B, float beta, sp<Allocation> C) { in SSYR2K() argument
1488 validateSYR2K(mRS, Element::F32(mRS), Trans, A, B, C); in SSYR2K()
1497 alpha, A->getID(), B->getID(), in SSYR2K()
1502 … sp<Allocation> A, sp<Allocation> B, double beta, sp<Allocation> C) { in DSYR2K() argument
1503 validateSYR2K(mRS, Element::F64(mRS), Trans, A, B, C); in DSYR2K()
1512 alpha, A->getID(), B->getID(), in DSYR2K()
1517 … sp<Allocation> A, sp<Allocation> B, Float2 beta, sp<Allocation> C) { in CSYR2K() argument
1518 validateSYR2K(mRS, Element::F32_2(mRS), Trans, A, B, C); in CSYR2K()
1527 alpha.x, alpha.y, A->getID(), B->getID(), in CSYR2K()
1532 … sp<Allocation> A, sp<Allocation> B, Double2 beta, sp<Allocation> C) { in ZSYR2K() argument
1533 validateSYR2K(mRS, Element::F64_2(mRS), Trans, A, B, C); in ZSYR2K()
1542 alpha.x, alpha.y, A->getID(), B->getID(), in ZSYR2K()
1547 sp<Allocation> A, sp<Allocation> B) { in validateTRMM() argument
1550 !B->getType()->getElement()->isCompatible(e)) { in validateTRMM()
1560 bM = B->getType()->getY(); in validateTRMM()
1561 bN = B->getType()->getX(); in validateTRMM()
1574 float alpha, sp<Allocation> A, sp<Allocation> B) { in STRMM() argument
1575 validateTRMM(mRS, Element::F32(mRS), Side, TransA, A, B); in STRMM()
1578 B->getType()->getY(), B->getType()->getX(), 0, in STRMM()
1579 alpha, A->getID(), B->getID(), 0.f, 0, 0, 0, 0, 0); in STRMM()
1583 double alpha, sp<Allocation> A, sp<Allocation> B) { in DTRMM() argument
1584 validateTRMM(mRS, Element::F64(mRS), Side, TransA, A, B); in DTRMM()
1587 B->getType()->getY(), B->getType()->getX(), 0, in DTRMM()
1588 alpha, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0); in DTRMM()
1592 Float2 alpha, sp<Allocation> A, sp<Allocation> B) { in CTRMM() argument
1593 validateTRMM(mRS, Element::F32_2(mRS), Side, TransA, A, B); in CTRMM()
1596 B->getType()->getY(), B->getType()->getX(), 0, in CTRMM()
1597 alpha.x, alpha.y, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0, 0); in CTRMM()
1601 Double2 alpha, sp<Allocation> A, sp<Allocation> B) { in ZTRMM() argument
1602 validateTRMM(mRS, Element::F64_2(mRS), Side, TransA, A, B); in ZTRMM()
1605 B->getType()->getY(), B->getType()->getX(), 0, in ZTRMM()
1606 alpha.x, alpha.y, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0, 0); in ZTRMM()
1610 sp<Allocation> A, sp<Allocation> B) { in validateTRSM() argument
1613 !B->getType()->getElement()->isCompatible(e)) { in validateTRSM()
1623 bM = B->getType()->getY(); in validateTRSM()
1624 bN = B->getType()->getX(); in validateTRSM()
1639 float alpha, sp<Allocation> A, sp<Allocation> B) { in STRSM() argument
1640 validateTRSM(mRS, Element::F32(mRS), Side, TransA, A, B); in STRSM()
1643 B->getType()->getY(), B->getType()->getX(), 0, in STRSM()
1644 alpha, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0); in STRSM()
1648 double alpha, sp<Allocation> A, sp<Allocation> B) { in DTRSM() argument
1649 validateTRSM(mRS, Element::F64(mRS), Side, TransA, A, B); in DTRSM()
1652 B->getType()->getY(), B->getType()->getX(), 0, in DTRSM()
1653 alpha, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0); in DTRSM()
1657 Float2 alpha, sp<Allocation> A, sp<Allocation> B) { in CTRSM() argument
1658 validateTRSM(mRS, Element::F32_2(mRS), Side, TransA, A, B); in CTRSM()
1661 B->getType()->getY(), B->getType()->getX(), 0, in CTRSM()
1662 alpha.x, alpha.y, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0, 0); in CTRSM()
1666 Double2 alpha, sp<Allocation> A, sp<Allocation> B) { in ZTRSM() argument
1667 validateTRSM(mRS, Element::F64_2(mRS), Side, TransA, A, B); in ZTRSM()
1670 B->getType()->getY(), B->getType()->getX(), 0, in ZTRSM()
1671 alpha.x, alpha.y, A->getID(), B->getID(), 0, 0, 0, 0, 0, 0, 0); in ZTRSM()
1675 sp<Allocation> A, sp<Allocation> B, sp<Allocation> C) { in validateHEMM() argument
1677 !B->getType()->getElement()->isCompatible(e) || in validateHEMM()
1687 if ((Side == RsBlasLeft && adim != (int)B->getType()->getY()) || in validateHEMM()
1688 (Side == RsBlasRight && adim != (int)B->getType()->getX())) { in validateHEMM()
1691 if (B->getType()->getX() != C->getType()->getX() || in validateHEMM()
1692 B->getType()->getY() != C->getType()->getY()) { in validateHEMM()
1698 sp<Allocation> A, sp<Allocation> B, Float2 beta, sp<Allocation> C) { in CHEMM() argument
1699 validateHEMM(mRS, Element::F32_2(mRS), Side, A, B, C); in CHEMM()
1703 alpha.x, alpha.y, A->getID(), B->getID(), in CHEMM()
1708 … sp<Allocation> A, sp<Allocation> B, Double2 beta, sp<Allocation> C) { in ZHEMM() argument
1709 validateHEMM(mRS, Element::F64_2(mRS), Side, A, B, C); in ZHEMM()
1713 alpha.x, alpha.y, A->getID(), B->getID(), in ZHEMM()
1772 sp<Allocation> A, sp<Allocation> B, sp<Allocation> C) { in validateHER2K() argument
1774 !B->getType()->getElement()->isCompatible(e) || in validateHER2K()
1794 …if (A->getType()->getX() != B->getType()->getX() || A->getType()->getY() != B->getType()->getY()) { in validateHER2K()
1800 sp<Allocation> A, sp<Allocation> B, float beta, sp<Allocation> C) { in CHER2K() argument
1801 validateHER2K(mRS, Element::F32_2(mRS), Trans, A, B, C); in CHER2K()
1810 alpha.x, alpha.y, A->getID(), B->getID(), in CHER2K()
1815 … sp<Allocation> A, sp<Allocation> B, double beta, sp<Allocation> C) { in ZHER2K() argument
1816 validateHER2K(mRS, Element::F64_2(mRS), Trans, A, B, C); in ZHER2K()
1825 alpha.x, alpha.y, A->getID(), B->getID(), in ZHER2K()
1831 void ScriptIntrinsicBLAS::BNNM(sp<Allocation> A, int a_offset, sp<Allocation> B, int b_offset, in BNNM() argument
1833 validateL3(mRS, Element::U8(mRS), RsBlasNoTrans, RsBlasTrans, 0, A, B, C); in BNNM()
1843 N = B->getType()->getY(); in BNNM()
1847 B->getID(), b_offset, C->getID(), c_offset, c_mult); in BNNM()