Searched refs:LstmCell (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 756 optimized_ops::LstmCell( in Eval()
802 optimized_ops::LstmCell<4>( in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 820 inline void LstmCell(const float* input_data, const Dims<4>& input_dims, in LstmCell() function
833 LstmCell(op_params, DimsToShape(input_dims), input_data, in LstmCell()
844 void LstmCell(const uint8* input_data_uint8, const Dims<4>& input_dims, in LstmCell() function
861 LstmCell<StateIntegerBits>( in LstmCell()
|
| D | optimized_ops.h | 3553 inline void LstmCell( in LstmCell() function
3674 inline void LstmCell( in LstmCell() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 519 inline void LstmCell(const float* input_data, const Dims<4>& input_dims, in LstmCell() function
532 LstmCell(op_params, DimsToShape(input_dims), input_data, in LstmCell()
543 void LstmCell(const uint8* input_data_uint8, const Dims<4>& input_dims, in LstmCell() function
560 LstmCell<StateIntegerBits>( in LstmCell()
|
| D | reference_ops.h | 1835 inline void LstmCell( in LstmCell() function
2037 inline void LstmCell( in LstmCell() function
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3644 inline void LstmCell( in LstmCell() function
3763 inline void LstmCell(const float* input_data, const Dims<4>& input_dims, in LstmCell() function
3776 LstmCell(op_params, DimsToShape(input_dims), input_data, in LstmCell()
3790 inline void LstmCell( in LstmCell() function
4079 void LstmCell(const uint8* input_data_uint8, const Dims<4>& input_dims, in LstmCell() function
4096 LstmCell<StateIntegerBits>( in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 328 HANDLE_OPERATORTYPENAME_CASE(LstmCell) in OperatorTypeName()
|