Searched refs:split_layer (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 133 std::unique_ptr<arm_compute::CLSplit> split_layer(new arm_compute::CLSplit()); in ClUnidirectionalSequenceLstmFloatWorkload() local
137 split_layer->configure(&m_PermuteFirstOut, m_SplitterOutputs, aclAxisSplit); in ClUnidirectionalSequenceLstmFloatWorkload()
141 split_layer->configure(&input, m_SplitterOutputs, aclAxisSplit); in ClUnidirectionalSequenceLstmFloatWorkload()
144 split_layer->prepare(); in ClUnidirectionalSequenceLstmFloatWorkload()
145 m_Splitter.reset(split_layer.release()); in ClUnidirectionalSequenceLstmFloatWorkload()
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 131 std::unique_ptr<arm_compute::NESplit> split_layer(new arm_compute::NESplit()); in NeonUnidirectionalSequenceLstmFloatWorkload() local
136 split_layer->configure(&m_PermuteFirstOut, m_SplitterOutputs, aclAxisSplit); in NeonUnidirectionalSequenceLstmFloatWorkload()
139 split_layer->configure(&input, m_SplitterOutputs, aclAxisSplit); in NeonUnidirectionalSequenceLstmFloatWorkload()
142 split_layer->prepare(); in NeonUnidirectionalSequenceLstmFloatWorkload()
143 m_Splitter.reset(split_layer.release()); in NeonUnidirectionalSequenceLstmFloatWorkload()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 133 std::unique_ptr<arm_compute::NESplit> split_layer(new arm_compute::NESplit()); in NeonUnidirectionalSequenceLstmWorkload() local
138 split_layer->configure(&m_PermuteFirstOut, m_SplitterOutputs, aclAxisSplit); in NeonUnidirectionalSequenceLstmWorkload()
141 split_layer->configure(&input, m_SplitterOutputs, aclAxisSplit); in NeonUnidirectionalSequenceLstmWorkload()
144 split_layer->prepare(); in NeonUnidirectionalSequenceLstmWorkload()
145 m_Splitter.reset(split_layer.release()); in NeonUnidirectionalSequenceLstmWorkload()
|