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 D | winograd_implementations.hpp | 97 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 D | input_transforms_fp16.cpp | 35 namespace input_transform { namespace
|
| H A D | input_transforms_fp32.cpp | 33 namespace input_transform { namespace
|
| H A D | input_transform.hpp | 38 namespace input_transform { namespace
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuWinogradConv2d.cpp | 177 …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 D | winograd.hpp | 129 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 D | a64_fp32_6x6.cpp | 31 namespace input_transform { namespace
|
| H A D | sme_fp32_mla_6x6.cpp | 31 namespace input_transform { namespace
|
| H A D | sve_fp32_6x6.cpp | 30 namespace input_transform { namespace
|
| H A D | arm_fp32_1x8.cpp | 30 namespace input_transform { namespace
|
| H A D | arm_fp32_6x6.cpp | 32 namespace input_transform { namespace
|
| H A D | arm_fp32_4x4.cpp | 30 namespace input_transform { namespace
|
| H A D | a64_fp16_6x6.cpp | 31 namespace input_transform { namespace
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuWinogradConv2dKernel.cpp | 55 _winograd_impl.input_transform->execute( in run_op()
|