| Commit message (Collapse) | Author | Age |
| |
|
|
|
|
| |
PiperOrigin-RevId: 209908710
|
|
|
|
|
|
| |
whenever one of the template arguments is not a POD, in particular for types such as string. A memset on a struct that contains a string invalidates the internals of that class (in particular the pointer that stores the characters of the string) and causes segfaults. This allows using weighted quantiles beyond int/float like template parameters. The upstream issue with memcopy of alignment bytes used for padding has been solved separately in the meantime.
PiperOrigin-RevId: 209906298
|
|
|
|
|
|
| |
DistributionStrategy.
PiperOrigin-RevId: 209891724
|
|
|
|
| |
PiperOrigin-RevId: 209884688
|
|
|
|
|
|
| |
This allows MirroredStrategy with one device and between-graph strategies with one tower on each worker.
PiperOrigin-RevId: 209884569
|
|
|
|
|
|
| |
coordinator in CollectiveAllReduceStrategy.
PiperOrigin-RevId: 209879722
|
|
|
|
| |
PiperOrigin-RevId: 209875487
|
|
|
|
| |
PiperOrigin-RevId: 209873671
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
- Adds an accuracy tool that can be used to develop evaluation pipelines to
evaluate model accuracies.
- The binary can be compiled for mobile platforms and
the tool can be used to evaluate accuracy of models by running the binary on
device.
- Adds an example implementation for imagenet ILSVRC classification task
accuracy evaluation.
- More documentations and details coming soon.
PiperOrigin-RevId: 209869774
|
|
|
|
|
|
|
|
| |
instructions.
This will make adding collective instructions slightly easier.
PiperOrigin-RevId: 209864869
|
|
|
|
| |
PiperOrigin-RevId: 209864694
|
|
|
|
| |
PiperOrigin-RevId: 209864144
|
|
|
|
| |
PiperOrigin-RevId: 209860270
|
|
|
|
| |
PiperOrigin-RevId: 209853753
|
|
|
|
|
|
| |
deterministic.
PiperOrigin-RevId: 209852547
|
|\
| |
| |
| | |
PiperOrigin-RevId: 209850481
|
| |
| |
| |
| |
| |
| | |
Apparently it's not inherited from its py_binary dep for pip tests
PiperOrigin-RevId: 209848894
|
| |
| |
| |
| |
| |
| | |
coordinator in MirroredStrategy.
PiperOrigin-RevId: 209848375
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209847371
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209847352
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209845613
|
| |
| |
| |
| |
| |
| |
| |
| |
| | |
input datasets are of type dictionaries with different set of key(s).
FIXES #20626
REL_NOTES: bug fix in `tf.data.Dataset.concatenate()`, now throws error if trying to concatenate two datasets of type dictionaries with different set of key(s).
PiperOrigin-RevId: 209845337
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209845109
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209842403
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209840812
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209839159
|
| |
| |
| |
| |
| |
| |
| |
| | |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209839057
|
| |
| |
| |
| |
| |
| |
| |
| | |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209839048
|
| |
| |
| |
| |
| |
| |
| |
| | |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209839032
|
| |
| |
| |
| |
| |
| |
| |
| | |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209839030
|
| |
| |
| |
| |
| |
| |
| |
| |
| | |
In the recent change to instantiate functions at iterator
initialization time, I forgot to make the corresponding change was to
MultiDeviceIterator. This change fixes that problem and unbreaks
MultiDeviceIterator.
PiperOrigin-RevId: 209838406
|
| |
| |
| |
| |
| |
| |
| |
| | |
context.executing_eagerly() in optimizer.variables()
Remove global collection usage from the Keras model to estimator flow.
PiperOrigin-RevId: 209837803
|
| |
| |
| |
| |
| |
| |
| |
| | |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209837298
|
| |
| |
| |
| |
| |
| |
| |
| | |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209836477
|
| |
| |
| |
| |
| |
| | |
which will replace dataset.map(parsing_ops.parse_example(..)).
PiperOrigin-RevId: 209836033
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209835865
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209833197
|
| |
| |
| |
| |
| |
| | |
ops when we have nodes with no examples.
PiperOrigin-RevId: 209830550
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209830234
|
| |
| |
| |
| |
| |
| | |
tensor is dynamic shaped.
PiperOrigin-RevId: 209829459
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209828735
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209828610
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209828317
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| | |
to FuncGraph.
This change makes tfe.defun() responsible for respecting call-site device
annotations, not PartitionedCallOp --- the latter doesn't have access to
the original requested device, and it was overly-aggressive in stamping its
assigned device onto unassigned function nodes.
It's not obvious how we could make PartitionedCallOp faithfully respect call-site device annotations, since directives like tf.device(None) are lost by the time control reaches PartitionedCallOp. As such, this change GraphModeFunction.__call__() raise an error if it's called under a device context / stack that's different than the one that was active when the python function was traced.
PiperOrigin-RevId: 209827111
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209824328
|
| |
| |
| |
| | |
PiperOrigin-RevId: 209823375
|
| |
| |
| |
| |
| |
| | |
Useful when loading only part of a larger checkpoint, to assert that the Variable objects which do exist in the current program are being restored.
PiperOrigin-RevId: 209823112
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| | |
This implements the semantics suggested in cr/209704042 for CPU and GPU:
X / 0 == -1
X % 0 == X
INT_SMIN / -1 = INT_SMIN
INT_SMIN % -1 = 0
PiperOrigin-RevId: 209821097
|
|\ \
| | |
| | |
| | | |
PiperOrigin-RevId: 209820178
|