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()
|