| Commit message (Collapse) | Author | Age |
... | |
|
|
|
|
|
|
|
| |
When comparing backends, it is useful to take an HLO optimized for one backend and perform transformations in order to match numerics. This can be thought of as finding a lowest common denominator.
Move this grouping of passes into its own HloPassPipeline that can be reused in a few different places.
PiperOrigin-RevId: 191914799
|
|
|
|
|
|
| |
cwise_lib_hdrs. This is for clients that want to use the cwise_ops machinery when making their own custom ops, including cwise_lib directly causes multiple definition linker errors.
PiperOrigin-RevId: 191914445
|
|
|
|
| |
PiperOrigin-RevId: 191914398
|
|
|
|
| |
PiperOrigin-RevId: 191913834
|
|
|
|
|
|
| |
the execution time. Until more sophisticated methods are added, we resort to the roofline model to calculate such cost.
PiperOrigin-RevId: 191913626
|
|
|
|
|
|
|
|
| |
HloInstruction.
This enables use of the function without an actual instruction (eg, if you just
have an HloProto).
PiperOrigin-RevId: 191905914
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
Before:
TFE_Context would check nullptr, and the function would fail straight away.
Now:
TFE_Context is nullptr, so it skips down to checking the status, and an error
is raised.
I'm not able to find in SWIG documentation how to order typemaps in the
generated code - ideally, I'd order it to check the status typemap first. This
code makes it not dependent on this ordering either way.
PiperOrigin-RevId: 191905893
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
Once checked in, this will be the only way I know of to save canonical weights
when executing eagerly. Eager's name-based saving support will only do the
opaque parameter buffer.
I'm not going to try converting everything in one go, but it's a start at
least. And everything else should raise a NotImplementedError rather than
silently not saving correctly.
Single-layer cuDNN cells can be swapped for un-wrapped cuDNN compatible cells or
single cells wrapped in MultiRNNCells. Multi-layer cells need MultiRNNCell
wrapping.
PiperOrigin-RevId: 191905703
|
|
|
|
| |
PiperOrigin-RevId: 191900015
|
|
|
|
|
|
|
|
|
| |
The added functionality is a substitute for the implicit prefetching in
`tfe.Iterator`, and the two paths will converge in a future change.
Fixes #18260.
PiperOrigin-RevId: 191897666
|
|
|
|
|
|
| |
This change allows advanced input pipelines (e.g. StreamingFilesDataset, or split-pipelines that use py_func's) to run in GKE- and GKE-like enviornments.
PiperOrigin-RevId: 191897639
|
|
|
|
| |
PiperOrigin-RevId: 191897098
|
|
|
|
|
|
| |
default session config.
PiperOrigin-RevId: 191895856
|
|
|
|
|
|
|
|
|
| |
times in the inputs/outputs list of an op. The (non-essential) computation
of the optimal workspace size had not been updated for that case, causing it
to fail on a simple test case. Moreover, the initial implementation had some
redundant usage of std::find that this CL moves to a shared helper function.
PiperOrigin-RevId: 191894081
|
|
|
|
|
|
|
|
| |
so the
self._variables will be empty list. So pass a function which returns list of trainable variables to estimator.
PiperOrigin-RevId: 191893084
|
|
|
|
| |
PiperOrigin-RevId: 191889484
|
|
|
|
|
|
| |
This would produce clusters where XLA cannot optimize anything.
PiperOrigin-RevId: 191887414
|
|
|
|
|
|
|
|
|
|
| |
with rationales.
Implement lightweight PRNG for seed generation in that style.
Enables incremental refactoring of existing code into this style.
PiperOrigin-RevId: 191884573
|
|
|
|
| |
PiperOrigin-RevId: 191881135
|
|
|
|
| |
PiperOrigin-RevId: 191869400
|
|
|
|
| |
PiperOrigin-RevId: 191860462
|
|
|
|
|
|
| |
Use the owned ClientGraph as the source for the node_to_name_ map, rather than the borrowed GraphExecutionState (which can be deleted while the ReffedClientGraph is in use).
PiperOrigin-RevId: 191847023
|
|
|
|
| |
PiperOrigin-RevId: 191840652
|
|
|
|
| |
PiperOrigin-RevId: 191838806
|
|
|
|
|
|
| |
adaptive sampling is off.
PiperOrigin-RevId: 191836004
|
|
|
|
|
|
|
|
|
|
| |
convolution routine.
Before: ConvolutionTest/0.StridedFilter (41812 ms)
After: ConvolutionTest/0.StridedFilter (28054 ms)
Speedup: 42 / 28 = 1.5x
PiperOrigin-RevId: 191835735
|
|
|
|
|
|
| |
usage visualization
PiperOrigin-RevId: 191832832
|
|
|
|
|
|
| |
allows the batch_and_drop_remainder function to be used instead of the default batch function.
PiperOrigin-RevId: 191831842
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
This speeds up the implementation of conv because Literal::Get calls
Literal::Piece::data, which is relatively slow.
Instead, we call Literal::Data() once and cache the result.
Before: ConvolutionTest/0.StridedFilter (59094 ms)
After: ConvolutionTest/0.StridedFilter (41812 ms)
Speedup: 59/42 = 1.4x
PiperOrigin-RevId: 191830741
|
|
|
|
|
|
|
|
|
|
|
|
| |
This change makes it so shapes are computed only when requested with
_USE_C_API = True. Note that the C API will still raise a shape error
if necessary when the op is created.
In addition, it cleans up the logic for _USE_C_SHAPES = True. In this
case, we lazily fetch and cache shapes directly from the C API. We no
longer need set_shapes_for_outputs at all in this case.
PiperOrigin-RevId: 191830565
|
|
|
|
| |
PiperOrigin-RevId: 191826636
|
|
|
|
|
|
| |
and output tensors.
PiperOrigin-RevId: 191825756
|
|
|
|
| |
PiperOrigin-RevId: 191824447
|
|
|
|
| |
PiperOrigin-RevId: 191822664
|
|
|
|
| |
PiperOrigin-RevId: 191819577
|
|
|
|
| |
PiperOrigin-RevId: 191811610
|
|
|
|
|
|
| |
This in turn allows to statically obtain a block's syntactic closure.
PiperOrigin-RevId: 191809965
|
|
|
|
| |
PiperOrigin-RevId: 191809110
|
|
|
|
| |
PiperOrigin-RevId: 191807973
|
|
|
|
| |
PiperOrigin-RevId: 191807862
|
|
|
|
| |
PiperOrigin-RevId: 191805682
|
|
|
|
| |
PiperOrigin-RevId: 191805453
|
|
|
|
|
|
| |
Mul reductions want a 1 as their init value, not a 0 or a random value.
PiperOrigin-RevId: 191802819
|
|
|
|
| |
PiperOrigin-RevId: 191797853
|
|
|
|
| |
PiperOrigin-RevId: 191794956
|
|
|
|
|
|
| |
and binary_classification_head.
PiperOrigin-RevId: 191793392
|
|
|
|
| |
PiperOrigin-RevId: 191788253
|
|
|
|
| |
PiperOrigin-RevId: 191786563
|
|
|
|
| |
PiperOrigin-RevId: 191784709
|
|
|
|
| |
PiperOrigin-RevId: 191784430
|