Searched defs:shape_of_op (Results 1 – 2 of 2) sorted by relevance
146 auto shape_of_op = op.tensor().getDefiningOp<ShapeOfOp>(); in matchAndRewrite() local195 auto shape_of_op = shape.getDefiningOp<ShapeOfOp>(); in matchAndRewrite() local
377 if (auto shape_of_op = llvm::dyn_cast<shape::ShapeOfOp>(it)) { in MaterializeFinalReshape() local812 if (auto shape_of_op = in FindNonScalarShapeEquivalences() local