Home
last modified time | relevance | path

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

/external/armnn/include/armnn/
DLstmParams.hpp153 const TensorInfo& GetCellToInputWeights() const in GetCellToInputWeights() function
/external/armnn/src/backends/neon/workloads/
DNeonQLstmWorkload.cpp306 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in NeonQLstmWorkloadValidate()
DNeonLstmFloatWorkload.cpp313 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in NeonLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp660 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmWorkload.cpp642 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in NeonUnidirectionalSequenceLstmWorkloadValidate()
/external/armnn/src/backends/cl/workloads/
DClQLstmWorkload.cpp310 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in ClQLstmWorkloadValidate()
DClLstmFloatWorkload.cpp298 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in ClLstmFloatWorkloadValidate()
DClUnidirectionalSequenceLstmFloatWorkload.cpp663 … aclCellToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellToInputWeights()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp1811 supported &= CheckSupportRule(TypesAreEqual(input, paramsInfo.GetCellToInputWeights()), in IsLstmSupported()
2856 … supported &= CheckSupportRule(TypeAnyOf(paramsInfo.GetCellToInputWeights(), supportedWeightTypes), in IsUnidirectionalSequenceLstmSupported()