Searched refs:createRankSpecializedBroadcastAndOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 370 void createRankSpecializedBroadcastAndOp(OpBuilder &if_builder, ChloOpTy op, in createRankSpecializedBroadcastAndOp() function
435 createRankSpecializedBroadcastAndOp(if_builder, op, lhs, rhs, 1); in HandleBroadcastAndOp()
445 createRankSpecializedBroadcastAndOp(if_builder, op, lhs, rhs, i); in HandleBroadcastAndOp()
458 createRankSpecializedBroadcastAndOp(else_builder, op, lhs, rhs, in HandleBroadcastAndOp()
|