Home
last modified time | relevance | path

Searched defs:fw_output (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn.cc349 TfLiteTensor* fw_output; in Prepare() local
385 TfLiteTensor* fw_hidden_state, TfLiteTensor* fw_output, in EvalFloat()
527 TfLiteTensor* fw_hidden_state, TfLiteTensor* fw_output, in EvalHybrid()
754 TfLiteTensor* fw_output; in Eval() local
Dbidirectional_sequence_lstm.cc532 TfLiteTensor* fw_output; in Prepare() local
1044 TfLiteTensor* fw_output; in Eval() local
Dbidirectional_sequence_rnn_test.cc1137 std::vector<float> fw_output = rnn.GetFwOutput(); in TEST() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dgroup_bidirectional_sequence_ops.cc39 Operator** fw_output, Operator** bw_output) { in MatchTwoUnpackOps()
61 Operator** fw_output, in MatchDynamicBidirectionalSequenceOutputs()
/external/pytorch/aten/src/ATen/native/
DRNN.cpp886 auto fw_output = at::stack(fw_result.outputs, 0); in operator ()() local
901 auto fw_output = at::stack(fw_result.outputs, 0); in operator ()() local