Align mask formatting of both masks more closely (#96286)
Summary: Align mask formatting of both masks more closely
Test Plan: sandcastle & github
Differential Revision: D43878634
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96286
Approved by: https://github.com/cpuhrsch
