Home
last modified time | relevance | path

Searched refs:inputNormWeightTensor (Results 1 – 3 of 3) sorted by relevance

/external/armnn/src/backends/cl/workloads/
DClUnidirectionalSequenceLstmFloatWorkload.cpp247 …auto inputNormWeightTensor = m_Data.m_Parameters.m_CifgEnabled ? nullptr : m_InputLayerNormWeights… in ClUnidirectionalSequenceLstmFloatWorkload() local
248 lstm_param.set_layer_normalization_params(inputNormWeightTensor, in ClUnidirectionalSequenceLstmFloatWorkload()
/external/armnn/src/backends/neon/workloads/
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp245 …auto inputNormWeightTensor = m_Data.m_Parameters.m_CifgEnabled ? nullptr : m_InputLayerNormWeights… in NeonUnidirectionalSequenceLstmFloatWorkload() local
246 lstm_param.set_layer_normalization_params(inputNormWeightTensor, in NeonUnidirectionalSequenceLstmFloatWorkload()
DNeonUnidirectionalSequenceLstmWorkload.cpp257 …auto inputNormWeightTensor = m_Data.m_Parameters.m_CifgEnabled ? nullptr : m_InputLayerNormWeights… in NeonUnidirectionalSequenceLstmWorkload() local
258 lstm_param.set_layer_normalization_params(inputNormWeightTensor, in NeonUnidirectionalSequenceLstmWorkload()