/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
H A D | CLRNNLayer.cpp | 47 …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 D | basic_rnn.cc | 74 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 D | unidirectional_sequence_rnn.cc | 75 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 D | NERNNLayer.cpp | 45 …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 D | io_memory.h | 73 uint16_t hidden_state[1024 * 4096]; member 135 uint16_t hidden_state[4096]; member
|
H A D | io_memory.cpp | 134 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/ |
D | unidirectional_sequence_rnn.mod.py | 19 def test(name, input, weights, recurrent_weights, bias, hidden_state, argument
|
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
D | unidirectional_sequence_rnn.mod.py | 19 def test(name, input, weights, recurrent_weights, bias, hidden_state, argument
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A D | RNNLayerFixture.h | 72 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 D | llama.py | 184 hidden_state, argument
|