Fixed type in ThresholdLayer.cu