Searched defs:collapseShape (Results 1 – 2 of 2) sorted by relevance
454 getCollapsedSize(memref::CollapseShapeOp collapseShape, OpBuilder &builder, in getCollapsedSize() argument 495 getCollapsedStride(memref::CollapseShapeOp collapseShape, OpBuilder &builder, in getCollapsedStride() argument
1473 llvm::SmallVector<int64_t> collapseShape{batch}; matchAndRewrite() local