Searched refs:prunedInShape (Results 1 – 1 of 1) sorted by relevance
434 auto prunedInShape = inputType.getShape().drop_back(numPrunedDims); in matchAndRewrite() local435 SmallVector<int64_t, 4> ones(prunedInShape.size(), 1); in matchAndRewrite()436 auto prunedInStrides = computeStrides(prunedInShape, ones); in matchAndRewrite()444 int64_t numTransposedElements = ShapedType::getNumElements(prunedInShape); in matchAndRewrite()