Home
last modified time | relevance | path

Searched defs:hidden_state (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLRNNLayer.cpp47 …ts, const ITensorInfo *recurrent_weights, const ITensorInfo *bias, const ITensorInfo *hidden_state, in validate()
76 …ghts, const ICLTensor *recurrent_weights, const ICLTensor *bias, ICLTensor *hidden_state, ICLTenso… in configure()
83 ICLTensor *hidden_state, in configure()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbasic_rnn.cc74 const TfLiteTensor* hidden_state; in Prepare() local
205 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat()
236 TfLiteTensor* hidden_state, TfLiteTensor* output, in EvalHybrid()
295 TfLiteTensor* hidden_state = in Eval() local
H A Dunidirectional_sequence_rnn.cc75 const TfLiteTensor* hidden_state; in Prepare() local
212 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat()
273 TfLiteTensor* hidden_state, TfLiteTensor* output, TfLiteTensor* zero_points, in EvalHybrid()
370 TfLiteTensor* hidden_state = in Eval() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNERNNLayer.cpp45 …ts, const ITensorInfo *recurrent_weights, const ITensorInfo *bias, const ITensorInfo *hidden_state, in validate()
72 … *weights, const ITensor *recurrent_weights, const ITensor *bias, ITensor *hidden_state, ITensor *… in configure()
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/
H A Dio_memory.h73 uint16_t hidden_state[1024 * 4096]; member
135 uint16_t hidden_state[4096]; member
H A Dio_memory.cpp134 Result<TensorInfo> hidden_state = in prepare_io() local
296 Result<TensorInfo> hidden_state = in prepare_io() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dunidirectional_sequence_rnn.mod.py19 def test(name, input, weights, recurrent_weights, bias, hidden_state, argument
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dunidirectional_sequence_rnn.mod.py19 def test(name, input, weights, recurrent_weights, bias, hidden_state, argument
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DRNNLayerFixture.h72 TensorType hidden_state = create_tensor<TensorType>(output_shape, data_type); in compute_target() local
/aosp_15_r20/external/executorch/examples/mediatek/model_export_scripts/
H A Dllama.py184 hidden_state, argument