NeonLstmFloatWorkload.cpp File Reference

#include "NeonLstmFloatWorkload.hpp"
#include "NeonWorkloadUtils.hpp"
#include <aclCommon/ArmComputeTensorUtils.hpp>
#include <aclCommon/ArmComputeUtils.hpp>
#include <armnn/utility/NumericCast.hpp>
#include "neon/NeonTensorHandle.hpp"

Include dependency graph for NeonLstmFloatWorkload.cpp:

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Functions
arm_compute::Status	NeonLstmFloatWorkloadValidate (const TensorInfo &input, const TensorInfo &outputStateIn, const TensorInfo &cellStateIn, const TensorInfo &scratchBuffer, const TensorInfo &outputStateOut, const TensorInfo &cellStateOut, const TensorInfo &output, const LstmDescriptor &descriptor, const LstmInputParamsInfo &paramsInfo)