| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 191805453
|
|
|
|
|
|
| |
Mul reductions want a 1 as their init value, not a 0 or a random value.
PiperOrigin-RevId: 191802819
|
|
|
|
| |
PiperOrigin-RevId: 191797853
|
|
|
|
| |
PiperOrigin-RevId: 191794956
|
|
|
|
|
|
| |
and binary_classification_head.
PiperOrigin-RevId: 191793392
|
|
|
|
| |
PiperOrigin-RevId: 191788253
|
|
|
|
| |
PiperOrigin-RevId: 191786563
|
|
|
|
| |
PiperOrigin-RevId: 191784709
|
|
|
|
| |
PiperOrigin-RevId: 191784430
|
|
|
|
|
|
| |
broadcast gradient shape nodes.
PiperOrigin-RevId: 191779263
|
|
|
|
|
|
|
|
| |
already.
Fixes #17423
PiperOrigin-RevId: 191773001
|
|
|
|
| |
PiperOrigin-RevId: 191771969
|
|
|
|
| |
PiperOrigin-RevId: 191769890
|
|
|
|
| |
PiperOrigin-RevId: 191769724
|
|
|
|
| |
PiperOrigin-RevId: 191763101
|
|
|
|
| |
PiperOrigin-RevId: 191761109
|
|
|
|
| |
PiperOrigin-RevId: 191756234
|
|
|
|
| |
PiperOrigin-RevId: 191756100
|
|
|
|
| |
PiperOrigin-RevId: 191754120
|
|
|
|
| |
PiperOrigin-RevId: 191753693
|
|
|
|
|
|
| |
checkpoint before the first training step.
PiperOrigin-RevId: 191753026
|
|
|
|
| |
PiperOrigin-RevId: 191751175
|
|
|
|
|
|
|
|
| |
Using fill saves on GraphDef size, but can slow down models since the
total number of ops is greater (fill + shape + constant op). This
change makes us only use fill for large shapes.
PiperOrigin-RevId: 191747456
|
|
|
|
| |
PiperOrigin-RevId: 191747417
|
|
|
|
|
|
| |
that we override: control flow, builtins, operators, etc.
PiperOrigin-RevId: 191729654
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
| |
PiperOrigin-RevId: 191696203
|
|
|
|
| |
PiperOrigin-RevId: 191679495
|
|
|
|
| |
PiperOrigin-RevId: 191677358
|
|
|
|
| |
PiperOrigin-RevId: 191673876
|
|
|
|
| |
PiperOrigin-RevId: 191672761
|
|
|
|
|
|
|
| |
Initialize the buffer for a scalar tensor to avoid uninitialized
accesses.
PiperOrigin-RevId: 191672257
|
|
|
|
|
|
| |
- Part 2
PiperOrigin-RevId: 191669725
|
|
|
|
| |
PiperOrigin-RevId: 191667378
|
|
|
|
| |
PiperOrigin-RevId: 191666300
|
|
|
|
| |
PiperOrigin-RevId: 191665098
|
|
|
|
|
|
| |
in fastbuild mode.
PiperOrigin-RevId: 191665014
|
|
|
|
|
|
| |
Don't propagate constants with control dependencies through Merge nodes.
PiperOrigin-RevId: 191663396
|
|
|
|
| |
PiperOrigin-RevId: 191660588
|
|
|
|
|
|
| |
/arch:AVX is the correpsonding option in MSVC for gcc's --march=native
PiperOrigin-RevId: 191660389
|
|
|
|
|
|
|
|
| |
optimize_for_inference script only use TOCO friendly ops. In particular, FusedResizeAndPadConv2D is not supported by TOCO.
This change does not alter existing behavior (the boolean is set to false by default).
PiperOrigin-RevId: 191660378
|
|
|
|
| |
PiperOrigin-RevId: 191659944
|
|
|
|
| |
PiperOrigin-RevId: 191659091
|
|
|
|
|
|
| |
The range of valid values for the `axis` argument should be `[-rank(input), rank(input))`, just like other reduction ops.
PiperOrigin-RevId: 191657767
|
|
|
|
|
|
| |
Otherwise, we run into a "dictionary changed size during iteration" once in a while, as we are modifying the values in the dictionary during the iteration.
PiperOrigin-RevId: 191655599
|
|
|
|
|
|
|
|
| |
model.
Had to modify the __call__ method in the base layer class so that it could work with feature style inputs in which case we lazily convert the inputs to tensors instead of providing tensors as inputs upfront.
PiperOrigin-RevId: 191655445
|
|
|
|
|
|
| |
clarify that its purpose is only for training.
PiperOrigin-RevId: 191654161
|
|
|
|
| |
PiperOrigin-RevId: 191652174
|