Lines Matching defs:inputTy

1075   auto inputTy = cast<ShapedType>(op->getOperand(0).getType());
1082   for (unsigned i = 0; i < inputTy.getRank(); i++) {
1084       reduceShape.push_back(inputTy.getDimSize(i));
1085       if (inputTy.isDynamicDim(i))
1173     auto inputTy = cast<ShapedType>(op.getInput().getType());
1175     unsigned rank = inputTy.getRank();
1182     if (!isa<IntegerType>(inputTy.getElementType()))
1358     auto inputTy = cast<RankedTensorType>(input.getType());
1362     auto inputH = inputTy.getDimSize(1);
1363     auto inputW = inputTy.getDimSize(2);
1376     if (inputTy == resultTy) {
1391         RankedTensorType::get({inputTy.getDimSize(0), inputTy.getDimSize(3)},
1392                               inputTy.getElementType());
1398     if (inputTy.isDynamicDim(0))
1400     if (inputTy.isDynamicDim(3))
1417           if (inputTy.getElementType() != resultTy.getElementType()) {
1455     auto inputTy = dyn_cast<RankedTensorType>(input.getType());
1458     if (!inputTy || !resultTy)
1462     auto batch = inputTy.getDimSize(0);
1463     auto channels = inputTy.getDimSize(3);
1464     auto inputH = inputTy.getDimSize(1);
1465     auto inputW = inputTy.getDimSize(2);
1509     if (inputTy.isDynamicDim(0))
1511     if (inputTy.isDynamicDim(3))
1551     auto inputTy = cast<ShapedType>(input.getType());
1558     auto imageH = inputTy.getShape()[1];
1559     auto imageW = inputTy.getShape()[2];
1844     auto inputTy = cast<ShapedType>(input.getType());
1849     for (int i = 0; i < inputTy.getRank(); i++) {
1850       if (inputTy.isDynamicDim(i)) {
1859                            .create<tensor::EmptyOp>(loc, inputTy.getShape(),
1860                                                     inputTy.getElementType(),
1871           for (unsigned int i = 0; i < inputTy.getRank(); i++) {
1906     auto inputTy = cast<ShapedType>(input.getType());
1907     auto inputShape = inputTy.getShape();
1909     auto elementTy = inputTy.getElementType();
1910     int64_t rank = inputTy.getRank();
1925     for (int i = 0; i < inputTy.getRank(); i++) {
1926       if (inputTy.isDynamicDim(i) || multiples[i] == -1) {
1983     auto inputTy = cast<ShapedType>(input.getType());
1985     auto inElementTy = inputTy.getElementType();
1996     for (int i = 0; i < inputTy.getRank(); i++) {
1997       if (inputTy.isDynamicDim(i) && i != axis) {
2038     iteratorTypes.resize(inputTy.getRank(), utils::IteratorType::parallel);
2043     for (int i = 0, rank = inputTy.getRank(); i != rank; ++i) {
2178     auto inputTy = cast<ShapedType>(input.getType());
2182     auto inputElementTy = inputTy.getElementType();
2188       if (inputTy.isDynamicDim(i)) {