* Fix more int overflow during quant. * Fix some more int overflow in softmax. * Revert back to int64_t.