Searched defs:ConvertUnrankedDynamicBroadcastBinaryOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 214 struct ConvertUnrankedDynamicBroadcastBinaryOp struct
218 LogicalResult matchAndRewrite( in matchAndRewrite()
310 Value IsSingleElementShape(OpBuilder &rewriter, ChloOpTy op, in IsSingleElementShape()
321 Value GreaterRankIsN(OpBuilder &builder, Location loc, Value actual_rank, in GreaterRankIsN()
328 scf::IfOp createIfOpForRankSpecializedBroadcastAndOp( in createIfOpForRankSpecializedBroadcastAndOp()
338 Value extendToBroadcastShape(OpBuilder &builder, Location loc, Value value, in extendToBroadcastShape()
349 Value createBroadcastToKnownRank(OpBuilder &builder, ChloOpTy op, Value value, in createBroadcastToKnownRank()
370 void createRankSpecializedBroadcastAndOp(OpBuilder &if_builder, ChloOpTy op, in createRankSpecializedBroadcastAndOp()
410 Value HandleBroadcastAndOp(OpBuilder &rewriter, ChloOpTy op, Value lhs, in HandleBroadcastAndOp()
|