Searched refs:_concat_scratch_buffer (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NELSTMLayer.h | 187 NEConcatenateLayer _concat_scratch_buffer; variable
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLLSTMLayer.h | 244 CLConcatenateLayer _concat_scratch_buffer; variable
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 55 …_copy_cell_state(), _copy_output(), _concat_scratch_buffer(), _concat_inputs_forget_gate(), _conca… in NELSTMLayer()
370 _concat_scratch_buffer.configure(scratch_inputs, scratch_buffer, Window::DimX); in configure()
709 _concat_scratch_buffer.run(); in run()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 59 …opyKernel>()), _copy_output(support::cpp14::make_unique<CLCopyKernel>()), _concat_scratch_buffer(), in CLLSTMLayer()
401 _concat_scratch_buffer.configure(compile_context, scratch_inputs, scratch_buffer, Window::DimX); in configure()
736 _concat_scratch_buffer.run(); in run()
|