1 2 3def _conv1d_flop_count(layer, rate): 4 return 2 * ((layer.in_channels + 1) * layer.out_channels * rate / layer.stride[0] ) * layer.kernel_size[0] 5 6 7def _dense_flop_count(layer, rate): 8 return 2 * ((layer.in_features + 1) * layer.out_features * rate )