Home
last modified time | relevance | path

Searched refs:GetVariableInput (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc536 GetVariableInput(context, node, kFwInputActivationStateTensor); in Prepare()
539 GetVariableInput(context, node, kFwInputCellStateTensor); in Prepare()
604 GetVariableInput(context, node, kBwInputActivationStateTensor); in Prepare()
607 GetVariableInput(context, node, kBwInputCellStateTensor); in Prepare()
1035 GetVariableInput(context, node, kFwInputActivationStateTensor); in Eval()
1038 GetVariableInput(context, node, kFwInputCellStateTensor); in Eval()
1103 GetVariableInput(context, node, kBwInputActivationStateTensor); in Eval()
1106 GetVariableInput(context, node, kBwInputCellStateTensor); in Eval()
Dunidirectional_sequence_lstm.cc58 GetVariableInput(context, node, lstm::full::kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
146 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in PopulateQuantizedLstmParams8x8_16()
712 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
918 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in Prepare()
921 GetVariableInput(context, node, lstm::full::kCellStateTensor); in Prepare()
1279 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in Eval()
1282 GetVariableInput(context, node, lstm::full::kCellStateTensor); in Eval()
Dkernel_util.h61 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node,
Dlstm.cc150 GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
231 GetVariableInput(context, node, kOutputStateTensor); in PopulateQuantizedLstmParams8x8_16()
533 GetVariableInput(context, node, kOutputStateTensor); in PopulateQuantizedLstmParams8x8_8()
535 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_8()
1158 GetVariableInput(context, node, kOutputStateTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
1352 GetVariableInput(context, node, kOutputStateTensor); in Prepare()
1354 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Prepare()
1896 GetVariableInput(context, node, kOutputStateTensor); in Eval()
1898 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Eval()
Dbasic_rnn.cc295 GetVariableInput(context, node, kHiddenStateTensor); in Eval()
Dsvdf.cc300 TfLiteTensor* state = GetVariableInput(context, node, kStateTensor); in Eval()
Dbidirectional_sequence_rnn.cc746 GetVariableInput(context, node, kFwHiddenStateTensor); in Eval()
749 GetVariableInput(context, node, kBwHiddenStateTensor); in Eval()
Dkernel_util.cc113 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node, in GetVariableInput() function
Dunidirectional_sequence_rnn.cc371 GetVariableInput(context, node, kHiddenStateTensor); in Eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc594 TfLiteTensor* activation_state = GetVariableInput( in lstm_eval()
597 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc561 TfLiteTensor* output_state = GetVariableInput( in lstm_eval()
564 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval()