| /frameworks/rs/cpp/ |
| D | ScriptIntrinsicBLAS.cpp | 51 int TransA, int TransB, int Side, int Uplo, int Diag, in setUpBLASCall() argument
61 call.transB = (RsBlasTranspose)TransB; in setUpBLASCall()
108 int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Single() argument
111 RsBlasCall call = setUpBLASCall(SINGLE, func, TransA, TransB, Side, Uplo, Diag, in nScriptIntrinsicBLAS_Single()
122 int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Double() argument
125 RsBlasCall call = setUpBLASCall(DOUBLE, func, TransA, TransB, Side, Uplo, Diag, in nScriptIntrinsicBLAS_Double()
135 int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Complex() argument
138 RsBlasCall call = setUpBLASCall(SINGLE_COMPLEX, func, TransA, TransB, Side, Uplo, Diag, in nScriptIntrinsicBLAS_Complex()
148 int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Z() argument
151 RsBlasCall call = setUpBLASCall(DOUBLE_COMPLEX, func, TransA, TransB, Side, Uplo, Diag, in nScriptIntrinsicBLAS_Z()
[all …]
|
| D | rsCppStructs.h | 3234 void SGEMM(RsBlasTranspose TransA, RsBlasTranspose TransB, float alpha, const sp<Allocation>& A,
3252 … void DGEMM(RsBlasTranspose TransA, RsBlasTranspose TransB, double alpha, const sp<Allocation>& A,
3269 … void CGEMM(RsBlasTranspose TransA, RsBlasTranspose TransB, Float2 alpha, const sp<Allocation>& A,
3286 … void ZGEMM(RsBlasTranspose TransA, RsBlasTranspose TransB, Double2 alpha, const sp<Allocation>& A,
|
| /frameworks/rs/support/java/src/androidx/renderscript/ |
| D | RenderScript.java | 929 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Single() argument
933 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Single() argument
937 …rsnScriptIntrinsicBLAS_Single(mContext, mIncCon, id, func, TransA, TransB, Side, Uplo, Diag, M, N,… in nScriptIntrinsicBLAS_Single()
941 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Double() argument
945 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Double() argument
949 …rsnScriptIntrinsicBLAS_Double(mContext, mIncCon, id, func, TransA, TransB, Side, Uplo, Diag, M, N,… in nScriptIntrinsicBLAS_Double()
953 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Complex() argument
957 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Complex() argument
961 …rsnScriptIntrinsicBLAS_Complex(mContext, mIncCon, id, func, TransA, TransB, Side, Uplo, Diag, M, N… in nScriptIntrinsicBLAS_Complex()
965 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Z() argument
[all …]
|
| D | ScriptIntrinsicBLAS.java | 2862 …static void validateL3(Element e, int TransA, int TransB, int Side, Allocation A, Allocation B, Al… in validateL3() argument
2899 if (TransB == TRANSPOSE || TransB == CONJ_TRANSPOSE) { in validateL3()
2943 public void SGEMM(@Transpose int TransA, @Transpose int TransB, float alpha, Allocation A, in SGEMM() argument
2946 validateTranspose(TransB); in SGEMM()
2947 validateL3(Element.F32(mRS), TransA, TransB, 0, A, B, C); in SGEMM() local
2957 if (TransB != NO_TRANSPOSE) { in SGEMM()
2972 …mRS.nScriptIntrinsicBLAS_Single(getID(mRS), RsBlas_sgemm, TransA, TransB, 0, 0, 0, M, N, K, alpha… in SGEMM() local
2990 public void DGEMM(@Transpose int TransA, @Transpose int TransB, double alpha, Allocation A, in DGEMM() argument
2993 validateTranspose(TransB); in DGEMM()
2994 validateL3(Element.F64(mRS), TransA, TransB, 0, A, B, C); in DGEMM() local
[all …]
|
| /frameworks/base/rs/java/android/renderscript/ |
| D | RenderScript.java | 980 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Single() argument
984 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Single() argument
988 …rsnScriptIntrinsicBLAS_Single(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alpha… in nScriptIntrinsicBLAS_Single()
992 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Double() argument
996 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Double() argument
1000 …rsnScriptIntrinsicBLAS_Double(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alpha… in nScriptIntrinsicBLAS_Double()
1004 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Complex() argument
1008 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in nScriptIntrinsicBLAS_Complex() argument
1012 …rsnScriptIntrinsicBLAS_Complex(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alph… in nScriptIntrinsicBLAS_Complex()
1016 … int TransB, int Side, int Uplo, int Diag, int M, int N, int K, in rsnScriptIntrinsicBLAS_Z() argument
[all …]
|
| D | ScriptIntrinsicBLAS.java | 2259 …static void validateL3(Element e, int TransA, int TransB, int Side, Allocation A, Allocation B, Al… in validateL3() argument
2296 if (TransB == TRANSPOSE || TransB == CONJ_TRANSPOSE) { in validateL3()
2340 public void SGEMM(@Transpose int TransA, @Transpose int TransB, float alpha, Allocation A, in SGEMM() argument
2343 validateTranspose(TransB); in SGEMM()
2344 validateL3(Element.F32(mRS), TransA, TransB, 0, A, B, C); in SGEMM() local
2354 if (TransB != NO_TRANSPOSE) { in SGEMM()
2359 …mRS.nScriptIntrinsicBLAS_Single(getID(mRS), RsBlas_sgemm, TransA, TransB, 0, 0, 0, M, N, K, alpha… in SGEMM() local
2377 public void DGEMM(@Transpose int TransA, @Transpose int TransB, double alpha, Allocation A, in DGEMM() argument
2380 validateTranspose(TransB); in DGEMM()
2381 validateL3(Element.F64(mRS), TransA, TransB, 0, A, B, C); in DGEMM() local
[all …]
|
| /frameworks/rs/cpu_ref/ |
| D | rsCpuBLASDispatch.h | 294 const enum CBLAS_TRANSPOSE TransB, const int M, const int N,
324 const enum CBLAS_TRANSPOSE TransB, const int M, const int N,
354 const enum CBLAS_TRANSPOSE TransB, const int M, const int N,
384 const enum CBLAS_TRANSPOSE TransB, const int M, const int N,
|
| D | rsCpuIntrinsicBLAS.cpp | 148 enum CBLAS_TRANSPOSE TransB = (enum CBLAS_TRANSPOSE)call->transB; in walk_tiled_gemm() local
165 int nStride = (TransB == CblasNoTrans) ? 1 : ldb; in walk_tiled_gemm()
183 blasFunc(CblasRowMajor, TransA, TransB, in walk_tiled_gemm()
246 enum CBLAS_TRANSPOSE TransB = (enum CBLAS_TRANSPOSE)call->transB; in invokeForEach() local
643 cblas_sgemm(CblasRowMajor, TransA, TransB, call->M, call->N, call->K, call->alpha.f, in invokeForEach()
680 cblas_dgemm(CblasRowMajor, TransA, TransB, call->M, call->N, call->K, call->alpha.d, in invokeForEach()
716 … cblas_cgemm(CblasRowMajor, TransA, TransB, call->M, call->N, call->K, (void*)&call->alpha.c, in invokeForEach()
752 … cblas_zgemm(CblasRowMajor, TransA, TransB, call->M, call->N, call->K, (void*)&call->alpha.z, in invokeForEach()
|
| /frameworks/rs/support/jni/ |
| D | android_renderscript_RenderScript.cpp | 586 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Single() argument
593 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Single()
626 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Double() argument
633 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Double()
666 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Complex() argument
673 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Complex()
709 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Z() argument
716 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Z()
|
| /frameworks/base/rs/jni/ |
| D | android_renderscript_RenderScript.cpp | 621 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Single() argument
628 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Single()
654 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Double() argument
661 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Double()
687 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Complex() argument
694 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Complex()
722 jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K, in nScriptIntrinsicBLAS_Z() argument
729 call.transB = (RsBlasTranspose)TransB; in nScriptIntrinsicBLAS_Z()
|