Searched refs:shape_of_lhs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 233 Value shape_of_lhs = rewriter.create<shape::ShapeOfOp>(loc, lhs); in matchAndRewrite() local
238 loc, result_type, IsSingleElementShape(rewriter, op, shape_of_lhs), in matchAndRewrite()
249 shape_of_lhs, shape_of_rhs); in matchAndRewrite()
272 shape_of_lhs, shape_of_rhs); in matchAndRewrite()
282 loc, shape_of_lhs, shape_of_rhs); in matchAndRewrite()
339 Value shape_of_lhs, Value shape_of_rhs) const { in extendToBroadcastShape()
344 shape_of_lhs, shape_of_rhs, nullptr); in extendToBroadcastShape()
|