| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 215553161
|
|
|
|
| |
PiperOrigin-RevId: 215534396
|
|
|
|
|
|
| |
Otherwise, when parsing a single instruction, the parsed module doesn't have a name, which won't pass the hlo verifier check.
PiperOrigin-RevId: 215519412
|
|
|
|
| |
PiperOrigin-RevId: 215518288
|
|
|
|
| |
PiperOrigin-RevId: 215517752
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
layout so that it can be used by the HLO verifier.
Change the function to a static member function of the LayoutAssignment class.
Add an std::function member to LayoutAssignment to store the function object
passed down from the backend compiler class and use it to decide whether an
instruction can change layouts.
Fix affected test cases.
PiperOrigin-RevId: 215515611
|
|
|
|
| |
PiperOrigin-RevId: 215512168
|
|
|
|
|
|
|
|
|
| |
tf.data objects
- Previously, when validation_steps was missing, the error message incorrectly says "please provide either batch_size or steps_per_epoch". Now it reads "please provide either batch_size or validation_steps".
- Some whitespace-related fixes.
PiperOrigin-RevId: 215503991
|
|
|
|
| |
PiperOrigin-RevId: 215503549
|
|
|
|
| |
PiperOrigin-RevId: 215501709
|
|
|
|
|
|
| |
one function.
PiperOrigin-RevId: 215501702
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
turning into a constant is a
discardable array. If it's not discardable, it means that the user wants this array to keep existing
in a way that is observable to them, i.e. not as weights.
Typical example: a Fill op outputs an array that is passed as a RNN state array (non-discardable).
It seems that so far we have been relying on accidental ordering of graph transformations for such state
arrays not to be accidentally turned into constants. Instead, the desired graph transformation here is
RemoveUnusedOp noticing that such a Fill can be discarded since its output is a RNN state array.
So I don't have a test for this, but this seems to be tightening existing behavior, and should be good
to have as long as it does not regress anything.
PiperOrigin-RevId: 215500760
|
|
|
|
|
|
|
|
| |
corresponding bugs fixed. The bugs that
were work-arounded were fixed and verified.
PiperOrigin-RevId: 215497418
|
|
|
|
|
|
| |
optimization parameter protos and removed uses of that functionality in tests.
PiperOrigin-RevId: 215494433
|
|
|
|
| |
PiperOrigin-RevId: 215492782
|
|
|
|
|
|
| |
`make_one_shot_iterator` which is to be deprecated in future.
PiperOrigin-RevId: 215491729
|
|
|
|
| |
PiperOrigin-RevId: 215487633
|
|
|
|
|
|
| |
https://github.com/pypa/auditwheel/issues/102
PiperOrigin-RevId: 215486669
|
|
|
|
| |
PiperOrigin-RevId: 215485333
|
|
|
|
|
|
|
| |
There is no known conceptual reason we can't use XLA, but in practice
we have some build issues that will need to be fixed.
PiperOrigin-RevId: 215484942
|
|\
| |
| |
| | |
PiperOrigin-RevId: 215483141
|
|\ \
| | |
| | |
| | | |
PiperOrigin-RevId: 215480021
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 215479788
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 215477724
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 215473351
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 215473319
|
|\ \ \
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 215473010
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
Also stop truncating operands in the canonical format.
PiperOrigin-RevId: 215466465
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
be SparseTensors as well.
PiperOrigin-RevId: 215466199
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
unintended effect of causing the second line not to run at all depending
on the result from the first line.
PiperOrigin-RevId: 215466006
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
consistent HLO instructions.
Fix a dot test that disables layout assignment pass to not generate layout
inconsistent HLO instructions. This includes only adding the dot result to an
addend with the same layout, and disabling algebraic simplification which may
transform a dot to a multiplication with inconsistent layouts.
PiperOrigin-RevId: 215463477
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
are equal to either the min or the max value, so that they are trivially
exactly quantized. This case does not normally occur for true learned weights,
which is what this warning is intended for.
PiperOrigin-RevId: 215463096
|
|\ \ \ \
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
kingofthebongo2008:version_info_cc_generated_only_once
PiperOrigin-RevId: 215462171
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215462000
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215460064
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215459075
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215454323
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215448397
|
|\ \ \ \ \
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 215447391
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
may be replaced by automatic shape inference in TF 2.0 (or before).
Add a output_shapes attr to While op to allow output shapes to be different from the incoming loop_vars.
PiperOrigin-RevId: 215446737
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Many of the HLO parser tests verify that an text form of an HLO module preserves all information when running through ToString then parsing. It makes sense to also use these tests to exercise proto serialization/deserialization. This is done by adding additional instantiations of the parameterized parsing tests. This caught several bugs which are fixed in this CL:
(1) Domain instructions were not being serialized properly.
(2) Host send/recv instructions did not preserve the is_host_transfer bit.
(3) Sparse literals could not be serialized or deserialized.
PiperOrigin-RevId: 215445200
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
tf.distributions API.
PiperOrigin-RevId: 215441733
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 215440829
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 215440356
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
list constructors.
PiperOrigin-RevId: 215435720
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
annotation.
PiperOrigin-RevId: 215435613
|
|\ \ \ \ \ \
| | | | | | |
| | | | | | |
| | | | | | | |
PiperOrigin-RevId: 215431884
|
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | | |
A hash map for 18 pointers is just a waste of space.
PiperOrigin-RevId: 215428176
|
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | | |
Also remove add_arg_scope.
PiperOrigin-RevId: 215426187
|
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | |
| | | | | | | |
Also convert unordered_map to flat/node_hash_map where the comments allow.
PiperOrigin-RevId: 215410566
|