Home
last modified time | relevance | path

Searched refs:GetProjectionWeights (Results 1 – 9 of 9) sorted by relevance

/external/armnn/include/armnn/
DLstmParams.hpp181 const TensorInfo& GetProjectionWeights() const in GetProjectionWeights() function
/external/armnn/src/backends/neon/workloads/
DNeonQLstmWorkload.cpp319 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in NeonQLstmWorkloadValidate()
DNeonLstmFloatWorkload.cpp330 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in NeonLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp678 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmWorkload.cpp660 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in NeonUnidirectionalSequenceLstmWorkloadValidate()
/external/armnn/src/backends/cl/workloads/
DClQLstmWorkload.cpp323 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in ClQLstmWorkloadValidate()
DClLstmFloatWorkload.cpp309 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in ClLstmFloatWorkloadValidate()
DClUnidirectionalSequenceLstmFloatWorkload.cpp681 aclProjectionWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetProjectionWeights()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp1825 …supported &= CheckSupportRule(TypesAreEqual(input, paramsInfo.GetProjectionWeights()), reasonIfUns… in IsLstmSupported()
2875 … supported &= CheckSupportRule(TypeAnyOf(paramsInfo.GetProjectionWeights(), supportedWeightTypes), in IsUnidirectionalSequenceLstmSupported()