Searched refs:shape_of_rhs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 234 Value shape_of_rhs = rewriter.create<shape::ShapeOfOp>(loc, rhs); in matchAndRewrite() local
249 shape_of_lhs, shape_of_rhs); in matchAndRewrite()
260 IsSingleElementShape(else_lhs_scalar_builder, op, shape_of_rhs), true); 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()
|