• Home
  • Raw
  • Download

Lines Matching refs:quantAttr

159   auto quantAttr = buildConvOpQuantizationAttr(builder, input, weight);  in buildConvOpWithQuantInfo()  local
160 if (quantAttr) { in buildConvOpWithQuantInfo()
161 result.addAttribute("quantization_info", quantAttr); in buildConvOpWithQuantInfo()
180 auto quantAttr = ::buildConvOpQuantizationAttr(builder, input, weight); in buildTransConvOpWithQuantInfo() local
182 if (quantAttr) { in buildTransConvOpWithQuantInfo()
183 result.addAttribute("quantization_info", quantAttr); in buildTransConvOpWithQuantInfo()
198 auto quantAttr = ::buildConvOpQuantizationAttr(builder, input, weight); in buildFCOpWithQuantInfo() local
199 if (quantAttr) { in buildFCOpWithQuantInfo()
200 result.addAttribute("quantization_info", quantAttr); in buildFCOpWithQuantInfo()
216 auto quantAttr = ::buildMatMulOpQuantizationAttr(builder, a, b); in buildMatMulOpWithQuantInfo() local
218 if (quantAttr) { in buildMatMulOpWithQuantInfo()
219 result.addAttribute("quantization_info", quantAttr); in buildMatMulOpWithQuantInfo()
259 auto quantAttr = buildUnaryOpQuantizationAttr(builder, input, outputType); in buildAvgPool2dOpWithQuantInfo() local
260 if (quantAttr) in buildAvgPool2dOpWithQuantInfo()
261 result.addAttribute("quantization_info", quantAttr); in buildAvgPool2dOpWithQuantInfo()
272 auto quantAttr = buildUnaryOpQuantizationAttr(builder, input, outputType); in buildUnaryOpWithQuantInfo() local
273 if (quantAttr) in buildUnaryOpWithQuantInfo()
274 result.addAttribute("quantization_info", quantAttr); in buildUnaryOpWithQuantInfo()
285 auto quantAttr = buildPadOpQuantizationAttr(builder, input); in buildPadOpWithQuantInfo() local
286 if (quantAttr) in buildPadOpWithQuantInfo()
287 result.addAttribute("quantization_info", quantAttr); in buildPadOpWithQuantInfo()