is_balanced = rho_x is None and rho_y is None
Source: torchmatch/transport/samples/kernels/cg_dense.py:479
