Home
last modified time | relevance | path

Searched refs:kFwHiddenStateTensor (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn.cc51 constexpr int kFwHiddenStateTensor = 4; variable
103 GetInput(context, node, kFwHiddenStateTensor); in Prepare()
616 GetVariableInput(context, node, kFwHiddenStateTensor); in Eval()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1251 constexpr int kFwHiddenStateTensor = 4; in ProcessBidirectionalSequenceRnnOperator() local
1254 model->GetArray(op->inputs[kFwHiddenStateTensor]).buffer.reset(); in ProcessBidirectionalSequenceRnnOperator()