| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | bidirectional_sequence_rnn.cc | 349 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
|
| H A D | bidirectional_sequence_lstm.cc | 532 TfLiteTensor* fw_output; in Prepare() local
1044 TfLiteTensor* fw_output; in Eval() local
|
| H A D | bidirectional_sequence_rnn_test.cc | 1137 std::vector<float> fw_output = rnn.GetFwOutput(); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | group_bidirectional_sequence_ops.cc | 39 Operator** fw_output, Operator** bw_output) { in MatchTwoUnpackOps()
61 Operator** fw_output, in MatchDynamicBidirectionalSequenceOutputs()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_rnn_1_3.mod.py | 41 time_major, merge_outputs, fw_output, bw_output, input_data, argument
|
| D | bidirectional_sequence_rnn_state_output.mod.py | 45 fw_output, bw_output, fw_output_hidden_state, bw_output_hidden_state, argument
|
| D | bidirectional_sequence_lstm_state_output.mod.py | 146 fw_output=Output("fw_output", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_output… variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | bidirectional_sequence_rnn.mod.py | 41 time_major, merge_outputs, fw_output, bw_output, input_data, argument
|
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 147 fw_output=Output("fw_output", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, variable
|
| D | bidirectional_sequence_lstm_merge_outputs.mod.py | 147 fw_output=Output("fw_output", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, 2 * n_ou… variable
|
| D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 147 fw_output=Output("fw_output", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_output… variable
|
| D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 146 fw_output=Output("fw_output", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_output… variable
|
| D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 146 fw_output=Output("fw_output", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_output… variable
|
| D | bidirectional_sequence_lstm_aux_input.mod.py | 148 fw_output=Output("fw_output", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_output… variable
|
| D | bidirectional_sequence_lstm.mod.py | 146 fw_output=Output("fw_output", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_output… variable
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 149 fw_output=Output("fw_output", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_output… variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 886 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
|