Searched refs:_conv_args (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuWinogradConv2dKernel.cpp | 32 : _winograd_impl{ w_impl }, _conv_args{ _c_args }, _nthreads{ nthreads } in CpuWinogradConv2dTransformInputKernel()
56 _conv_args, in run_op()
69 : _winograd_impl{ w_impl }, _conv_args{ _c_args }, _nthreads{ nthreads } in CpuWinogradConv2dTransformOutputKernel()
101 _conv_args, in run_op()
|
| D | CpuWinogradConv2dKernel.h | 68 arm_conv::ConvolutionArgs &_conv_args; variable
98 const arm_conv::ConvolutionArgs &_conv_args; variable
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuWinogradConv2d.cpp | 144 _conv_args{ nullptr }, in CpuWinogradConv2d()
174 …lementation(src, weights, dst, conv_info, act_info, enable_fast_math, &_winograd_impl, _conv_args); in configure()
185 …ut_workspace_size = _winograd_impl.input_transform->get_working_space_size(*_conv_args, nthreads); in configure()
186 …ut_workspace_size = _winograd_impl.output_transform->get_working_space_size(*_conv_args, nthreads); in configure()
256 …l = std::make_unique<CpuWinogradConv2dTransformInputKernel>(_winograd_impl, *_conv_args, nthreads); in configure()
262 … = std::make_unique<CpuWinogradConv2dTransformOutputKernel>(_winograd_impl, *_conv_args, nthreads); in configure()
407 *_conv_args, in prepare()
|
| D | CpuWinogradConv2d.h | 119 …std::unique_ptr<arm_conv::ConvolutionArgs> _conv_args; // Make it unique ptr because this type doe… variable
|