Home
last modified time | relevance | path

Searched refs:_winograd_impl (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/cpu/operators/
DCpuWinogradConv2d.cpp145 _winograd_impl{},
174 …lementation(src, weights, dst, conv_info, act_info, enable_fast_math, &_winograd_impl, _conv_args); in configure()
177 …arm_compute::logging::LogLevel::INFO, "Using input transform: %s\n", _winograd_impl.input_transfor… in configure()
178 …rm_compute::logging::LogLevel::INFO, "Using weight transform: %s\n", _winograd_impl.input_transfor… in configure()
179 …rm_compute::logging::LogLevel::INFO, "Using output transform: %s\n", _winograd_impl.input_transfor… in configure()
181 …nst bool has_impl = ((_winograd_impl.input_transform != nullptr) && (_winograd_impl.output_transfo… in configure()
185 …const size_t input_workspace_size = _winograd_impl.input_transform->get_working_space_size(*_conv… in configure()
186 …const size_t output_workspace_size = _winograd_impl.output_transform->get_working_space_size(*_con… in configure()
193 const auto &wds = _winograd_impl.winograd_spec; in configure()
197 … const uint32_t m = _winograd_impl.gemm_args->_Msize; // Total number of tiles in configure()
[all …]
DCpuWinogradConv2d.h120 arm_conv::winograd::WinogradImpl _winograd_impl; variable
/external/ComputeLibrary/src/cpu/kernels/
DCpuWinogradConv2dKernel.cpp32 : _winograd_impl{ w_impl }, _conv_args{ _c_args }, _nthreads{ nthreads } in CpuWinogradConv2dTransformInputKernel()
55 _winograd_impl.input_transform->execute( in run_op()
62 _winograd_impl.winograd_spec, in run_op()
69 : _winograd_impl{ w_impl }, _conv_args{ _c_args }, _nthreads{ nthreads } in CpuWinogradConv2dTransformOutputKernel()
100 _winograd_impl.output_transform->execute( in run_op()
103 _winograd_impl.winograd_spec, in run_op()
DCpuWinogradConv2dKernel.h67 arm_conv::winograd::WinogradImpl &_winograd_impl;
97 arm_conv::winograd::WinogradImpl &_winograd_impl;