Searched refs:outputShape (Results 1 – 3 of 3) sorted by relevance
174 ArrayAttr dilation, ArrayAttr outputShape) { in buildTransConvOpWithQuantInfo() argument179 result.addAttribute("out_shape", outputShape); in buildTransConvOpWithQuantInfo()233 auto outputShape = outputShapedType.getShape(); in buildMatMulOpWithQuantInfo() local240 auto accType = RankedTensorType::get(outputShape, accElementType); in buildMatMulOpWithQuantInfo()
267 auto outputShape = outputShapedType.getShape(); in buildConvOpResultTypeInfo() local274 auto accType = RankedTensorType::get(outputShape, accElementType); in buildConvOpResultTypeInfo()
127 "ArrayAttr":$outputShape),132 outputShape);