24 #ifndef ARM_COMPUTE_CLLSTMLAYER_H 25 #define ARM_COMPUTE_CLLSTMLAYER_H 48 class CLCompileContext;
65 CLLSTMLayer(std::shared_ptr<IMemoryManager> memory_manager =
nullptr);
171 const ICLTensor *recurrent_to_forget_weights,
const ICLTensor *recurrent_to_cell_weights,
const ICLTensor *recurrent_to_output_weights,
227 void prepare()
override;
242 std::unique_ptr<opencl::kernels::ClTransposeKernel> _transpose_cell_state;
307 bool _run_peephole_opt;
309 bool _perform_cell_clipping;
310 bool _has_projection_weights;
311 bool _perform_projection_clipping;
313 bool _is_layer_norm_lstm;
314 const ICLTensor *_recurrent_to_cell_weights{
nullptr };
OpenCL kernel to transpose a 2D tensor.
Basic function to compute a Fully Connected layer on OpenCL.
Base class for all functions.
Basic function to run opencl::kernels::ClCopyKernel.
Basic function to execute mean and standard deviation normalization by calling CLMeanStdDevNormalizat...
Basic function to run opencl::kernels::ClSaturatedArithmeticKernel for addition.
auto recurrent_to_forget_weights
Store the tensor's metadata.
Basic function to run opencl::kernels::ClActivationKernel.
Basic function to execute concatenate tensors along a given axis.
Activation Layer Information class.
Copyright (c) 2017-2022 Arm Limited.
auto input_to_cell_weights
auto recurrent_to_output_weights
auto input_to_output_weights
Basic function to execute GEMM on OpenCL.
This function performs a single time step in a Long Short-Term Memory (LSTM) layer.
auto recurrent_to_cell_weights
Basic function to run opencl::ClMul.
Interface for OpenCL tensor.
Basic function to run opencl::kernels::ClFillKernel.
im2col_func configure(src_target.info(), dst_target.info(), spatial_kernel, conv_info, has_bias)
Basic function to run opencl::kernels::ClSaturatedArithmeticKernel for subtraction.
auto input_to_forget_weights
Status validate(const ITensorInfo *scores_in, const ITensorInfo *boxes_in, const ITensorInfo *batch_splits_in, const ITensorInfo *scores_out, const ITensorInfo *boxes_out, const ITensorInfo *classes, const ITensorInfo *batch_splits_out, const ITensorInfo *keeps, const ITensorInfo *keeps_size, const BoxNMSLimitInfo info)
Basic implementation of the OpenCL tensor interface.