Searched defs:output_state_data (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | unidirectional_sequence_rnn.mod.py | 22 output_state_data): argument
142 output_state_data = [ variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_gru.cc | 60 float* output_state_data = GetTensorData<float>(output_state); in GruImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 37 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
|
| H A D | legacy_reference_ops.h | 576 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 2987 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
3102 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
|
| H A D | optimized_ops.h | 2458 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
|