supporting both CPU and GPU modes in im2col_speed_test
removed old TODOs