Home
last modified time | relevance | path

Searched refs:compute_rnn_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNERNNLayer.cpp63 …auto shape_info = TensorInfo(misc::shape_calculator::compute_rnn_shape(recurrent_weights, hidden_s… in validate()
80 …TensorShape shape = misc::shape_calculator::compute_rnn_shape(recurrent_weights->info(), hidd… in configure()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLRNNLayer.cpp66 …auto shape_info = TensorInfo(compute_rnn_shape(recurrent_weights, hidden_state->dimension(idx_heig… in validate()
91 …TensorShape shape = compute_rnn_shape(recurrent_weights->info(), hidden_state->info()->dimens… in configure()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h888 inline TensorShape compute_rnn_shape(const ITensorInfo *input, const unsigned int batch_size) in compute_rnn_shape() function