Revert "Revert "Update tf and replace xla::int64 to xla::int64_t"" (#3175)
* Revert "Revert "Update tf and replace xla::int64 to xla::int64_t (#3140)" (#3174)"
This reverts commit f8310d3613c7fb1f0899aa795afe7daaccd1e93d.
* Only link cublas files if exist