Searched defs:hidden_state_scale (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/common/ |
H A D | LSTMParams.h | 188 LSTMParams &set_hidden_state_params(int32_t hidden_state_zero, float hidden_state_scale) in set_hidden_state_params() 290 float hidden_state_scale() const in hidden_state_scale() function
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
H A D | NEQLSTMLayer.cpp | 591 …const float hidden_state_scale = std::pow(2, -15) / lstm_params.hidden_state_scale() * std::pow(2,… in configure() local 989 …const float hidden_state_scale = std::pow(2, -15) / lstm_params.hidden_state_scale() * std::pow(2,… in validate() local
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
H A D | CLQLSTMLayer.cpp | 518 …const float hidden_state_scale = std::pow(2, -15) / lstm_params.hidden_state_scale() * std::pow(2,… in configure() local 889 …const float hidden_state_scale = std::pow(2, -15) / lstm_params.hidden_state_scale() * std::pow(2,… in validate() local
|
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
D | qlstm_noprojection.mod.py | 77 hidden_state_scale = Float32Scalar("hidden_state_scale", 0.007) variable
|
D | qlstm_projection.mod.py | 77 hidden_state_scale = Float32Scalar("hidden_state_scale", 0.007) variable
|
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
D | TestValidateOperations.cpp | 4411 ANeuralNetworksOperandType hidden_state_scale = float32Scalar; in TEST() local
|