Searched refs:output_state_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 826 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell() argument
837 DimsToShape(output_state_dims), output_state_data, in LstmCell()
850 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell() argument
866 DimsToShape(output_state_dims), output_state_data_int16, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 525 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell() argument
536 DimsToShape(output_state_dims), output_state_data, in LstmCell()
549 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell() argument
565 DimsToShape(output_state_dims), output_state_data_int16, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3769 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell() argument
3780 DimsToShape(output_state_dims), output_state_data, in LstmCell()
4085 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell() argument
4101 DimsToShape(output_state_dims), output_state_data_int16, in LstmCell()
|