Home
last modified time | relevance | path

Searched defs:prev_state_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h576 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
600 const Dims<4>& prev_state_dims, int16* output_state_data_int16, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3102 const Dims<4>& prev_state_dims, float* output_state_data, in LstmCell()
3413 const Dims<4>& prev_state_dims, int16* output_state_data_int16, in LstmCell()