Removes host_from_gpu and gpu_from_host around specify_shapes node,
making it run transperently on the GPU