and use it for Dense layers and 1x1 convolutions
and use it for Dense layers and 1x1 convolutions