Searched refs:BroadcastConstant (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1453 BroadcastOp BroadcastConstant(Location loc, Shape shape, int32_t constant, in BroadcastConstant() function in mlir::mhlo::__anon5ef947ee0111::ConvertMatrixDiagPartV3Op
1536 Value b_zero = BroadcastConstant(loc, indices_shape, 0, 32, rewriter); in matchAndRewrite()
1537 Value b_false = BroadcastConstant(loc, indices_shape, 0, 1, rewriter); in matchAndRewrite()
1538 Value b_true = BroadcastConstant(loc, indices_shape, 1, 1, rewriter); in matchAndRewrite()
1539 Value b_k1 = BroadcastConstant(loc, indices_shape, k[1], 32, rewriter); in matchAndRewrite()
1540 Value b_rows = BroadcastConstant(loc, indices_shape, rows, 32, rewriter); in matchAndRewrite()
1541 Value b_cols = BroadcastConstant(loc, indices_shape, cols, 32, rewriter); in matchAndRewrite()
1543 BroadcastConstant(loc, indices_shape, max_diag_len, 32, rewriter); in matchAndRewrite()
|