Searched refs:inputTy (Results 1 – 5 of 5) sorted by relevance
260 ShapedType inputTy = input.getType().cast<ShapedType>(); in matchAndRewrite() local261 Type elementTy = inputTy.getElementType(); in matchAndRewrite()453 ShapedType inputTy = input().getType().cast<ShapedType>(); \454 if (!inputTy.hasRank()) \456 if (inputTy.getDimSize(axis()) == 1) \470 auto inputTy = getInput1().getType().dyn_cast<RankedTensorType>(); in REDUCE_FOLDER() local473 if (!inputTy || !outputTy || inputTy != outputTy) in REDUCE_FOLDER()491 auto inputTy = getInput().getType().dyn_cast<RankedTensorType>(); in fold() local494 if (!inputTy || !outputTy || inputTy != outputTy) in fold()496 if (inputTy.hasStaticShape()) in fold()
40 Type inputETy = inputTy.getElementType(); in applyPad()41 auto inputShape = inputTy.getShape(); in applyPad()105 Type inputETy = inputTy.getElementType(); in inferDynamicDimsForConv()106 int64_t inputRank = inputTy.getRank(); in inferDynamicDimsForConv()118 if (inputTy.isDynamicDim(heightDim)) { in inferDynamicDimsForConv()131 if (inputTy.isDynamicDim(weightDim)) { in inferDynamicDimsForConv()178 Type inputETy = inputTy.getElementType(); in matchAndRewrite()347 Type inputETy = inputTy.getElementType(); in matchAndRewrite()579 if (!inputTy.hasRank() || inputTy.isDynamicDim(0)) { in matchAndRewrite()693 Type resultETy = inputTy.getElementType(); in matchAndRewrite()[all …]
802 if (inputTy.isDynamicDim(i)) in reduceMatchAndRewriteHelper()1158 unsigned rank = inputTy.getRank(); in matchAndRewrite()1354 auto imageH = inputTy.getShape()[1]; in matchAndRewrite()1355 auto imageW = inputTy.getShape()[2]; in matchAndRewrite()1723 if (inputTy.isDynamicDim(i)) { in matchAndRewrite()1734 inputTy.getShape(), inputTy.getElementType()) in matchAndRewrite()1780 auto inputShape = inputTy.getShape(); in matchAndRewrite()1782 auto elementTy = inputTy.getElementType(); in matchAndRewrite()1783 int64_t rank = inputTy.getRank(); in matchAndRewrite()1845 int64_t rank = inputTy.getRank(); in matchAndRewrite()[all …]
93 ShapedType inputTy = input.getType().cast<ShapedType>(); in matchAndRewrite() local110 if (!inputTy.hasStaticShape() || !weightTy.hasStaticShape() || in matchAndRewrite()157 ShapedType inputTy = input.getType().cast<ShapedType>(); in matchAndRewrite() local162 Type inputETy = inputTy.getElementType(); in matchAndRewrite()181 if (!inputTy.hasStaticShape() || !weightTy.hasStaticShape() || in matchAndRewrite()185 int64_t batch = inputTy.getDimSize(0); in matchAndRewrite()
133 # Unmerge(merge) with incompatible types: unmerge destTy != merge inputTy.159 # Unmerge(concatvectors) with incompatible types: unmerge destTy != merge inputTy160 # but destTy.size() == inputTy.size().