Searched refs:_dwc_assembly_kernel (Results 1 – 1 of 1) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/assembly/ |
| D | NEDepthwiseConvolutionAssemblyDispatch.cpp | 335 std::unique_ptr<depthwise::IDepthwiseConvolution> _dwc_assembly_kernel{ nullptr }; member
380 …_pImpl->_dwc_assembly_kernel = create_convolver(input, weights, output, conv_info, act_info, dilat… in configure()
381 ARM_COMPUTE_ERROR_ON(_pImpl->_dwc_assembly_kernel == nullptr); in configure()
384 _pImpl->_dwc_acl_kernel.configure(_pImpl->_dwc_assembly_kernel.get()); in configure()
390 …const size_t workspace_size = _pImpl->_dwc_assembly_kernel->get_working_space_size(num_threa… in configure()
397 const size_t pack_tensor_size = _pImpl->_dwc_assembly_kernel->get_packed_params_size(); in configure()
523 _pImpl->_dwc_assembly_kernel->set_working_space(static_cast<void *>(_workspace.buffer())); in run()
531 …_pImpl->_dwc_assembly_kernel->set_input(input_ptr, input_batch_stride, input_row_stride, input_col… in run()
539 …_pImpl->_dwc_assembly_kernel->set_output(output_ptr, output_batch_stride, output_row_stride, outpu… in run()
556 _pImpl->_dwc_assembly_kernel->pack_params(_packed_weights.buffer(), in prepare()
[all …]
|