modify fprop and test_layer_stats