Searched refs:prev_state_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 445 const float* prev_state_data, in LstmCell() argument
546 prev_state_data[Offset(prev_state_shape, b, h, w, c)]; in LstmCell()
|
| D | legacy_reference_ops.h | 573 const Dims<4>& bias_dims, const float* prev_state_data, in LstmCell() argument
585 bias_data, DimsToShape(prev_state_dims), prev_state_data, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 2985 const float* prev_state_data, in LstmCell() argument
3078 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
3100 const Dims<4>& bias_dims, const float* prev_state_data, in LstmCell() argument
3112 bias_data, DimsToShape(prev_state_dims), prev_state_data, in LstmCell()
|
| D | optimized_ops.h | 2780 const float* prev_state_data, in LstmCell() argument
2876 MapAsArrayWithLastDimAsRows(prev_state_data, prev_state_shape); in LstmCell()
|