Searched refs:output_state_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 825 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell() argument
837 DimsToShape(output_state_dims), output_state_data, in LstmCell()
|
| D | optimized_ops.h | 3560 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell() argument
3654 MapAsArrayWithLastDimAsRows(output_state_data, output_state_shape); in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 524 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell() argument
536 DimsToShape(output_state_dims), output_state_data, in LstmCell()
|
| D | reference_ops.h | 1842 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell() argument
1943 output_state_data[Offset(output_state_shape, b, h, w, c)] = new_state; in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3651 const RuntimeShape& unextended_output_state_shape, float* output_state_data, in LstmCell() argument
3745 MapAsArrayWithLastDimAsRows(output_state_data, output_state_shape); in LstmCell()
3768 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell() argument
3780 DimsToShape(output_state_dims), output_state_data, in LstmCell()
|