Home
last modified time | relevance | path

Searched refs:GetCellBias (Results 1 – 14 of 14) sorted by relevance

/external/armnn/include/armnn/
DQuantizedLstmParams.hpp108 const ConstTensor& GetCellBias() const in GetCellBias() function
207 const TensorInfo& GetCellBias() const in GetCellBias() function
DLstmParams.hpp173 const TensorInfo& GetCellBias() const in GetCellBias() function
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval_test.cc364 TfLiteTensor* GetCellBias() { in GetCellBias() function in tflite::__anon353329140111::QuantizedLstmParam
618 one_parameter.GetForgetBias(), one_parameter.GetCellBias(), in TestOneFullyQuantizedLSTM()
755 TfLiteTensor* GetCellBias() { in GetCellBias() function in tflite::__anon353329140111::HybridLstmParam
928 one_parameter.GetForgetBias(), one_parameter.GetCellBias(), in TestOneHybridAsymmLSTM()
/external/armnn/src/backends/cl/workloads/
DClQuantizedLstmWorkload.cpp51 …rm_compute::TensorInfo aclCellBiasInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in ClQuantizedLstmWorkloadValidate()
DClQLstmWorkload.cpp283 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in ClQLstmWorkloadValidate()
DClLstmFloatWorkload.cpp275 …onst arm_compute::TensorInfo aclCellBiasInfo = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in ClLstmFloatWorkloadValidate()
DClUnidirectionalSequenceLstmFloatWorkload.cpp640 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/external/armnn/src/backends/neon/workloads/
DNeonQuantizedLstmWorkload.cpp169 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in NeonQuantizedLstmWorkloadValidate()
DNeonQLstmWorkload.cpp279 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in NeonQLstmWorkloadValidate()
DNeonLstmFloatWorkload.cpp290 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in NeonLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp637 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmWorkload.cpp620 = BuildArmComputeTensorInfo(paramsInfo.GetCellBias()); in NeonUnidirectionalSequenceLstmWorkloadValidate()
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp1796 … supported &= CheckSupportRule(TypesAreEqual(input, paramsInfo.GetCellBias()), reasonIfUnsupported, in IsLstmSupported()
2838 …supported &= CheckSupportRule(TypeAnyOf(paramsInfo.GetCellBias(), supportedBiasTypes), reasonIfUns… in IsUnidirectionalSequenceLstmSupported()
/external/armnn/src/armnn/
DNetwork.cpp2601 std::make_shared<ScopedTensorHandle>(params.GetCellBias()); in AddQuantizedLstmLayer()