Searched refs:LSTMType (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.h | 26 enum class LSTMType { enum
|
| D | lstm.cc | 465 LSTMType lstm_type, Logger* logger, in lstm_eval()
583 case LSTMType::kLSTM: { in lstm_eval()
588 case LSTMType::kUnidirectionalSequenceLSTM: { in lstm_eval()
640 return lstm_eval(context, node, LSTMType::kLSTM, logger, error_reporter); in lstm_logging_kernel()
646 return lstm_eval(context, node, LSTMType::kUnidirectionalSequenceLSTM, logger, in unidirectional_sequence_lstm_logging_kernel()
|