Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dshape_simplification.cc182 auto shapeOfOp = shape.getDefiningOp<ShapeOfOp>(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Drank_specialization.cc383 if (auto shapeOfOp = llvm::dyn_cast<shape::ShapeOfOp>(it)) { in materializeFinalReshape() local
816 if (auto shapeOfOp = in findNonScalarShapeEquivalences() local