Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_shape_computations.cc155 Value dimOp = rewriter.create<tensor::DimOp>(loc, rewriter.getIndexType(), in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dsame_shape_propagation.cc292 auto dimOp = val.value().getDefiningOp<memref::DimOp>(); in tryEvaluateShapeToRoot() local