Lines Matching refs:quantAttr

182   auto quantAttr = buildConvOpQuantizationAttr(builder, input, weight);  in buildConvOpWithQuantInfo()  local
183 if (quantAttr) { in buildConvOpWithQuantInfo()
184 result.addAttribute("quantization_info", quantAttr); in buildConvOpWithQuantInfo()
203 auto quantAttr = ::buildConvOpQuantizationAttr(builder, input, weight); in buildTransConvOpWithQuantInfo() local
205 if (quantAttr) { in buildTransConvOpWithQuantInfo()
206 result.addAttribute("quantization_info", quantAttr); in buildTransConvOpWithQuantInfo()
221 auto quantAttr = ::buildConvOpQuantizationAttr(builder, input, weight); in buildFCOpWithQuantInfo() local
222 if (quantAttr) { in buildFCOpWithQuantInfo()
223 result.addAttribute("quantization_info", quantAttr); in buildFCOpWithQuantInfo()
239 auto quantAttr = ::buildMatMulOpQuantizationAttr(builder, a, b); in buildMatMulOpWithQuantInfo() local
241 if (quantAttr) { in buildMatMulOpWithQuantInfo()
242 result.addAttribute("quantization_info", quantAttr); in buildMatMulOpWithQuantInfo()
280 auto quantAttr = buildUnaryOpQuantizationAttr(builder, input, outputType); in buildAvgPool2dOpWithQuantInfo() local
281 if (quantAttr) in buildAvgPool2dOpWithQuantInfo()
282 result.addAttribute("quantization_info", quantAttr); in buildAvgPool2dOpWithQuantInfo()
293 auto quantAttr = buildUnaryOpQuantizationAttr(builder, input, outputType); in buildUnaryOpWithQuantInfo() local
294 if (quantAttr) in buildUnaryOpWithQuantInfo()
295 result.addAttribute("quantization_info", quantAttr); in buildUnaryOpWithQuantInfo()
306 auto quantAttr = buildPadOpQuantizationAttr(builder, input); in buildPadOpWithQuantInfo() local
307 if (quantAttr) in buildPadOpWithQuantInfo()
308 result.addAttribute("quantization_info", quantAttr); in buildPadOpWithQuantInfo()
320 auto quantAttr = buildPadOpQuantizationAttr(builder, input); in buildExplicitValuePadOpWithQuantInfo() local
321 if (quantAttr) in buildExplicitValuePadOpWithQuantInfo()
322 result.addAttribute("quantization_info", quantAttr); in buildExplicitValuePadOpWithQuantInfo()