Add sample weight support; pass `training=True` in SAM's `train_step` to support layers like BatchNorm and Dropout.#858

copybara-service[bot] merged 1 commit intomasterkeras-team/tf-keras:masterfrom

test_799861878keras-team/tf-keras:test_799861878Copy head branch name to clipboard