Searched refs:prev_state_dims (Results 1 – 3 of 3) 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
836 bias_data, DimsToShape(prev_state_dims), prev_state_data, in LstmCell()
849 const Dims<4>& prev_state_dims, int16* output_state_data_int16, in LstmCell() argument
865 bias_data_int32, DimsToShape(prev_state_dims), prev_state_data_int16, 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
535 bias_data, DimsToShape(prev_state_dims), prev_state_data, in LstmCell()
548 const Dims<4>& prev_state_dims, int16* output_state_data_int16, in LstmCell() argument
564 bias_data_int32, DimsToShape(prev_state_dims), prev_state_data_int16, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3768 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell() argument
3779 bias_data, DimsToShape(prev_state_dims), prev_state_data, in LstmCell()
4084 const Dims<4>& prev_state_dims, int16* output_state_data_int16, in LstmCell() argument
4100 bias_data_int32, DimsToShape(prev_state_dims), prev_state_data_int16, in LstmCell()
|