Lines Matching refs:ConvolutionMethod
62 case ConvolutionMethod::WINOGRAD: in configure()
69 case ConvolutionMethod::GEMM: in configure()
76 case ConvolutionMethod::GEMM_CONV2D: in configure()
83 case ConvolutionMethod::DIRECT: in configure()
90 case ConvolutionMethod::FFT: in configure()
111 case ConvolutionMethod::WINOGRAD: in validate()
114 case ConvolutionMethod::GEMM: in validate()
117 case ConvolutionMethod::GEMM_CONV2D: in validate()
120 case ConvolutionMethod::DIRECT: in validate()
123 case ConvolutionMethod::FFT: in validate()
134 ConvolutionMethod NEConvolutionLayer::get_convolution_method(const ITensorInfo *input, const ITenso… in get_convolution_method()
149 using ConfigurationMethod = std::pair<ConvolutionConfiguration, ConvolutionMethod>; in get_convolution_method()
154 … 27U), Size2D(5U, 5U), Size2D(48U, 128U), PadStrideInfo(1U, 1U, 2U, 2U)), ConvolutionMethod::GEMM), in get_convolution_method()
156 …, 224U), Size2D(3U, 3U), Size2D(3U, 64U), PadStrideInfo(1U, 1U, 1U, 1U)), ConvolutionMethod::GEMM), in get_convolution_method()
158 …U), PadStrideInfo(2U, 2U, 0U, 1U, 0U, 1U, DimensionRoundingType::FLOOR)), ConvolutionMethod::GEMM), in get_convolution_method()
160 …4U), PadStrideInfo(2U, 2U, 0U, 1U, 0U, 1U, DimensionRoundingType::FLOOR)), ConvolutionMethod::GEMM) in get_convolution_method()
181 return ConvolutionMethod::GEMM; in get_convolution_method()
190 return ConvolutionMethod::DIRECT; in get_convolution_method()
194 return ConvolutionMethod::FFT; in get_convolution_method()
198 return ConvolutionMethod::GEMM; in get_convolution_method()
229 return ConvolutionMethod::GEMM; in get_convolution_method()
236 return ConvolutionMethod::GEMM; in get_convolution_method()
241 return ConvolutionMethod::WINOGRAD; in get_convolution_method()
245 return ConvolutionMethod::GEMM_CONV2D; in get_convolution_method()
247 return ConvolutionMethod::GEMM; in get_convolution_method()