op.getBiasMutable().assign(none_value);
      return success();
    }

    return failure();
  }
};

// Return true if the given Add operation has the CPU kernel supported shapes.
bool VerifyAddOpShapeConstraints(AddOp op) {
  auto element_type = getElementTypeOrSelf(op.getOutput().getType());

  // Allows F32, QI8, QUI8 and I32 outputs when the operands have valid shapes,

}

def TFL_AddOp : TFL_Op<"add", [
    TFL_RuntimePredOpTrait<"Operands should have valid shapes and element type needs to match",
      And<[CPred<"TFL::VerifyAddOpShapeConstraints(llvm::cast<AddOp>($_op))">,
           OperandsSameElementTypeConstraintBasePred]>>,
    ResultsBroadcastableShape,
    Pure,
    Commutative,
    QuantizableResult,