Home
last modified time | relevance | path

Searched defs:ShapeReificationPattern (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dshape_reification_pass.cc35 struct ShapeReificationPattern : public OpRewritePattern<shape::ShapeOfOp> { struct
36 explicit ShapeReificationPattern(MLIRContext *ctx) in ShapeReificationPattern() function
42 LogicalResult matchAndRewrite(shape::ShapeOfOp op, in matchAndRewrite()
H A Dmerge_assuming_ops.cc45 struct ShapeReificationPattern : public OpRewritePattern<shape::ShapeOfOp> { struct
46 explicit ShapeReificationPattern(MLIRContext *context) in ShapeReificationPattern() function
52 LogicalResult matchAndRewrite(shape::ShapeOfOp op, in matchAndRewrite()