Home
last modified time | relevance | path

Searched refs:kHiddenStateTensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc44 constexpr int kHiddenStateTensor = 4; variable
70 GetInput(context, node, kHiddenStateTensor); in Prepare()
227 &context->tensors[node->inputs->data[kHiddenStateTensor]]; in Eval()
Dunidirectional_sequence_rnn.cc39 constexpr int kHiddenStateTensor = 4; variable
65 GetInput(context, node, kHiddenStateTensor); in Prepare()
298 const_cast<TfLiteTensor*>(GetInput(context, node, kHiddenStateTensor)); in Eval()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1131 constexpr int kHiddenStateTensor = 4; in ProcessUnidirectionalSequenceRnnOperator() local
1137 model->GetArray(op->inputs[kHiddenStateTensor]).buffer.reset(); in ProcessUnidirectionalSequenceRnnOperator()