Searched refs:m_ConvolutionLayer (Results 1 – 8 of 8) sorted by relevance
62 , m_ConvolutionLayer() in ClConvolution3dWorkload()88 m_ConvolutionLayer.configure(clCompileContext, in ClConvolution3dWorkload()109 m_ConvolutionLayer.prepare(); in ClConvolution3dWorkload()115 RunClFunction(m_ConvolutionLayer, CHECK_LOCATION()); in Execute()
78 , m_ConvolutionLayer(memoryManager) in ClConvolution2dWorkload()113 m_ConvolutionLayer.configure(clCompileContext, in ClConvolution2dWorkload()126 m_ConvolutionLayer.get_convolution_method(input.info(), in ClConvolution2dWorkload()158 RunClFunction(m_ConvolutionLayer, CHECK_LOCATION()); in Execute()
37 mutable arm_compute::CLConv3D m_ConvolutionLayer; member in armnn::ClConvolution3dWorkload
61 mutable arm_compute::CLConvolutionLayer m_ConvolutionLayer; member in armnn::ClConvolution2dWorkload
103 m_ConvolutionLayer.reset(convolutionLayer.release()); in NeonConvolution3dWorkload()105 ARMNN_ASSERT(m_ConvolutionLayer); in NeonConvolution3dWorkload()107 m_ConvolutionLayer->prepare(); in NeonConvolution3dWorkload()113 m_ConvolutionLayer->run(); in Execute()
147 m_ConvolutionLayer.reset(convolutionLayer.release()); in NeonConvolution2dWorkload()149 ARMNN_ASSERT(m_ConvolutionLayer); in NeonConvolution2dWorkload()170 m_ConvolutionLayer->prepare(); in Execute()175 m_ConvolutionLayer->run(); in Execute()
40 std::unique_ptr<arm_compute::IFunction> m_ConvolutionLayer; member in armnn::NeonConvolution3dWorkload
42 std::unique_ptr<arm_compute::IFunction> m_ConvolutionLayer; member in armnn::NeonConvolution2dWorkload