Searched refs:activ_temp_dims (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | legacy_optimized_ops.h | 829 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/ |
D | legacy_reference_ops.h | 528 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/ |
D | optimized_ops.h | 3772 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()
|