Make GpuSoftmax and GpuSoftmaxWithBias compatible with float64 and adjust unit tests to test this