| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/common/ |
| H A D | LSTMParams.h | 128 …TMParams &set_layer_normalization_params(T *input_layer_norm_weights, T *forget_layer_norm_weights, in set_layer_normalization_params()
240 T *forget_layer_norm_weights() const in forget_layer_norm_weights() function
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 79 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
251 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| D | unidirectional_sequence_lstm_f16_batch_major.mod.py | 84 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT16", variable
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| D | unidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py | 82 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", "{0}") variable
|
| D | unidirectional_sequence_lstm_1step.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| D | unidirectional_sequence_lstm_f16_norm_peephole_projection.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT16", variable
|
| D | unidirectional_sequence_lstm_cifg_peephole.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| D | unidirectional_sequence_lstm_batch_major_norm_peephole_projection.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| D | unidirectional_sequence_lstm_norm_peephole_projection.mod.py | 83 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 82 forget_layer_norm_weights = Input("forget_layer_norm_weights", "TENSOR_FLOAT32", variable
|
| D | qlstm_noprojection.mod.py | 65 forget_layer_norm_weights = Input("forget_layer_norm_weights", LayerNormType) variable
|
| D | qlstm_projection.mod.py | 65 forget_layer_norm_weights = Input("forget_layer_norm_weights", LayerNormType) variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 148 const RunTimeOperandInfo* forget_layer_norm_weights, in CheckInputTensorDimensions()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 4401 ANeuralNetworksOperandType forget_layer_norm_weights = quant16SymTensor1D; in TEST() local
|