Home
last modified time | relevance | path

Searched refs:input_transform (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/
H A Dwinograd_implementations.hpp97 namespace input_transform namespace
185 inline std::vector<const input_transform::ITransform *> get_input_transforms( in get_input_transforms()
193 std::vector<const input_transform::ITransform *> input_transforms; in get_input_transforms()
194 for (auto impl = input_transform::implementation_list<TIn, TWinogradIn>(); in get_input_transforms()
272 for (auto input_transform = input_transforms.cbegin(); in get_implementation() local
273 !success && input_transform != input_transforms.cend(); in get_implementation()
274 input_transform++) in get_implementation()
278 if ((*input_transform)->get_input_rows() == (*output_transform)->get_input_rows() && in get_implementation()
279 (*input_transform)->get_input_cols() == (*output_transform)->get_input_cols()) in get_implementation()
282 dest.input_transform = *input_transform; in get_implementation()
[all …]
H A Dinput_transforms_fp16.cpp35 namespace input_transform { namespace
H A Dinput_transforms_fp32.cpp33 namespace input_transform { namespace
H A Dinput_transform.hpp38 namespace input_transform { namespace
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuWinogradConv2d.cpp177 …gging::LogLevel::INFO, "Using input transform: %s\n", _winograd_impl.input_transform->get_name().c… in configure()
178 …ging::LogLevel::INFO, "Using weight transform: %s\n", _winograd_impl.input_transform->get_name().c… in configure()
179 …ging::LogLevel::INFO, "Using output transform: %s\n", _winograd_impl.input_transform->get_name().c… in configure()
181 …const bool has_impl = ((_winograd_impl.input_transform != nullptr) && (_winograd_impl.output_trans… in configure()
185 …const size_t input_workspace_size = _winograd_impl.input_transform->get_working_space_size(*_conv… in configure()
310 …ogging::LogLevel::INFO, "Using input transform: %s\n", winograd_impl.input_transform->get_name().c… in validate()
311 …gging::LogLevel::INFO, "Using weight transform: %s\n", winograd_impl.input_transform->get_name().c… in validate()
312 …gging::LogLevel::INFO, "Using output transform: %s\n", winograd_impl.input_transform->get_name().c… in validate()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/assembly/
H A Dwinograd.hpp129 namespace input_transform namespace
213 const input_transform::ITransform *input_transform = nullptr; member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/input_transforms/
H A Da64_fp32_6x6.cpp31 namespace input_transform { namespace
H A Dsme_fp32_mla_6x6.cpp31 namespace input_transform { namespace
H A Dsve_fp32_6x6.cpp30 namespace input_transform { namespace
H A Darm_fp32_1x8.cpp30 namespace input_transform { namespace
H A Darm_fp32_6x6.cpp32 namespace input_transform { namespace
H A Darm_fp32_4x4.cpp30 namespace input_transform { namespace
H A Da64_fp16_6x6.cpp31 namespace input_transform { namespace
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuWinogradConv2dKernel.cpp55 _winograd_impl.input_transform->execute( in run_op()