Home
last modified time | relevance | path

Searched refs:set_layer_normalization_params (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQLstmWorkload.cpp114 qLstmParams.set_layer_normalization_params( in NeonQLstmWorkload()
344 aclParamsInfo.set_layer_normalization_params( in NeonQLstmWorkloadValidate()
H A DNeonLstmFloatWorkload.cpp126 lstm_param.set_layer_normalization_params(m_Data.m_Parameters.m_CifgEnabled ? in NeonLstmFloatWorkload()
355 lstm_params_info.set_layer_normalization_params(descriptor.m_CifgEnabled ? in NeonLstmFloatWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp246 lstm_param.set_layer_normalization_params(inputNormWeightTensor, in NeonUnidirectionalSequenceLstmFloatWorkload()
702 lstm_params_info.set_layer_normalization_params(descriptor.m_CifgEnabled ? nullptr : in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp258 lstm_param.set_layer_normalization_params(inputNormWeightTensor, in NeonUnidirectionalSequenceLstmWorkload()
684 lstm_params_info.set_layer_normalization_params(descriptor.m_CifgEnabled ? nullptr : in NeonUnidirectionalSequenceLstmWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQLstmWorkload.cpp116 qLstmParams.set_layer_normalization_params( in ClQLstmWorkload()
348 aclParamsInfo.set_layer_normalization_params( in ClQLstmWorkloadValidate()
H A DClLstmFloatWorkload.cpp129 lstm_param.set_layer_normalization_params(m_Data.m_Parameters.m_CifgEnabled ? nullptr : in ClLstmFloatWorkload()
347 lstm_params_info.set_layer_normalization_params(descriptor.m_CifgEnabled ? in ClLstmFloatWorkloadValidate()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp248 lstm_param.set_layer_normalization_params(inputNormWeightTensor, in ClUnidirectionalSequenceLstmFloatWorkload()
705 lstm_params_info.set_layer_normalization_params(descriptor.m_CifgEnabled ? nullptr : in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DInfoHelpers.h108 … lstm_params_info->set_layer_normalization_params(input_info, forget_info, cell_info, output_info); in build_lstm_params_tensor_info()
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/common/
H A DLSTMParams.h128 …LSTMParams &set_layer_normalization_params(T *input_layer_norm_weights, T *forget_layer_norm_weigh… in set_layer_normalization_params() function
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DLSTMLayerFixture.h154 …lstm_params.set_layer_normalization_params(&input_layer_norm_w, &forget_layer_norm_w, &cell_layer_… in compute_target()
158 …lstm_params.set_layer_normalization_params(nullptr, &forget_layer_norm_w, &cell_layer_norm_w, &out… in compute_target()