Home
last modified time | relevance | path

Searched defs:assuming_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dbroadcast_propagation.cc108 auto assuming_op = in MoveIntoAssumingOpMatchAndRewrite() local
202 auto assuming_op = op->template getParentOfType<shape::AssumingOp>(); in matchAndRewrite() local
Dchlo_legalize_to_hlo.cc1254 auto assuming_op = rewriter.create<shape::AssumingOp>( in matchAndRewrite() local
1400 auto assuming_op = rewriter.create<shape::AssumingOp>( in matchAndRewrite() local
Drank_specialization.cc798 if (auto assuming_op = op->getParentOfType<shape::AssumingOp>()) { in FindNonScalarShapeEquivalences() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3206 auto assuming_op = in matchAndRewrite() local