Home
last modified time | relevance | path

Searched refs:activ_temp_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h829 const Dims<4>& activ_temp_dims) { in LstmCell() argument
840 DimsToShape(activ_temp_dims), activ_temp_data); in LstmCell()
853 const Dims<4>& activ_temp_dims, int32 weights_zero_point, in LstmCell() argument
869 DimsToShape(activ_temp_dims), activ_temp_data_int16, gemm_context); in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h528 const Dims<4>& activ_temp_dims) { in LstmCell() argument
539 DimsToShape(activ_temp_dims), activ_temp_data); in LstmCell()
552 const Dims<4>& activ_temp_dims, int32 weights_zero_point, in LstmCell() argument
568 DimsToShape(activ_temp_dims), activ_temp_data_int16, gemm_context); in LstmCell()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h3772 const Dims<4>& activ_temp_dims) { in LstmCell() argument
3783 DimsToShape(activ_temp_dims), activ_temp_data); in LstmCell()
4088 const Dims<4>& activ_temp_dims, int32 weights_zero_point, in LstmCell() argument
4104 DimsToShape(activ_temp_dims), activ_temp_data_int16, gemm_context); in LstmCell()