only work with broadcasted layer operations

for consistency