Searched refs:conv_algorithm (Results 1 – 3 of 3) sorted by relevance
548 const ConvolutionMethod conv_algorithm = conv_node->convolution_method(); in fuse_convolution_with_post_ops() local549 if(conv_algorithm != ConvolutionMethod::GEMM) in fuse_convolution_with_post_ops()646 const ConvolutionMethod conv_algorithm = conv_node->convolution_method(); in fuse_convolution_batch_normalization_with_post_ops() local647 if(conv_algorithm != ConvolutionMethod::GEMM) in fuse_convolution_batch_normalization_with_post_ops()
155 const ConvolutionMethod conv_algorithm = node.convolution_method(); in validate_convolution_layer() local161 switch(conv_algorithm) in validate_convolution_layer()
483 const ConvolutionMethod conv_algorithm = node.convolution_method(); in create_convolution_layer() local492 if(conv_algorithm == ConvolutionMethod::Winograd) in create_convolution_layer()499 else if(conv_algorithm == ConvolutionMethod::Direct) in create_convolution_layer()506 else if(conv_algorithm == ConvolutionMethod::GEMM) in create_convolution_layer()