Home
last modified time | relevance | path

Searched refs:GetInputToInputWeights (Results 1 – 13 of 13) sorted by relevance

/external/armnn/include/armnn/
DQuantizedLstmParams.hpp58 const ConstTensor& GetInputToInputWeights() const in GetInputToInputWeights() function
165 const TensorInfo& GetInputToInputWeights() const in GetInputToInputWeights() function
DLstmParams.hpp121 const TensorInfo& GetInputToInputWeights() const in GetInputToInputWeights() function
/external/armnn/src/backends/cl/workloads/
DClQuantizedLstmWorkload.cpp34 = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in ClQuantizedLstmWorkloadValidate()
DClQLstmWorkload.cpp357 … aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in ClQLstmWorkloadValidate()
DClLstmFloatWorkload.cpp293 aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in ClLstmFloatWorkloadValidate()
DClUnidirectionalSequenceLstmFloatWorkload.cpp665 aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/external/armnn/src/backends/neon/workloads/
DNeonQuantizedLstmWorkload.cpp147 = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in NeonQuantizedLstmWorkloadValidate()
DNeonQLstmWorkload.cpp353 … aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in NeonQLstmWorkloadValidate()
DNeonLstmFloatWorkload.cpp315 aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in NeonLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp662 aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmWorkload.cpp644 aclInputToInputWeightsInfo = BuildArmComputeTensorInfo(paramsInfo.GetInputToInputWeights()); in NeonUnidirectionalSequenceLstmWorkloadValidate()
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp1802 …supported &= CheckSupportRule(TypesAreEqual(input, paramsInfo.GetInputToInputWeights()), reasonIfU… in IsLstmSupported()
2844 …supported &= CheckSupportRule(TypeAnyOf(paramsInfo.GetInputToInputWeights(), supportedWeightTypes), in IsUnidirectionalSequenceLstmSupported()
/external/armnn/src/armnn/
DNetwork.cpp2577 std::make_shared<ScopedTensorHandle>(params.GetInputToInputWeights()); in AddQuantizedLstmLayer()