20.02
|
#include "NeonLstmFloatWorkload.hpp"
#include "NeonWorkloadUtils.hpp"
#include "aclCommon/ArmComputeTensorUtils.hpp"
#include "neon/NeonTensorHandle.hpp"
Go to the source code of this file.
Namespaces | |
armnn | |
Copyright (c) 2020 ARM Limited. | |
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 ¶msInfo) |