Searched refs:concat_temp_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 828 const Dims<4>& concat_temp_dims, float* activ_temp_data, in LstmCell() argument
839 DimsToShape(concat_temp_dims), concat_temp_data, in LstmCell()
852 const Dims<4>& concat_temp_dims, int16* activ_temp_data_int16, in LstmCell() argument
868 DimsToShape(concat_temp_dims), concat_temp_data_uint8, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 527 const Dims<4>& concat_temp_dims, float* activ_temp_data, in LstmCell() argument
538 DimsToShape(concat_temp_dims), concat_temp_data, in LstmCell()
551 const Dims<4>& concat_temp_dims, int16* activ_temp_data_int16, in LstmCell() argument
567 DimsToShape(concat_temp_dims), concat_temp_data_uint8, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3771 const Dims<4>& concat_temp_dims, float* activ_temp_data, in LstmCell() argument
3782 DimsToShape(concat_temp_dims), concat_temp_data, in LstmCell()
4087 const Dims<4>& concat_temp_dims, int16* activ_temp_data_int16, in LstmCell() argument
4103 DimsToShape(concat_temp_dims), concat_temp_data_uint8, in LstmCell()
|