ClLstmFloatWorkload.cpp File Reference

#include "ClLstmFloatWorkload.hpp"
#include <cl/ClTensorHandle.hpp>
#include <armnn/backends/TensorHandle.hpp>
#include <cl/ClLayerSupport.hpp>
#include <aclCommon/ArmComputeUtils.hpp>
#include <aclCommon/ArmComputeTensorUtils.hpp>
#include <armnn/utility/NumericCast.hpp>
#include <arm_compute/runtime/CL/functions/CLLSTMLayer.h>
#include "ClWorkloadUtils.hpp"

Include dependency graph for ClLstmFloatWorkload.cpp:

Go to the source code of this file.

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

Functions
arm_compute::Status	ClLstmFloatWorkloadValidate (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)