Lines Matching defs:inputTy

147     auto inputTy = llvm::cast<ShapedType>(input.getType());
148     if (!inputTy.hasRank())
152     for (int i = 0; i < inputTy.getRank(); ++i)
153       if (inputTy.isDynamicDim(i))
166       auto sz = inputTy.getDimSize(idx);
177     newShape.reserve(inputTy.getRank());
178     for (int i = 0, s = inputTy.getRank(); i < s; ++i)
179       newShape.push_back(inputTy.getDimSize(permValues[i]));
204     ShapedType inputTy = llvm::cast<ShapedType>(input.getType());
205     Type elementTy = inputTy.getElementType();
585   auto inputTy = llvm::dyn_cast<RankedTensorType>(getInput().getType());
587   if (!inputTy || !outputTy || !inputTy.hasStaticShape() ||
591   if (inputTy.getDimSize(getAxis()) == 1)
893     ShapedType inputTy = llvm::cast<ShapedType>(getInput().getType());         \
894     if (!inputTy.hasRank())                                                    \
896     if (inputTy != getType())                                                  \
898     if (inputTy.getRank() == 0 || inputTy.getDimSize(getAxis()) == 1)          \
912   auto inputTy = llvm::dyn_cast<RankedTensorType>(getInput1().getType());
915   if (!inputTy || !outputTy)
921   if (inputTy == outputTy && inputTy.getNumDynamicDims() < 2)
932   if (!inputTy.getElementType().isIntOrIndexOrFloat())
994   auto inputTy = llvm::cast<RankedTensorType>(input.getType());
996   if (inputTy != resultTy)
1020   auto inputTy = llvm::dyn_cast<RankedTensorType>(getInput1().getType());
1023   if (!inputTy || !outputTy)
1026   if (inputTy == outputTy && inputTy.hasStaticShape())
1033   if (!inputTy.getElementType().isIntOrIndexOrFloat() ||
1042   if (inputTy.hasStaticShape() && outputTy.hasStaticShape() &&