| Commit message (Collapse) | Author | Age |
... | |
|
|
|
| |
PiperOrigin-RevId: 192637289
|
|
|
|
|
|
| |
dtype.
PiperOrigin-RevId: 192634133
|
|
|
|
|
|
|
| |
Also modify code to ensure aliased forwarding happens whenever
possible with DT_VARIANT objects in ResourceVariables and in the new op.
PiperOrigin-RevId: 192632202
|
|
|
|
|
|
|
|
|
| |
random data and outfeeding the data generated at each step.
The arguments of the computation can be either read from the session module, or randomly generated.
The tool uses the raw transfer manager API to infeed and outfeed the data.
PiperOrigin-RevId: 192628605
|
|
|
|
|
|
|
|
|
| |
BaseCollectiveExecutor and RingReducer.
This change is part of a series of changes introducing infrastructure
for collective ops and initial implementations of reduction and broadcast.
PiperOrigin-RevId: 192624521
|
|
|
|
| |
PiperOrigin-RevId: 192624191
|
|
|
|
|
|
| |
original and optimized graphs and check if the outputs are same.
PiperOrigin-RevId: 192616402
|
|
|
|
|
|
| |
top-level name scopes to begin with underscores. Also use the transformer scope information to get to the enclosing function name.
PiperOrigin-RevId: 192600256
|
|
|
|
| |
PiperOrigin-RevId: 192591402
|
|
|
|
| |
PiperOrigin-RevId: 192560111
|
|
|
|
| |
PiperOrigin-RevId: 192550101
|
|
|
|
| |
PiperOrigin-RevId: 192549109
|
|
|
|
| |
PiperOrigin-RevId: 192548652
|
|
|
|
| |
PiperOrigin-RevId: 192548367
|
|
|
|
| |
PiperOrigin-RevId: 192546579
|
|
|
|
|
|
| |
CreateFromProto requires that the config has proper entry_computation_layout, so give the config the program shape.
PiperOrigin-RevId: 192546316
|
|
|
|
|
|
| |
See:
https://source.cloud.google.com/results/invocations/31632a30-3728-4635-a456-f89b9e8b9dfe/log
PiperOrigin-RevId: 192544848
|
|
|
|
|
|
| |
Also set the sharding to the instruction when created from proto.
PiperOrigin-RevId: 192543024
|
|
|
|
| |
PiperOrigin-RevId: 192542956
|
|
|
|
|
|
|
|
| |
variables(which assumes there exists a session). Otherwise create a session so that we can run get_config().
Actually fix #18193.
PiperOrigin-RevId: 192541442
|
|
|
|
|
|
| |
This is a starting point for Dataset-compatible weighted sampling across a list of datasets.
PiperOrigin-RevId: 192540412
|
|
|
|
| |
PiperOrigin-RevId: 192537874
|
|
|
|
| |
PiperOrigin-RevId: 192536085
|
|
|
|
| |
PiperOrigin-RevId: 192534931
|
|
|
|
|
|
| |
modes too.
PiperOrigin-RevId: 192532297
|
|
|
|
| |
PiperOrigin-RevId: 192530231
|
|
|
|
|
|
| |
Implement a vectorized way to compute the same thing instead.
PiperOrigin-RevId: 192524667
|
|
|
|
|
|
| |
data are saved with host prefix.
PiperOrigin-RevId: 192523668
|
|
|
|
|
|
| |
estimators.
PiperOrigin-RevId: 192521398
|
|
|
|
| |
PiperOrigin-RevId: 192520351
|
|
|
|
| |
PiperOrigin-RevId: 192519639
|
|
|
|
|
|
| |
the metagraph.
PiperOrigin-RevId: 192518307
|
|
|
|
|
|
| |
updated checkpoints from the contents.
PiperOrigin-RevId: 192517819
|
|
|
|
| |
PiperOrigin-RevId: 192516190
|
|
|
|
| |
PiperOrigin-RevId: 192511036
|
|
|
|
|
|
|
|
| |
mutating
NodeDef before creating the copied operation.
PiperOrigin-RevId: 192505209
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
`log_det_jacobian` methods to take event_ndims.
The class level event_ndims parameter is being deprecated in favor of passing it in
to the `log_det_jacobian` methods.
Specific changes:
- `log_det_jacobian` signatures are now `log_det_jacobian(input, event_ndims)`
- Constructors no long have event_ndims passed in (e.g. Affine() vs. Affine(event_ndims=0)).
- All bijectors must specify a subset of [forward_min_event_ndims, inverse_min_event_ndims]. This is the minimal dimensionality the bijector operates on, with it being "broadcasted" to any passed in event_ndims (e.g. Exp has forward_min_event_ndims = 0. That means it operates on scalars. However, we can use the bijector on any event_ndims > 0 (i.e. we've broadcasted the transformation to work on any amount of event_ndims > 0), and jacobian reduction will work in those cases.
As a result of this change, all bijectors should "broadcast" (e.g. Sigmoid now works on any number of event_ndims).
Other changes (internal and documentation):
- Added clarifications on Jacobian Determinant vs. Jacobian Matrix.
- Added clarifications on min_event_ndims, and what the jacobian reduction is over.
- Changed caching of ildj to be keyed on event_ndims.
- Several bug fixes to bugs unearthed while writing this code (e.g. transformed distribution shape computation being incorrect)
PiperOrigin-RevId: 192504919
|
|
|
|
| |
PiperOrigin-RevId: 192504411
|
|
|
|
| |
PiperOrigin-RevId: 192503204
|
|
|
|
|
|
|
| |
https://github.com/tensorflow/hub/tree/master/examples/image_retraining
has the same tool, upgraded to use TensorFlow Hub instead of raw graph defs.
PiperOrigin-RevId: 192502469
|
|
|
|
|
|
| |
Choosing not to do this at static analysis because it exposes the scope to any node, making it easier to use by any specialization of a transformer.
PiperOrigin-RevId: 192502309
|
|
|
|
|
|
| |
(not yet supported by TPU compilation).
PiperOrigin-RevId: 192502020
|
|
|
|
| |
PiperOrigin-RevId: 192499401
|
|
|
|
| |
PiperOrigin-RevId: 192498471
|
|
|
|
| |
PiperOrigin-RevId: 192491335
|
|
|
|
| |
PiperOrigin-RevId: 192491201
|
|
|
|
|
|
| |
tensorflow/contrib/data/python/kernel_tests:batch_dataset_op_test to "medium".
PiperOrigin-RevId: 192484895
|
|
|
|
| |
PiperOrigin-RevId: 192482242
|
|
|
|
| |
PiperOrigin-RevId: 192481690
|
|
|
|
| |
PiperOrigin-RevId: 192480452
|