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()
|