"$0.getType().cast<ShapedType>().getShape() =="
          "$1.getType().cast<ShapedType>().getShape()">,
    "have the same static shape">;

def IsPermutationNCHW : Constraint<CPred<"IsPermutationNCHW($0)">>;

def IsBiasShape : Constraint<
    CPred<"$0.getType().cast<ShapedType>().getRank() == 4 && "
          "$0.getType().cast<ShapedType>().getShape()[2] == 1 && "

    if (!input_type) {
      return failure();
    }
    const ArrayRef<int64_t> input_shape = input_type.getShape();

    const int input_size = input_shape.size();

    RankedTensorType begin_type = mlir::cast<RankedTensorType>(begin.getType());
    const ArrayRef<int64_t> begin_shape = begin_type.getShape();
    const int begin_dim = begin_shape.size();

    if (begin_dim != 1) return failure();

  Type dtype = getElementTypeOrSelf(type);
  if (RankedTensorType ty = type.dyn_cast<RankedTensorType>()) {
    llvm::SmallVector<int64_t, 4> shape = {dim};
    shape.append(ty.getShape().begin(), ty.getShape().end());
    return tensorflow::GetTypeFromTFTensorShape(shape, dtype);
  }
  return type;
}

Type GetTensorTypeForTensorList(Type element_type, TF::VariantType handle_dtype,