Searched refs:kBwOutputTensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 62 constexpr int kBwOutputTensor = 1; // Only if merge_outputs is false. variable
360 context, GetOutputSafe(context, node, kBwOutputTensor, &bw_output)); in Prepare()
757 : GetOutput(context, node, kBwOutputTensor); in Eval()
|
| D | bidirectional_sequence_lstm.cc | 127 constexpr int kBwOutputTensor = 1; // Ignored if merge_outputs is set. variable
614 context, GetOutputSafe(context, node, kBwOutputTensor, &bw_output)); in Prepare()
1110 : GetOutput(context, node, kBwOutputTensor); in Eval()
|