| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 200443383
|
|
|
|
| |
PiperOrigin-RevId: 200442886
|
|
|
|
|
|
| |
The `defun_args()` helper flattens a nested structure down into the flat tuple of tensor types expected by `Defun`.
PiperOrigin-RevId: 200441074
|
|
|
|
|
|
| |
slow-running tests and re-enables it as a 'large' test.
PiperOrigin-RevId: 200440883
|
|
|
|
|
|
|
|
|
|
| |
InputTensor and OutputTensor classes from TF core.
Add equality and hash methods to InputTensor and OutputTensor.
No functional changes intended.
PiperOrigin-RevId: 200440015
|
|
|
|
|
|
| |
Otherwise passing in name="" results in trying to name the If op "".
PiperOrigin-RevId: 200439070
|
|
|
|
| |
PiperOrigin-RevId: 200435608
|
|
|
|
| |
PiperOrigin-RevId: 200434656
|
|
|
|
| |
PiperOrigin-RevId: 200432478
|
|
|
|
| |
PiperOrigin-RevId: 200431906
|
|
|
|
| |
PiperOrigin-RevId: 200431713
|
|
|
|
| |
PiperOrigin-RevId: 200430314
|
|
|
|
| |
PiperOrigin-RevId: 200427133
|
|
|
|
|
|
| |
This cuts down on the amount of repeated (or near-repeated) code in Dataset wrappers.
PiperOrigin-RevId: 200424152
|
|
|
|
|
|
| |
TensorFlow eager mode execution.
PiperOrigin-RevId: 200422481
|
|
|
|
| |
PiperOrigin-RevId: 200421914
|
|
|
|
|
|
|
|
| |
- Passes use_nnapi from TestDriver to interpeter
- Adds command line flag to generated tests for NNAPI
- Fixes logic for allocating im2col tensor so that tests pass without NNAPI
PiperOrigin-RevId: 200416472
|
|
|
|
| |
PiperOrigin-RevId: 200414970
|
|
|
|
|
|
|
|
| |
This support is unused and adds complexity to an already very complicated piece of code.
No (observable) functional changes intended.
PiperOrigin-RevId: 200411522
|
|
|
|
| |
PiperOrigin-RevId: 200411493
|
|
|
|
|
|
|
|
|
|
| |
Add CompileAheadOfTime parameter that can optionally be populated during
compilation process.
This change is to allow populating metadata even if the CompileAheadOfTime
fails.
PiperOrigin-RevId: 200407917
|
|
|
|
|
|
| |
for debugging purposes.
PiperOrigin-RevId: 200397151
|
|
|
|
| |
PiperOrigin-RevId: 200395171
|
|
|
|
| |
PiperOrigin-RevId: 200392587
|
|
|
|
|
|
| |
into a helper.
PiperOrigin-RevId: 200386950
|
|
|
|
|
|
|
| |
style guide,
https://google.github.io/styleguide/cppguide.html#Run-Time_Type_Information__RTTI_
PiperOrigin-RevId: 200380532
|
|
|
|
| |
PiperOrigin-RevId: 200378252
|
|
|
|
| |
PiperOrigin-RevId: 200370679
|
|
|
|
| |
PiperOrigin-RevId: 200362771
|
|
|
|
|
|
|
|
|
| |
We need to be careful how we compare reduce shapes.
- If comparing against non-reduce shapes, we should compare the operand shape of a reduce with the other shape.
- If comparing two reduce shapes, we need to compare both the operand shapes and the reduce shapes.
Also, if we already have a multi-output fusion node, we should pick one of its reduce instructions for comparison, because it has more constraints than the other instructions.
PiperOrigin-RevId: 200353595
|
|
|
|
| |
PiperOrigin-RevId: 200352941
|
|
|
|
|
|
| |
subclasses from HloInstruction.
PiperOrigin-RevId: 200337508
|
|
|
|
| |
PiperOrigin-RevId: 200334496
|
|
|
|
|
|
| |
HloInstruction.
PiperOrigin-RevId: 200333922
|
|
|
|
| |
PiperOrigin-RevId: 200333639
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
and deprecate tf.contrib.distributions.
This CL:
- copies Distributions code into TFP
- Updates it to use TFP idioms, e.g., 'tf.matmul' instead of 'math_ops.matmul'.
- Creates a new `internal` subdirectory for modules that must be importable but not visible externally by way of `__init__.py`.
- Adds a deprecation notice to tf.contrib.distributions BUILD files and to the class constructors of all tf.contrib Distributions and Bijectors. (and removes tests from the deprecated copies of `distribution_test` and `normal_conjugate_posteriors_test` whose `instance()` calls would be broken by the deprecation wrapper).
- Updates miscellaneous references to distributions in TFP. It does not attempt to migrate *all* TFP code to use TFP Distributions; this will be done in a separate CL.
PiperOrigin-RevId: 200333629
|
|
|
|
|
|
|
|
| |
All callables returned from this private API now accept a
"run_metadata" keyword argument whose behavior matches the
`run_metadata` argument accepted by `Session.run()`.
PiperOrigin-RevId: 200331667
|
|
|
|
| |
PiperOrigin-RevId: 200327849
|
|
|
|
| |
PiperOrigin-RevId: 200322035
|
|
|
|
|
|
|
| |
The test is primarily an example of what approaches currently work
for sparse operations when we mostly want to run on TPU.
PiperOrigin-RevId: 200320045
|
|
|
|
|
|
| |
Checkpointing datasets with in-memory caching is not supported yet.
PiperOrigin-RevId: 200316958
|
|
|
|
| |
PiperOrigin-RevId: 200314093
|
|
|
|
|
|
| |
This is unused, and, as it turns out, is broken for sparse shapes.
PiperOrigin-RevId: 200313641
|
|
|
|
| |
PiperOrigin-RevId: 200309129
|
|
|
|
| |
PiperOrigin-RevId: 200306833
|
|
|
|
|
|
| |
bias_add_grad and reduce_sum.
PiperOrigin-RevId: 200306546
|
|
|
|
| |
PiperOrigin-RevId: 200294932
|
|
|
|
|
|
|
| |
kDynamicSlice only needs to reads the part of the operand that is sliced out.
kDynamicUpdateSlice needs to read the update, and write it out to the relevant part of the updated operand, but does not need to read the updated operand at all.
PiperOrigin-RevId: 200293681
|
|
|
|
|
|
|
|
|
|
|
|
| |
After scheduling HLOs it is very beneficial to try more copy elision: The
sequential ordering from the schedule is stricter than the data-dependency ordering used
during copy insertion.
Also, allow more operands to share a buffer with their user. In particular, the user has to be element-wise only wrt to the specified operand, and not wrt to all operands.
These two changes allow more copies to be eliminated.
PiperOrigin-RevId: 200292049
|
|
|
|
| |
PiperOrigin-RevId: 200289114
|