Update build script for tf-random