Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for DirectBinaryPat (0.17 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td

    // Check that two values can be broadcasted together
    def AreBroadcastCompatible : Constraint<CPred<"AreBroadcastCompatible($0, $1)">,
        "types must be broadcastable">;
    
    class DirectBinaryPat<Op FromOp, Op ToOp>
      : Pat<(FromOp AnyTensor:$l, AnyTensor:$r),
            (ToOp $l, $r, (BinBroadcastDimensions $l, $r))>;
    
    foreach fromToBinPair = [[TF_AddV2Op, CHLO_BroadcastAddOp],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 34.8K bytes
    - Viewed (0)
Back to top