Lines Matching refs:inputShape
346 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
350 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
356 outShape.reserve(inputShape.getRank() - 1); in inferReturnTypeComponents()
357 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
360 outShape.push_back(inputShape.getDimSize(i)); in inferReturnTypeComponents()
428 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
436 if (inputShape.hasRank()) { in inferReturnTypeComponents()
437 outShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
487 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
493 if (!inputShape.hasRank() && !paddingShape.hasRank()) { in inferReturnTypeComponents()
500 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
514 outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
524 outputShape.reserve(inputShape.getRank()); in inferReturnTypeComponents()
525 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
526 if (inputShape.isDynamicDim(i)) { in inferReturnTypeComponents()
531 outputShape.push_back(inputShape.getDimSize(i) + paddingValues[i * 2] + in inferReturnTypeComponents()
558 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
560 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
566 inputShape.getDims(inferredReturnShapes[0]); in inferReturnTypeComponents()
576 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
578 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
593 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
594 int dim = inputShape.getDimSize(i); in inferReturnTypeComponents()
609 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
617 if (!inputShape.hasRank() || !inputShape.hasStaticShape()) { in inferReturnTypeComponents()
625 int64_t numElements = inputShape.getNumElements(); in inferReturnTypeComponents()
647 ShapeAdaptor inputShape = operands.getShape(0); in inferReturnTypeComponents() local
652 if (!inputShape.hasRank() || !permsShape.hasRank() || in inferReturnTypeComponents()
660 if (permsShape.getDimSize(0) != inputShape.getRank()) { in inferReturnTypeComponents()
667 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
674 if (inputShape.getRank() == 0) { in inferReturnTypeComponents()
681 for (int i = 1, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
682 if (inputShape.getDimSize(0) != inputShape.getDimSize(i)) { in inferReturnTypeComponents()
691 outputShape.resize(inputShape.getRank(), inputShape.getDimSize(0)); in inferReturnTypeComponents()
696 outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
700 outputShape.reserve(inputShape.getRank()); in inferReturnTypeComponents()
701 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
702 outputShape[i] = inputShape.getDimSize(permShape.getDimSize(i)); in inferReturnTypeComponents()
746 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents() local
747 if (inputShape.hasRank()) { in inferReturnTypeComponents()
748 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
749 outputShape[3] = inputShape.getDimSize(3); in inferReturnTypeComponents()
751 inHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
752 inWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
826 ShapeAdaptor inputShape = operands.getShape(2); in inferReturnTypeComponents() local
827 if (inputShape.hasRank()) { in inferReturnTypeComponents()
829 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
831 outputShape[2] = inputShape.getDimSize(2); in inferReturnTypeComponents()
974 ShapeAdaptor inputShape = operands.getShape(0); in NARY_SHAPE_INFER() local
979 if (!inputShape) { in NARY_SHAPE_INFER()
985 outputShape[0] = inputShape.getDimSize(0); in NARY_SHAPE_INFER()
986 outputShape[3] = inputShape.getDimSize(3); in NARY_SHAPE_INFER()
988 int32_t height = inputShape.getDimSize(1); in NARY_SHAPE_INFER()
989 int32_t width = inputShape.getDimSize(2); in NARY_SHAPE_INFER()
1027 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents() local
1028 if (inputShape.hasRank()) { in inferReturnTypeComponents()
1029 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
1030 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1031 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1096 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents() local
1097 if (inputShape.hasRank()) { in inferReturnTypeComponents()
1098 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
1099 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1100 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1101 inputDepth = inputShape.getDimSize(3); in inferReturnTypeComponents()
1188 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents() local
1189 if (inputShape.hasRank()) { in inferReturnTypeComponents()
1190 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
1191 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1192 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1193 inputChannels = inputShape.getDimSize(3); in inferReturnTypeComponents()
1266 ShapeAdaptor inputShape = operands.getShape(adaptor.getInput()); in inferReturnTypeComponents() local
1267 if (inputShape.hasRank()) { in inferReturnTypeComponents()
1269 ? inputShape.getDimSize(0) in inferReturnTypeComponents()
1271 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1272 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()