Searched refs:src_ch_count (Results 1 – 2 of 2) sorted by relevance
151 const int src_ch_count = std::min(4, weights_shape.i - s * 4); in GenerateConvolutionConstantCode() local153 src_ch_count == 1 ? "" : std::to_string(src_ch_count); in GenerateConvolutionConstantCode()155 const std::string s_postfix = postfixes[src_ch_count - 1]; in GenerateConvolutionConstantCode()182 c += GenerateConv(src_ch_count, dst_ch_count, use_dot_conv, in GenerateConvolutionConstantCode()185 filters_counter += use_dot_conv ? dst_ch_count : src_ch_count; in GenerateConvolutionConstantCode()
441 const int src_ch_count = std::min(4, attr.weights.shape.i - d * 4); in AddDepthwiseConvNode() local442 const std::string s_postfix = postfixes[src_ch_count - 1]; in AddDepthwiseConvNode()