Searched defs:merge_outputs (Results 1 – 5 of 5) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| H A D | bidirectional_sequence_rnn_1_3.mod.py | 25 def merge_outputs(a, a_shape, b, b_shape): function
41 time_major, merge_outputs, fw_output, bw_output, input_data, argument
|
| H A D | bidirectional_sequence_rnn_state_output.mod.py | 25 def merge_outputs(a, a_shape, b, b_shape): function
44 fw_aux_weights, bw_aux_weights, activation, time_major, merge_outputs, argument
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| H A D | bidirectional_sequence_rnn.mod.py | 25 def merge_outputs(a, a_shape, b, b_shape): function
41 time_major, merge_outputs, fw_output, bw_output, input_data, argument
|
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | LSTM.h | 41 bool merge_outputs; member
|
| /aosp12/packages/modules/NeuralNetworks/common/ |
| H A D | CpuExecutor.cpp | 998 const auto merge_outputs = getScalarData<bool>( in executeOperation() local
|