Skip to content

Modify custom BatchNorm layer such that the mean and variance variables have the same rank whether train=False or train=True.#838

Merged
copybara-service[bot] merged 1 commit intomasterfrom
test_862475302
Jan 30, 2026
Merged

Modify custom BatchNorm layer such that the mean and variance variables have the same rank whether train=False or train=True.#838
copybara-service[bot] merged 1 commit intomasterfrom
test_862475302

Commits