| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 194031845
|
|
|
|
|
|
|
|
| |
library
Closes #12102.
PiperOrigin-RevId: 193696537
|
|
|
|
| |
PiperOrigin-RevId: 193119953
|
|
|
|
| |
PiperOrigin-RevId: 192821482
|
|
|
|
| |
PiperOrigin-RevId: 191840652
|
|
|
|
|
|
|
|
| |
use. The default is still NCCL 1 from GitHub. If the user chooses NCCL 2, it asks for the install directory.
The nccl_configure.bzl generates two different BUILD files based on the chose NCCL version. For NCCL 1, it aliases to the existing 'nccl_archive' http_repo on GitHub. For NCCL 2, it creates a target containing the NCCL 2 library and headers from the chosen install directory.
PiperOrigin-RevId: 191718007
|
|
|
|
|
|
|
| |
They added big endian preprocessor macros in recent releases. Hopefully
this should do the right thing on IBM mainframes.
PiperOrigin-RevId: 191705207
|
|
|
|
| |
PiperOrigin-RevId: 191696213
|
|
|
|
|
|
| |
DataTypes.h is no longer a generated header. X86DisassemblerDecoderCommon.h is now part of :support.
PiperOrigin-RevId: 191438031
|
|
|
|
| |
PiperOrigin-RevId: 190953197
|
|
|
|
| |
PiperOrigin-RevId: 190878279
|
|
|
|
| |
PiperOrigin-RevId: 190858242
|
|
|
|
| |
PiperOrigin-RevId: 190835392
|
|
|
|
|
|
|
|
|
|
| |
Makes the tf.keras.Layer.__call__ signature identical to tf.layers.Layer.__call__, but makes passing positional arguments other than "inputs" an error in most cases. The only case it's allowed is subclassed Models which do not have an "inputs" argument to their call() method.
This means subclassed Models no longer need to pass all but the first argument as a keyword argument (or do list packing/unpacking) when call() takes multiple Tensor arguments.
Includes errors for cases where whether an argument indicates an input is ambiguous, but otherwise doesn't do much to support non-"inputs" call() signatures for shape inference or deferred Tensors. The definition of an input/non-input is pretty clear, so that cleanup will mostly be tracking down all of the users of "self.call" and getting them to pass inputs as positional arguments if necessary.
PiperOrigin-RevId: 190787899
|
|
|
|
|
|
|
|
| |
checkpointing.
Uses a more recursive/functional tracking style which avoids numbering layers. Maybe this is too magical and we should adapt tf.keras.Sequential first? Let me know what you think.
PiperOrigin-RevId: 190282346
|
|
|
|
| |
PiperOrigin-RevId: 190161440
|
|
|
|
|
|
|
|
|
|
|
| |
Previously we only allowed to download clang when doing GPU builds.
The added skylark files use bazel's autoconf scripts, which were only added in
0.10.0. To provide nice error message for older versions of bazel
(i.e. 'version is less than 0.10' vs 'can't load @bazel_tools/cpp/...'), we
move the bazel version check into WORKSPACE file from workspace.bzl.
PiperOrigin-RevId: 190050798
|
|
|
|
| |
PiperOrigin-RevId: 189945839
|
|
|
|
|
|
|
|
|
|
|
| |
By disabling openmp when building with clang.
If we want to enable openmp with clang, we'll probably have to have libomp as
an explicit dependency.
This fixes a breakage found by OS CI:
https://ci.tensorflow.org/view/Nightly/job/nightly-matrix-linux-gpu-clang/215/
PiperOrigin-RevId: 189644968
|
|
|
|
| |
PiperOrigin-RevId: 189258641
|
|
|
|
| |
PiperOrigin-RevId: 189231636
|
|
|
|
|
|
| |
corresponding script.
PiperOrigin-RevId: 188324090
|
|
|
|
|
|
| |
This detects negativo17's CUDA packages for Fedora.
PiperOrigin-RevId: 187923472
|
|
|
|
| |
PiperOrigin-RevId: 187643585
|
|
|
|
| |
PiperOrigin-RevId: 187191730
|
|
|
|
|
|
| |
* updated ShuffleIndex because of API change
PiperOrigin-RevId: 186822637
|
|
|
|
| |
PiperOrigin-RevId: 186674197
|
|
|
|
| |
PiperOrigin-RevId: 186592198
|
|
|
|
| |
PiperOrigin-RevId: 186073337
|
|
|
|
|
|
| |
in inference output.
PiperOrigin-RevId: 185939805
|
|
|
|
|
|
| |
* Also de-flake a test.
PiperOrigin-RevId: 185637742
|
|
|
|
|
|
| |
an Orc API change.
PiperOrigin-RevId: 185185088
|
|
|
|
|
|
|
| |
The same patch was sent via PR to cub upstream:
https://github.com/NVlabs/cub/pull/125
PiperOrigin-RevId: 184975304
|
|
|
|
| |
PiperOrigin-RevId: 184897758
|
|
|
|
| |
PiperOrigin-RevId: 183728562
|
|
|
|
|
|
| |
be necessary any longer.
PiperOrigin-RevId: 183474194
|
|
|
|
| |
PiperOrigin-RevId: 183347199
|
|
|
|
|
|
| |
They don't make sense in the open source repository.
PiperOrigin-RevId: 183140889
|
|
|
|
| |
PiperOrigin-RevId: 183100142
|
|
|
|
| |
PiperOrigin-RevId: 182829120
|
|
|
|
| |
PiperOrigin-RevId: 182511847
|
|
|
|
| |
PiperOrigin-RevId: 182258809
|
|
|
|
| |
PiperOrigin-RevId: 182247687
|
|
|
|
|
|
|
| |
This is necessary to compile with Clang. Since the TF GCP code doesn't have any
try/catch statements, this could be a bug fix.
PiperOrigin-RevId: 182143536
|
|
|
|
|
|
|
|
|
| |
This makes it 400kB smaller. This is worth it because even
-O6 would only make SQLite 3% faster.
https://www.sqlite.org/footprint.html
PiperOrigin-RevId: 182143488
|
|
|
|
| |
PiperOrigin-RevId: 181494416
|
|
|
|
| |
PiperOrigin-RevId: 181469026
|
|
|
|
|
|
|
|
| |
It matches the way bazel's autoconf works and seems to be the right thing to
do.
This change should fix #14380.
PiperOrigin-RevId: 181305871
|
|
|
|
|
|
| |
counterparts instead of removing them. Re-enable a few tests to verify.
PiperOrigin-RevId: 181035075
|
|
|
|
| |
PiperOrigin-RevId: 180746153
|