198                                           ArrayAttr outputShape) {  in buildTransConvOpWithQuantInfo()  argument
202   result.addAttribute("out_shape", outputShape);  in buildTransConvOpWithQuantInfo()
374   llvm::SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
383       outputShape.resize(operandShape.getRank(), ShapedType::kDynamicSize);  in inferReturnTypeComponents()
389       if (outputShape[i] == ShapedType::kDynamicSize)  in inferReturnTypeComponents()
390         outputShape[i] = operandShape.getDimSize(i);  in inferReturnTypeComponents()
391       if (outputShape[i] != operandShape.getDimSize(i))  in inferReturnTypeComponents()
418   outputShape[axis] = concatDimSize;  in inferReturnTypeComponents()
420   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
489   SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
506     outputShape.resize(paddingShape.getDimSize(0), ShapedType::kDynamicSize);  in inferReturnTypeComponents()
507     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
514     outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize);  in inferReturnTypeComponents()
515     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
524   outputShape.reserve(inputShape.getRank());  in inferReturnTypeComponents()
527       outputShape.push_back(ShapedType::kDynamicSize);  in inferReturnTypeComponents()
531     outputShape.push_back(inputShape.getDimSize(i) + paddingValues[i * 2] +  in inferReturnTypeComponents()
535   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
544   SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
545   outputShape.reserve(sizes.size());  in inferReturnTypeComponents()
547     outputShape.push_back(val.cast<IntegerAttr>().getValue().getSExtValue());  in inferReturnTypeComponents()
550   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
577   SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
579     outputShape.resize(multiples.size(), ShapedType::kDynamicSize);  in inferReturnTypeComponents()
580     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
592   outputShape.reserve(multiples.size());  in inferReturnTypeComponents()
597     outputShape.push_back(dim);  in inferReturnTypeComponents()
600   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
666   SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
668     outputShape.resize(permsShape.getDimSize(0), ShapedType::kDynamicSize);  in inferReturnTypeComponents()
669     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
675     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
691     outputShape.resize(inputShape.getRank(), inputShape.getDimSize(0));  in inferReturnTypeComponents()
692     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
696   outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize);  in inferReturnTypeComponents()
700     outputShape.reserve(inputShape.getRank());  in inferReturnTypeComponents()
702       outputShape[i] = inputShape.getDimSize(permShape.getDimSize(i));  in inferReturnTypeComponents()
706   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
714   llvm::SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
715   outputShape.resize(3, ShapedType::kDynamicSize);  in inferReturnTypeComponents()
719     outputShape[0] = valuesShape.getDimSize(0);  in inferReturnTypeComponents()
720     outputShape[2] = valuesShape.getDimSize(2);  in inferReturnTypeComponents()
725     if (outputShape[0] == ShapedType::kDynamicSize)  in inferReturnTypeComponents()
726       outputShape[0] = indicesShape.getDimSize(0);  in inferReturnTypeComponents()
727     if (outputShape[1] == ShapedType::kDynamicSize)  in inferReturnTypeComponents()
728       outputShape[1] = indicesShape.getDimSize(1);  in inferReturnTypeComponents()
731   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
740   llvm::SmallVector<int64_t, 4> outputShape;  in inferReturnTypeComponents()  local
741   outputShape.resize(4, ShapedType::kDynamicSize);  in inferReturnTypeComponents()
748     outputShape[0] = inputShape.getDimSize(0);  in inferReturnTypeComponents()
749     outputShape[3] = inputShape.getDimSize(3);  in inferReturnTypeComponents()
758   outputShape[1] = newShape[0];  in inferReturnTypeComponents()
759   outputShape[2] = newShape[1];  in inferReturnTypeComponents()
778   if (outputShape[1] == ShapedType::kDynamicSize && inHeight >= 0 && fpMode) {  in inferReturnTypeComponents()
781     outputShape[1] = std::ceil(sizeFp) + round;  in inferReturnTypeComponents()
784   if (outputShape[2] == ShapedType::kDynamicSize && inWidth >= 0 && fpMode) {  in inferReturnTypeComponents()
787     outputShape[2] = std::ceil(sizeFp) + round;  in inferReturnTypeComponents()
790   if (outputShape[1] == ShapedType::kDynamicSize && inHeight >= 0 && !fpMode) {  in inferReturnTypeComponents()
793     outputShape[1] = size + 1;  in inferReturnTypeComponents()
796   if (outputShape[2] == ShapedType::kDynamicSize && inWidth >= 0 && !fpMode) {  in inferReturnTypeComponents()
799     outputShape[2] = size + 1;  in inferReturnTypeComponents()
802   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
810   llvm::SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
811   outputShape.resize(3, ShapedType::kDynamicSize);  in inferReturnTypeComponents()
815     outputShape[0] = valuesInShape.getDimSize(0);  in inferReturnTypeComponents()
816     outputShape[1] = valuesInShape.getDimSize(1);  in inferReturnTypeComponents()
817     outputShape[2] = valuesInShape.getDimSize(2);  in inferReturnTypeComponents()
822     if (outputShape[0] == ShapedType::kDynamicSize)  in inferReturnTypeComponents()
823       outputShape[0] = indicesShape.getDimSize(0);  in inferReturnTypeComponents()
828     if (outputShape[0] == ShapedType::kDynamicSize)  in inferReturnTypeComponents()
829       outputShape[0] = inputShape.getDimSize(0);  in inferReturnTypeComponents()
830     if (outputShape[2] == ShapedType::kDynamicSize)  in inferReturnTypeComponents()
831       outputShape[2] = inputShape.getDimSize(2);  in inferReturnTypeComponents()
834   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
846   SmallVector<int64_t> outputShape;  in ReduceInferReturnTypes()  local
847   operandShape.getDims(outputShape);  in ReduceInferReturnTypes()
849   outputShape[axisVal] = 1;  in ReduceInferReturnTypes()
850   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in ReduceInferReturnTypes()
975   llvm::SmallVector<int64_t> outputShape;  in NARY_SHAPE_INFER()  local
976   outputShape.resize(4, -1);  in NARY_SHAPE_INFER()
980     inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in NARY_SHAPE_INFER()
985   outputShape[0] = inputShape.getDimSize(0);  in NARY_SHAPE_INFER()
986   outputShape[3] = inputShape.getDimSize(3);  in NARY_SHAPE_INFER()
1001     outputShape[1] = padded / stride[0] + 1;  in NARY_SHAPE_INFER()
1006     outputShape[2] = padded / stride[1] + 1;  in NARY_SHAPE_INFER()
1009   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in NARY_SHAPE_INFER()
1017   llvm::SmallVector<int64_t> outputShape(4, ShapedType::kDynamicSize);  in inferReturnTypeComponents()  local
1029     outputShape[0] = inputShape.getDimSize(0);  in inferReturnTypeComponents()
1037     outputShape[3] = weightShape.getDimSize(0);  in inferReturnTypeComponents()
1045     outputShape[3] = ShapedType::isDynamic(outputShape[3])  in inferReturnTypeComponents()
1047                          : outputShape[3];  in inferReturnTypeComponents()
1063     outputShape[1] = (unstridedResult - 1) / stride[0] + 1;  in inferReturnTypeComponents()
1071     outputShape[2] = (unstridedResult - 1) / stride[1] + 1;  in inferReturnTypeComponents()
1074   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
1084   llvm::SmallVector<int64_t> outputShape(5, ShapedType::kDynamicSize);  in inferReturnTypeComponents()  local
1098     outputShape[0] = inputShape.getDimSize(0);  in inferReturnTypeComponents()
1107     outputShape[4] = weightShape.getDimSize(0);  in inferReturnTypeComponents()
1116     outputShape[4] =  in inferReturnTypeComponents()
1117         (outputShape[4] == -1) ? biasShape.getDimSize(0) : outputShape[4];  in inferReturnTypeComponents()
1133     outputShape[1] = (unstridedResult - 1) / stride[0] + 1;  in inferReturnTypeComponents()
1141     outputShape[2] = (unstridedResult - 1) / stride[1] + 1;  in inferReturnTypeComponents()
1149     outputShape[3] = (unstridedResult - 1) / stride[2] + 1;  in inferReturnTypeComponents()
1152   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
1176   llvm::SmallVector<int64_t> outputShape(4, ShapedType::kDynamicSize);  in inferReturnTypeComponents()  local
1190     outputShape[0] = inputShape.getDimSize(0);  in inferReturnTypeComponents()
1211     outputShape[3] = inputChannels * depthChannels;  in inferReturnTypeComponents()
1217     outputShape[3] = ShapedType::isDynamic(outputShape[3])  in inferReturnTypeComponents()
1219                          : outputShape[3];  in inferReturnTypeComponents()
1235     outputShape[1] = (unstridedResult - 1) / stride[0] + 1;  in inferReturnTypeComponents()
1243     outputShape[2] = (unstridedResult - 1) / stride[1] + 1;  in inferReturnTypeComponents()
1246   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()
1257   llvm::SmallVector<int64_t> outputShape;  in inferReturnTypeComponents()  local
1258   getI64Values(adaptor.getOutShape(), outputShape);  in inferReturnTypeComponents()
1268     outputShape[0] = ShapedType::isDynamic(outputShape[0])  in inferReturnTypeComponents()
1270                          : outputShape[0];  in inferReturnTypeComponents()
1278     outputShape[3] = ShapedType::isDynamic(outputShape[3])  in inferReturnTypeComponents()
1280                          : outputShape[3];  in inferReturnTypeComponents()
1288     outputShape[3] = ShapedType::isDynamic(outputShape[3])  in inferReturnTypeComponents()
1290                          : outputShape[3];  in inferReturnTypeComponents()
1303     outputShape[1] = outputShape[1] == -1 ? calculateSize : outputShape[1];  in inferReturnTypeComponents()
1310     outputShape[2] = outputShape[2] == -1 ? calculateSize : outputShape[2];  in inferReturnTypeComponents()
1313   inferredReturnShapes.push_back(ShapedTypeComponents(outputShape));  in inferReturnTypeComponents()