Lines Matching refs:ConvolutionMethod
66 case ConvolutionMethod::WINOGRAD: in configure()
74 case ConvolutionMethod::DIRECT: in configure()
82 case ConvolutionMethod::GEMM: in configure()
89 case ConvolutionMethod::FFT: in configure()
112 case ConvolutionMethod::WINOGRAD: in validate()
119 case ConvolutionMethod::DIRECT: in validate()
126 case ConvolutionMethod::GEMM: in validate()
132 case ConvolutionMethod::FFT: in validate()
146 ConvolutionMethod CLConvolutionLayer::get_convolution_method(const ITensorInfo *input, const ITenso… in get_convolution_method()
161 using ConfigurationMethod = std::pair<ConvolutionConfiguration, ConvolutionMethod>; in get_convolution_method()
166 …), Size2D(48U, 128U), PadStrideInfo(1U, 1U, 2U, 2U), DataLayout::NCHW), ConvolutionMethod::DIRECT), in get_convolution_method()
168 …3U), Size2D(3U, 64U), PadStrideInfo(1U, 1U, 1U, 1U), DataLayout::NCHW), ConvolutionMethod::DIRECT), in get_convolution_method()
170 …2U, 2U, 0U, 1U, 0U, 1U, DimensionRoundingType::FLOOR), DataLayout::NCHW), ConvolutionMethod::GEMM), in get_convolution_method()
172 …2U, 2U, 0U, 1U, 0U, 1U, DimensionRoundingType::FLOOR), DataLayout::NCHW), ConvolutionMethod::GEMM), in get_convolution_method()
174 …2U, 2U, 0U, 1U, 0U, 1U, DimensionRoundingType::FLOOR), DataLayout::NHWC), ConvolutionMethod::GEMM), in get_convolution_method()
176 …2U, 2U, 0U, 1U, 0U, 1U, DimensionRoundingType::FLOOR), DataLayout::NHWC), ConvolutionMethod::GEMM), in get_convolution_method()
198 return ConvolutionMethod::GEMM; in get_convolution_method()
206 return ConvolutionMethod::DIRECT; in get_convolution_method()
210 return ConvolutionMethod::FFT; in get_convolution_method()
214 return ConvolutionMethod::GEMM; in get_convolution_method()
216 …r, output, conv_info, act_info, enable_fast_math)) ? ConvolutionMethod::WINOGRAD : ConvolutionMeth… in get_convolution_method()