Searched defs:conv_args (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/ |
H A D | winograd_implementations.hpp | 70 … *transform, const MethodConstraints &c, const CPUInfo *ci, const ConvolutionArgs &conv_args, cons… in output_transform_constraints_met() 155 const CPUInfo *ci, const ConvolutionArgs &conv_args, const WinogradConfig *cfg in get_weight_transforms() 186 const CPUInfo *ci, const ConvolutionArgs &conv_args, const WinogradConfig *cfg in get_input_transforms() 213 const CPUInfo *ci, const ConvolutionArgs &conv_args, const WinogradConfig *cfg in get_output_transforms() 240 const ConvolutionArgs &conv_args, in get_implementation()
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
H A D | CpuWinogradConv2d.cpp | 85 …arm_conv::winograd::WinogradImpl *winograd_impl, std::unique_ptr<arm_conv::ConvolutionArgs> &conv_… in get_winograd_kernel_implementation() 306 std::unique_ptr<arm_conv::ConvolutionArgs> conv_args; in validate() local
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
H A D | Im2Col.cpp | 46 const auto conv_args = combine(combine(combine(combine(conv_filter_sizes, framework::datase… variable
|