Added output_dtype to PoolingLayer.