Searched defs:output_state_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_gru.cc | 60 float* output_state_data = GetTensorData<float>(output_state); in GruImpl() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | lstm_cell.h | 37 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
|
| D | legacy_reference_ops.h | 576 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| 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()
|
| D | optimized_ops.h | 2458 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell()
|