Searched refs:IsSingleElementShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 238 loc, result_type, IsSingleElementShape(rewriter, op, shape_of_lhs), in matchAndRewrite()
260 IsSingleElementShape(else_lhs_scalar_builder, op, shape_of_rhs), true); in matchAndRewrite()
310 Value IsSingleElementShape(OpBuilder &rewriter, ChloOpTy op, in IsSingleElementShape() function
|