| Commit message (Collapse) | Author | Age |
... | |
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211633744
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211631516
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
* Split CondState into CondState (which corresponds to scope previously) and
AncestorState (which tracks which switch/merge nodes are an ancestor of a
ndoe). Previously CondState tracked both but that resulted in difficult to
follow meet rules. Instead by splitting these out the meet for merge and
non-merge are straight forward set operations. The ancestor relation is
similarly easy to compute along with CondState computation.
* Enhance the redundant switch checking: previously we only considered the
predicates but
%s=switch(val=%P, pred=switch(%P_1, %P):then)
is also redundant as if %P is true then %s:else is dead.
* Enhance in-edge testing to insert a switch if a value from an outer context
is consumed inside an inner context.
* Rename CondStateMap to StateMap to match new usage.
PiperOrigin-RevId: 211622021
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211621189
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211598349
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211592901
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
Add a missing check to InferConvolveShape(), the output feature dimension needs to be divisible by feature_group_count.
Also fix some tests which took a const reference to the return value of
a function which doesn't return a reference.
PiperOrigin-RevId: 211592011
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211588937
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 211586062
|
|\ \ \ \
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211584024
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211581486
|
|\ \ \ \ \
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211581348
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
strategy.
PiperOrigin-RevId: 211576839
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
the user doesn't have to pass it again to session_config.
PiperOrigin-RevId: 211576564
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Also minor fix of enabling quantization of shared weights if hybrid evaluation is true.
PiperOrigin-RevId: 211573947
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Parameter server strategy where variables are shared across sessions.
PiperOrigin-RevId: 211573447
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
This is implemented as custom op instead of builtin op because Relu1 is not
supported in Tensorflow and not commonly used.
PiperOrigin-RevId: 211571619
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211570665
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
function.
PiperOrigin-RevId: 211564198
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
- Remove unnecessary use of test_session() in tests that run with eager
execution enabled.
- Use cached_session() instead of test_session()
(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: 211562969
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211562900
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
If the feature_group_count is 1, don't bother showing it as it is not very
informative and a very common scenario. This is consistent with the
HloCustomCall's feature_group_count attribute.
PiperOrigin-RevId: 211560372
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211557743
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211557740
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
This patch uses take by value and move idiom to optimize copying of constructor arguments.
PiperOrigin-RevId: 211553877
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211552101
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
value}}` and `^^key:value^^`. This change consolidate these two format.
PiperOrigin-RevId: 211550259
|
| | | | | | |
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
switching from assertAllEqual to assertAllClose.
PiperOrigin-RevId: 211543406
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
microcontrollers
PiperOrigin-RevId: 211543125
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211542593
|
| |/ / / /
|/| | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
- Remove unnecessary test_session() boilerplate when executing eagerly
- Use self.cached_session() instead of self.test_session() when using graphs
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: 211542360
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211541639
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211540844
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211535930
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211534283
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211532963
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211531374
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 211524810
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
shapes.
PiperOrigin-RevId: 211523104
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
HLO instruction to its operands.
PiperOrigin-RevId: 211521410
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
self -> _parameters -> self cycles were creating work for Python's garbage collector in training loops, where Distribution objects may be created repeatedly when executing eagerly. This CL just fixes that narrow memory issue; I'm not convinced dict(locals()) is super efficient, so we may want to follow up on that for performance.
Adds a few unit tests tests with run_test_in_graph_and_eager_modes(assert_no_eager_garbage=True). It'd be nice to expand this coverage over time.
Includes a small test_util simplification to support this (TFP tests don't like reset_default_graph for some reason). Testing for cycles in the TFP repo will need to wait on the Normal changes from the TF repo syncing.
PiperOrigin-RevId: 211520394
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
`MapDataset`.
PiperOrigin-RevId: 211520001
|
|\ \ \ \ \
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211519911
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211519679
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211519628
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211519250
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
* simplify contraction by collapsing inner dims into single dimension
* get rid of expensive reverse op
~5X improvement when compiled with AVX.
PiperOrigin-RevId: 211518363
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Fixes #21266
PiperOrigin-RevId: 211515918
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 211514287
|