| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 216536298
|
|
|
|
|
|
| |
Sometimes the actual number of outputs is dictated by one of the attributes of the NodeDef.
PiperOrigin-RevId: 216530696
|
|
|
|
| |
PiperOrigin-RevId: 216475683
|
|
|
|
| |
PiperOrigin-RevId: 216471178
|
|
|
|
|
|
| |
Support peephole and num_proj as well.
PiperOrigin-RevId: 216467578
|
|
|
|
| |
PiperOrigin-RevId: 216463491
|
|
|
|
| |
PiperOrigin-RevId: 216463443
|
|
|
|
|
|
| |
values.
PiperOrigin-RevId: 216461637
|
|
|
|
| |
PiperOrigin-RevId: 216455772
|
|
|
|
| |
PiperOrigin-RevId: 216452447
|
|
|
|
| |
PiperOrigin-RevId: 216451263
|
|
|
|
| |
PiperOrigin-RevId: 216447412
|
|
|
|
| |
PiperOrigin-RevId: 216445964
|
|
|
|
| |
PiperOrigin-RevId: 216443201
|
|
|
|
| |
PiperOrigin-RevId: 216442906
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
The core of the change is have the gradient tape capture
distributed variables instead of plain ResourceVariables.
In other words, we move the distribution awareness from defun
down to tape and rely on distributed variable magic to provide us
with the right variable at runtime.
In tower context, we always watch the container (e.g. MirroredVariable).
In cross tower context, we always watch all the components.
PiperOrigin-RevId: 216430530
|
|
|
|
|
|
| |
attributes, set the attributes of all the contained variables. This fixes a bug that tf.train.init_from_checkpoint doesn't overwrite the initialization values correctly for TPUMirroredVariable.
PiperOrigin-RevId: 216429476
|
|
|
|
| |
PiperOrigin-RevId: 216425002
|
|
|
|
| |
PiperOrigin-RevId: 216422334
|
|
|
|
| |
PiperOrigin-RevId: 216419983
|
|
|
|
| |
PiperOrigin-RevId: 216419037
|
|
|
|
| |
PiperOrigin-RevId: 216418324
|
|
|
|
|
|
|
|
|
|
|
|
| |
function calls.
E.g. register_kl calls would trigger such warnings. This spam was exacerbated
by the fact that it happens before logging is initialized, so it is dumped
prominently to STDERR. Worse yet it also happened no matter whether the user
imported any symbols from tf.distributions or not as the relevant code is
executed when you import TensorFlow.
PiperOrigin-RevId: 216396036
|
|
|
|
| |
PiperOrigin-RevId: 216392908
|
|
|
|
| |
PiperOrigin-RevId: 216386450
|
|
|
|
| |
PiperOrigin-RevId: 216385202
|
|
|
|
| |
PiperOrigin-RevId: 216370193
|
|
|
|
| |
PiperOrigin-RevId: 216367867
|
|
|
|
|
|
|
| |
If the graph contains not constant array with strings it fails because the
array's size can't be estimated.
PiperOrigin-RevId: 216356162
|
|
|
|
| |
PiperOrigin-RevId: 216303340
|
|
|
|
| |
PiperOrigin-RevId: 216270385
|
|
|
|
| |
PiperOrigin-RevId: 216260437
|
|
|
|
|
|
|
| |
Use the ArgDef::type field when available for propagating
the output types from a given unsupported operator.
PiperOrigin-RevId: 216257741
|
|
|
|
| |
PiperOrigin-RevId: 216248418
|
|\
| |
| |
| | |
PiperOrigin-RevId: 216245934
|
| |
| |
| |
| |
| |
| | |
sqrt(v + epsilon**2) and changed flag name accordingly.
PiperOrigin-RevId: 216240045
|
| |
| |
| |
| |
| |
| | |
must be on axis 0, and can only be on 1 or 2-d inputs.
PiperOrigin-RevId: 216226776
|
| |
| |
| |
| | |
PiperOrigin-RevId: 216225505
|
| |
| |
| |
| | |
PiperOrigin-RevId: 216210141
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| | |
Prior to this change, tf.colocate_with(v) would insert spurious operations (a ReadVariableOp and an Identity) in the graph when v is a resource variable, and then
colocate the operations within the block with those newly added, otherwise disconnected, operations.
This commit avoids adding the unnecessary ReadVariableOp/Identity nodes and colocates
operations within the block with the VarHandleOp.
PiperOrigin-RevId: 216201638
|
| |
| |
| |
| | |
PiperOrigin-RevId: 216200439
|
| |
| |
| |
| | |
PiperOrigin-RevId: 216191084
|
|\ \
| | |
| | |
| | | |
PiperOrigin-RevId: 216185979
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 216066634
|
| | |
| | |
| | |
| | |
| | |
| | | |
It fails 1/1000 runs in OSS builds.
PiperOrigin-RevId: 216050192
|
|\ \ \
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 216046506
|
|\ \ \ \
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 216040541
|
|\ \ \ \ \
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 215995215
|
|\ \ \ \ \ \
| | | | | | |
| | | | | | |
| | | | | | | |
PiperOrigin-RevId: 215981773
|
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | | |
PiperOrigin-RevId: 215978771
|