| Commit message (Collapse) | Author | Age |
|
|
|
|
|
| |
compatibility.
PiperOrigin-RevId: 198893078
|
|
|
|
|
|
|
| |
When implementing a custom layer, it's necessary to call the Layer constructor
from the custom layer's constructor.
PiperOrigin-RevId: 198892503
|
|
|
|
|
|
| |
edges to a single node from the same source device. Instead, build an intermediate NoOp node on the source device and use a single cross-device control edge.
PiperOrigin-RevId: 198891614
|
|
|
|
| |
PiperOrigin-RevId: 198887868
|
|
|
|
| |
PiperOrigin-RevId: 198886485
|
|
|
|
| |
PiperOrigin-RevId: 198880096
|
|
|
|
| |
PiperOrigin-RevId: 198878259
|
|
|
|
| |
PiperOrigin-RevId: 198876135
|
|
|
|
|
|
| |
reduce_{sum,prod,etc}
PiperOrigin-RevId: 198874465
|
|
|
|
|
|
| |
include the cuDNN header file.
PiperOrigin-RevId: 198869605
|
|
|
|
| |
PiperOrigin-RevId: 198863091
|
|
|
|
|
|
| |
optonly due to flakiness.
PiperOrigin-RevId: 198862313
|
|
|
|
| |
PiperOrigin-RevId: 198859282
|
|
|
|
| |
PiperOrigin-RevId: 198850955
|
|
|
|
|
|
| |
For large constants, creating an llvm::Constant for each element can get prohibitively large compile times.
PiperOrigin-RevId: 198843141
|
|
|
|
| |
PiperOrigin-RevId: 198837256
|
|
|
|
| |
PiperOrigin-RevId: 198836479
|
|
|
|
| |
PiperOrigin-RevId: 198823349
|
|
|
|
|
|
| |
continuous build due to flakiness.
PiperOrigin-RevId: 198817129
|
|
|
|
|
|
|
|
| |
Allows fit() when call's signature looks something like call(x, training=True).
Calling conventions are "inputs", single positional, and multiple positional. Right now the distinction between "inputs" and single positional calling conventions is the text of one error message. Both support shape inference (which just hasn't been implemented for multiple positional input arguments yet).
PiperOrigin-RevId: 198815483
|
|
|
|
| |
PiperOrigin-RevId: 198815200
|
|
|
|
|
|
| |
flakiness.
PiperOrigin-RevId: 198813273
|
|
|
|
|
|
| |
correctness
PiperOrigin-RevId: 198812512
|
|
|
|
|
|
| |
TensorFlow Optimizers compiled with a Model will now have their state saved and restored with save_weights/load_weights.
PiperOrigin-RevId: 198811639
|
|
|
|
| |
PiperOrigin-RevId: 198810875
|
|
|
|
|
|
|
| |
containing a ShuffleDataset with reshuffle_each_iteration=True.
This is a temporary fix till we figure out how to handle this use-case.
PiperOrigin-RevId: 198805344
|
|
|
|
|
|
|
| |
CustomCall can have a window and convolution-dimension-numbers, so
HloInstruction needs to handle this in Clone() and Identical().
PiperOrigin-RevId: 198805211
|
|
|
|
| |
PiperOrigin-RevId: 198805143
|
|
|
|
|
|
| |
due to timeouts without optimization.
PiperOrigin-RevId: 198804880
|
|
|
|
|
|
|
|
| |
(such as tf.while_loop()).
Register the remaining control-flow operators on XLA devices.
PiperOrigin-RevId: 198803131
|
|
|
|
|
|
| |
simplifier reshape change is now covered by ReshapeMover.
PiperOrigin-RevId: 198802494
|
|
|
|
|
|
| |
This fixes an error when running resnet50_batch128_fp16 with --v=2.
PiperOrigin-RevId: 198798196
|
|
|
|
|
|
| |
removed references to content that is being deleted.
PiperOrigin-RevId: 198797662
|
|
|
|
| |
PiperOrigin-RevId: 198797254
|
|
|
|
|
|
| |
By marking DebugString() as const we can make some error messages more descriptive. Because DatasetIterator marks the return value of the dataset() function const, DebugString() cannot be called.
PiperOrigin-RevId: 198796894
|
|
|
|
| |
PiperOrigin-RevId: 198795738
|
|
|
|
| |
PiperOrigin-RevId: 198795625
|
|
|
|
|
|
|
| |
Gets rid of the tricky algorithm/use-tensor-cores operands to cudnn
convolution customcalls, using instead a backend_config.
PiperOrigin-RevId: 198794988
|
|
|
|
| |
PiperOrigin-RevId: 198794897
|
|
|
|
| |
PiperOrigin-RevId: 198794845
|
|
|
|
| |
PiperOrigin-RevId: 198793502
|
|
|
|
| |
PiperOrigin-RevId: 198793295
|
|
|
|
| |
PiperOrigin-RevId: 198789495
|
|
|
|
| |
PiperOrigin-RevId: 198788352
|
|
|
|
| |
PiperOrigin-RevId: 198787391
|
|
|
|
|
|
| |
in tf.map_fn to be compatible with XLA.
PiperOrigin-RevId: 198786266
|
|
|
|
| |
PiperOrigin-RevId: 198783364
|
|
|
|
| |
PiperOrigin-RevId: 198782348
|
|
|
|
| |
PiperOrigin-RevId: 198778945
|
|
|
|
| |
PiperOrigin-RevId: 198775276
|