Searched defs:outputTy (Results 1 – 3 of 3) sorted by relevance
921 if (inputTy == outputTy && inputTy.getNumDynamicDims() < 2) in fold() local 517 auto outputTy = llvm::dyn_cast<RankedTensorType>(getType()); fold() local 817 auto outputTy = llvm::dyn_cast<RankedTensorType>(getType()); REDUCE_FOLDER() local [all...]
565 auto outputTy = cast<ShapedType>(op.getType()); matchAndRewrite() local 621 auto outputTy = cast<ShapedType>(op.getType()); matchAndRewrite() local [all...]
1140 auto outputTy = cast<ShapedType>(op.getOutput().getType()); matchAndRewrite() local