Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h575 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell()
599 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3102 const Dims<4>& output_state_dims, float* output_activ_data, in LstmCell()
3413 const Dims<4>& output_state_dims, uint8* output_activ_data_uint8, in LstmCell()