Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-155.md11 TfLiteTensor* state = GetVariableInput(context, node, kStateTensor);
16 The [`GetVariableInput`
22 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node,
29 Furthermore, because `GetVariableInput` calls
/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm.cc536 GetVariableInput(context, node, kFwInputActivationStateTensor); in Prepare()
539 GetVariableInput(context, node, kFwInputCellStateTensor); in Prepare()
606 GetVariableInput(context, node, kBwInputActivationStateTensor); in Prepare()
609 GetVariableInput(context, node, kBwInputCellStateTensor); in Prepare()
1039 GetVariableInput(context, node, kFwInputActivationStateTensor); in Eval()
1042 GetVariableInput(context, node, kFwInputCellStateTensor); in Eval()
1107 GetVariableInput(context, node, kBwInputActivationStateTensor); in Eval()
1110 GetVariableInput(context, node, kBwInputCellStateTensor); in Eval()
Dunidirectional_sequence_lstm.cc59 GetVariableInput(context, node, lstm::full::kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
151 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in PopulateQuantizedLstmParams8x8_16()
720 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
928 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in Prepare()
931 GetVariableInput(context, node, lstm::full::kCellStateTensor); in Prepare()
1292 GetVariableInput(context, node, lstm::full::kOutputStateTensor); in Eval()
1295 GetVariableInput(context, node, lstm::full::kCellStateTensor); in Eval()
Dlstm.cc151 GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
232 GetVariableInput(context, node, kOutputStateTensor); in PopulateQuantizedLstmParams8x8_16()
534 GetVariableInput(context, node, kOutputStateTensor); in PopulateQuantizedLstmParams8x8_8()
536 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in PopulateQuantizedLstmParams8x8_8()
1159 GetVariableInput(context, node, kOutputStateTensor); in PopulatePrecomputedZPTimesWeightsWithBias()
1353 GetVariableInput(context, node, kOutputStateTensor); in Prepare()
1355 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Prepare()
1900 GetVariableInput(context, node, kOutputStateTensor); in Eval()
1902 TfLiteTensor* cell_state = GetVariableInput(context, node, kCellStateTensor); in Eval()
Dkernel_util.h64 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node,
Dbasic_rnn.cc296 GetVariableInput(context, node, kHiddenStateTensor); in Eval()
Dbidirectional_sequence_rnn.cc748 GetVariableInput(context, node, kFwHiddenStateTensor); in Eval()
751 GetVariableInput(context, node, kBwHiddenStateTensor); in Eval()
Dsvdf.cc310 TfLiteTensor* state = GetVariableInput(context, node, kStateTensor); in Eval()
Dunidirectional_sequence_rnn.cc371 GetVariableInput(context, node, kHiddenStateTensor); in Eval()
Dkernel_util.cc120 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node, in GetVariableInput() function
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc562 TfLiteTensor* output_state = GetVariableInput( in lstm_eval()
565 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc606 TfLiteTensor* activation_state = GetVariableInput( in lstm_eval()
609 TfLiteTensor* cell_state = GetVariableInput( in lstm_eval()