// CHECK-SAME: value = dense<1> : tensor<i32>

// Check that the constants that are input to XlaConvV2's 3rd and 4th
// arguments are not duplicated.
// CHECK-NOT: "tf.Const"()

// Check that the constants used for tf.XlaConvV2 and tf.AddV2s are different.

def IsEinsumOpSupported : Constraint<
  CPred<"IsEinsumOpSupported($0, $1, $2)">,
  "Check if the given einsum op could be converted into a XlaDotV2 op.">;

// Converts inlined Conv2D pattern to TF XlaConvV2 op. This pattern doesn't
// support non-constant weights.
def ConvertTFConv2DToXLAConvOp : Pat<
  (TF_Conv2DOp:$conv
    (TF_SubOp (TF_CastOp $input, $truncate), $input_zp),

    new absl::flat_hash_set<std::string>{"XlaBroadcastHelper",
                                         "XlaCallModule",
                                         "XlaConv",
                                         "XlaConvV2",
                                         "XlaDequantize",
                                         "XlaDot",
                                         "XlaDotV2",
                                         "XlaDynamicSlice",