Revert the changes in test_eltwise and vision_layer