Home
last modified time | relevance | path

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

12

/frameworks/ml/nn/common/operations/
DSoftmax.cpp42 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
74 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
90 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
104 bool softmaxQuant8Impl(const uint8_t* inputData, const Shape& inputShape, const float beta, in softmaxQuant8Impl()
186 bool softmaxQuant8(const uint8_t* inputData, const Shape& inputShape, const float beta, in softmaxQuant8()
256 float beta = (input.type == OperandType::TENSOR_FLOAT16) in prepare() local
DNormalization.cpp30 int32_t radius, float bias, float alpha, float beta, in localResponseNormFloat32Impl()
60 float bias, float alpha, float beta, int32_t axis, in localResponseNormFloat16()
75 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNormFloat32()
/frameworks/ml/nn/runtime/test/generated/models/
Dsoftmax_float_1_relaxed.model.cpp8 auto beta = model->addOperand(&type1); in CreateModel() local
34 auto beta = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
Dsoftmax_float_1.model.cpp8 auto beta = model->addOperand(&type1); in CreateModel() local
32 auto beta = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
Dsoftmax_float_2.model.cpp8 auto beta = model->addOperand(&type1); in CreateModel() local
32 auto beta = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
Dsoftmax_float_2_relaxed.model.cpp8 auto beta = model->addOperand(&type1); in CreateModel() local
34 auto beta = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
Dsoftmax_quant8_2.model.cpp9 auto beta = model->addOperand(&type1); in CreateModel() local
33 auto beta = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
Dsoftmax_quant8_1.model.cpp9 auto beta = model->addOperand(&type1); in CreateModel() local
33 auto beta = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_3.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
46 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_1.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
46 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_4.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
46 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_4_relaxed.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
48 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_2_relaxed.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
48 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_2.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
46 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_3_relaxed.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
48 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
Dlocal_response_norm_float_1_relaxed.model.cpp12 auto beta = model->addOperand(&type2); in CreateModel() local
48 auto beta = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
/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_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
/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
/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 …]

12