Home
last modified time | relevance | path

Searched defs:beta (Results 1 – 25 of 27) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DSoftmax.cpp51 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
83 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
99 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
114 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8Impl()
201 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8()
268 float beta = (input.type == OperandType::TENSOR_FLOAT16) in prepare() local
DLocalResponseNormalization.cpp51 int32_t radius, float bias, float alpha, float beta, in localResponseNormFloat32Impl()
86 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNorm()
107 _Float16 bias, _Float16 alpha, _Float16 beta, int32_t axis, in localResponseNorm()
DInstanceNormalization.cpp48 inline bool instanceNormNhwc(const T* inputData, const Shape& inputShape, T gamma, T beta, in instanceNormNhwc()
91 inline bool instanceNorm(const T* inputData, const Shape& inputShape, T gamma, T beta, T epsilon, in instanceNorm()
DLogSoftmax.cpp43 inline bool compute(const T* input, const Shape& shape, T beta, uint32_t axis, T* output) { in compute()
/frameworks/rs/cpp/
DScriptIntrinsicBLAS.cpp110 float beta, RsAllocation C, int incX, int incY, int KL, int KU) { in nScriptIntrinsicBLAS_Single()
124 double beta, RsAllocation C, int incX, int incY, int KL, int KU) { in nScriptIntrinsicBLAS_Double()
214 int incX, float beta, const sp<Allocation>& Y, int incY) { in SGEMV()
225 int incX, double beta, const sp<Allocation>& Y, int incY) { in DGEMV()
236 int incX, Float2 beta, const sp<Allocation>& Y, int incY) { in CGEMV()
247 int incX, Double2 beta, const sp<Allocation>& Y, int incY) { in ZGEMV()
258 … const sp<Allocation>& X, int incX, float beta, const sp<Allocation>& Y, int incY) { in SGBMV()
274 … const sp<Allocation>& X, int incX, double beta, const sp<Allocation>& Y, int incY) { in DGBMV()
290 … const sp<Allocation>& X, int incX, Float2 beta, const sp<Allocation>& Y, int incY) { in CGBMV()
306 … const sp<Allocation>& X, int incX, Double2 beta, const sp<Allocation>& Y, int incY) { in ZGBMV()
[all …]
/frameworks/ml/nn/runtime/test/specs/V1_3/
Dsoftmax_quant8_signed.mod.py19 beta = Float32Scalar("beta", 0.00001) # close to 0 variable
38 beta = Float32Scalar("beta", 1.) variable
/frameworks/ml/nn/runtime/test/specs/V1_0/
Dsoftmax_float_2.mod.py5 beta = Float32Scalar("beta", 1.) variable
Dsoftmax_quant8_2.mod.py5 beta = Float32Scalar("beta", 1.) variable
Dsoftmax_quant8_1.mod.py5 beta = Float32Scalar("beta", 0.00001) # close to 0 variable
Dsoftmax_float_1.mod.py5 beta = Float32Scalar("beta", 0.000001) variable
Dlocal_response_norm_float_4.mod.py6 beta = Float32Scalar("beta", .5) variable
Dlocal_response_norm_float_1.mod.py6 beta = Float32Scalar("beta", .5) variable
/frameworks/ml/nn/runtime/test/specs/V1_2/
Dlog_softmax.mod.py19 def test(input0, output0, input_data, beta, axis, output_data): argument
/frameworks/ml/nn/runtime/test/specs/V1_1/
Dsoftmax_float_1_relaxed.mod.py21 beta = Float32Scalar("beta", 0.000001) variable
Dsoftmax_float_2_relaxed.mod.py21 beta = Float32Scalar("beta", 1.) variable
Dlocal_response_norm_float_4_relaxed.mod.py22 beta = Float32Scalar("beta", .5) variable
Dlocal_response_norm_float_1_relaxed.mod.py22 beta = Float32Scalar("beta", .5) variable
/frameworks/base/rs/java/android/renderscript/
DScriptIntrinsicBLAS.java320 …anspose int TransA, float alpha, Allocation A, Allocation X, int incX, float beta, Allocation Y, i… in SGEMV()
324 …as_sgemv, TransA, 0, 0, 0, 0, M, N, 0, alpha, A.getID(mRS), X.getID(mRS), beta, Y.getID(mRS), incX… in SGEMV() local
342 …nspose int TransA, double alpha, Allocation A, Allocation X, int incX, double beta, Allocation Y, … in DGEMV()
346 …as_dgemv, TransA, 0, 0, 0, 0, M, N, 0, alpha, A.getID(mRS), X.getID(mRS), beta, Y.getID(mRS), incX… in DGEMV() local
364 …nspose int TransA, Float2 alpha, Allocation A, Allocation X, int incX, Float2 beta, Allocation Y, … in CGEMV()
386 …pose int TransA, Double2 alpha, Allocation A, Allocation X, int incX, Double2 beta, Allocation Y, … in ZGEMV()
417 …sA, int KL, int KU, float alpha, Allocation A, Allocation X, int incX, float beta, Allocation Y, i… in SGBMV()
425 …as_sgbmv, TransA, 0, 0, 0, 0, M, N, 0, alpha, A.getID(mRS), X.getID(mRS), beta, Y.getID(mRS), incX… in SGBMV() local
452 …A, int KL, int KU, double alpha, Allocation A, Allocation X, int incX, double beta, Allocation Y, … in DGBMV()
460 …as_dgbmv, TransA, 0, 0, 0, 0, M, N, 0, alpha, A.getID(mRS), X.getID(mRS), beta, Y.getID(mRS), incX… in DGBMV() local
[all …]
DRenderScript.java981 … float alpha, long A, long B, float beta, long C, int incX, int incY, in rsnScriptIntrinsicBLAS_Single()
985 … float alpha, long A, long B, float beta, long C, int incX, int incY, in nScriptIntrinsicBLAS_Single()
993 … double alpha, long A, long B, double beta, long C, int incX, int incY, in rsnScriptIntrinsicBLAS_Double()
997 … double alpha, long A, long B, double beta, long C, int incX, int incY, in nScriptIntrinsicBLAS_Double()
/frameworks/rs/cpu_ref/
DrsCpuIntrinsicBLAS.cpp144 static void walk_tiled_gemm(Func blasFunc, T_param alpha, T_param beta, int vecSize, in walk_tiled_gemm()
197 float beta = call->beta.f; in walk_2d_sgemm() local
208 double beta = call->beta.d; in walk_2d_dgemm() local
219 void * beta = (void *)&call->beta.c; in walk_2d_cgemm() local
230 void * beta = (void *)&call->beta.z; in walk_2d_zgemm() local
/frameworks/av/tools/resampler_tools/
Dfir.cpp61 static double kaiser(int k, int N, double beta) { in kaiser()
135 double beta = 7.865; in main() local
/frameworks/rs/support/java/src/androidx/renderscript/
DScriptIntrinsicBLAS.java327 …anspose int TransA, float alpha, Allocation A, Allocation X, int incX, float beta, Allocation Y, i… in SGEMV()
341 …e(getID(mRS), RsBlas_sgemv, TransA, 0, 0, 0, 0, M, N, 0, alpha, aID, xID, beta, yID, incX, incY, 0… in SGEMV() local
359 …nspose int TransA, double alpha, Allocation A, Allocation X, int incX, double beta, Allocation Y, … in DGEMV()
373 …e(getID(mRS), RsBlas_dgemv, TransA, 0, 0, 0, 0, M, N, 0, alpha, aID, xID, beta, yID, incX, incY, 0… in DGEMV() local
391 …nspose int TransA, Float2 alpha, Allocation A, Allocation X, int incX, Float2 beta, Allocation Y, … in CGEMV()
423 …pose int TransA, Double2 alpha, Allocation A, Allocation X, int incX, Double2 beta, Allocation Y, … in ZGEMV()
464 …sA, int KL, int KU, float alpha, Allocation A, Allocation X, int incX, float beta, Allocation Y, i… in SGBMV()
482 …e(getID(mRS), RsBlas_sgbmv, TransA, 0, 0, 0, 0, M, N, 0, alpha, aID, xID, beta, yID, incX, incY, K… in SGBMV() local
509 …A, int KL, int KU, double alpha, Allocation A, Allocation X, int incX, double beta, Allocation Y, … in DGBMV()
527 …e(getID(mRS), RsBlas_dgbmv, TransA, 0, 0, 0, 0, M, N, 0, alpha, aID, xID, beta, yID, incX, incY, K… in DGBMV() local
[all …]
DRenderScript.java930 … float alpha, long A, long B, float beta, long C, int incX, int incY, in rsnScriptIntrinsicBLAS_Single()
934 … float alpha, long A, long B, float beta, long C, int incX, int incY, in nScriptIntrinsicBLAS_Single()
942 … double alpha, long A, long B, double beta, long C, int incX, int incY, in rsnScriptIntrinsicBLAS_Double()
946 … double alpha, long A, long B, double beta, long C, int incX, int incY, in nScriptIntrinsicBLAS_Double()
/frameworks/rs/
DrsDefines.h464 RsBlasScalar beta; member
/frameworks/rs/support/jni/
Dandroid_renderscript_RenderScript.cpp587 … jfloat alpha, jlong A, jlong B, jfloat beta, jlong C, jint incX, jint incY, in nScriptIntrinsicBLAS_Single()
627 … jdouble alpha, jlong A, jlong B, jdouble beta, jlong C, jint incX, jint incY, in nScriptIntrinsicBLAS_Double()

12