| Commit message (Collapse) | Author | Age |
|
|
|
|
|
| |
since xla::DotGeneral was added.
PiperOrigin-RevId: 213052269
|
|
|
|
| |
PiperOrigin-RevId: 213049674
|
|
|
|
|
|
|
| |
Also added some experimental C APIs for facilitate the use of eager C APIs in
S4TF compiler.
PiperOrigin-RevId: 213041780
|
|
|
|
| |
PiperOrigin-RevId: 213040362
|
|
|
|
| |
PiperOrigin-RevId: 213037039
|
|
|
|
| |
PiperOrigin-RevId: 213034078
|
|
|
|
| |
PiperOrigin-RevId: 213028338
|
|
|
|
| |
PiperOrigin-RevId: 213027176
|
|
|
|
| |
PiperOrigin-RevId: 213026080
|
|
|
|
| |
PiperOrigin-RevId: 213025676
|
|
|
|
| |
PiperOrigin-RevId: 213023382
|
|
|
|
| |
PiperOrigin-RevId: 213023245
|
|
|
|
| |
PiperOrigin-RevId: 213022233
|
|
|
|
| |
PiperOrigin-RevId: 213017532
|
|
|
|
|
|
|
|
| |
Stateless MapDatasets can be paralellized by switching to ParallelMapDataset. We set `num_parallel_calls` to 2 for now, but in the future a special value will be used that result in the optimal value to be selected dynamically at runtime.
This patch also exposed a memory leak which was fixed.
PiperOrigin-RevId: 213015223
|
|
|
|
|
|
|
|
| |
There is no reason to limit ourselves to float32 dot product operations, we
simply convert to and from float32 around the reduction to simulate the
precision change.
PiperOrigin-RevId: 213014410
|
|
|
|
| |
PiperOrigin-RevId: 213012717
|
|
|
|
| |
PiperOrigin-RevId: 213010458
|
|
|
|
| |
PiperOrigin-RevId: 213008707
|
|
|
|
| |
PiperOrigin-RevId: 213008118
|
|
|
|
| |
PiperOrigin-RevId: 213007905
|
|
|
|
| |
PiperOrigin-RevId: 213007422
|
|
|
|
|
|
|
|
|
|
| |
tf.gradients() can't be used to take the gradient of eager
computations, but it should handle ops that take captured eager
tensors as input as long as the gradient computation doesn't depend on
that input. This change makes sure the gradient algorithm doesn't try
to access the op, etc. of EagerTensors.
PiperOrigin-RevId: 213007155
|
|
|
|
|
|
| |
closes #21790, fixes #21735
PiperOrigin-RevId: 213003724
|
|
|
|
| |
PiperOrigin-RevId: 212997520
|
|
|
|
| |
PiperOrigin-RevId: 212991181
|
|
|
|
| |
PiperOrigin-RevId: 212989480
|
|
|
|
| |
PiperOrigin-RevId: 212986442
|
|
|
|
| |
PiperOrigin-RevId: 212972521
|
|
|
|
|
|
| |
Make the test large to prevent occasional timeouts on CPU. This should normally complete in well under a minute.
PiperOrigin-RevId: 212953337
|
|
|
|
|
|
| |
string and std::string are not necessarily the same thing in TF, but this code assumed that they are.
PiperOrigin-RevId: 212952877
|
|
|
|
| |
PiperOrigin-RevId: 212949973
|
|
|
|
|
|
|
|
|
|
|
| |
FloorDiv (which corresponds to the // operator in python) supports uint8 and
uint16 (but not uint32) in TF. Using xla::Abs on unsigned types throws an error,
but the rounding logic is trivial for unsigned types so just do a plain Div.
This isn't tested yet because we don't have any targets supporting uint8 or
uint16 yet.
PiperOrigin-RevId: 212946132
|
|
|
|
| |
PiperOrigin-RevId: 212936412
|
|
|
|
|
|
|
| |
- XLAContext is never used in the class, remove it from member list.
- Be more clear in the comment that the result is written to the context from the compilation device.
PiperOrigin-RevId: 212924213
|
|
|
|
| |
PiperOrigin-RevId: 212921868
|
|
|
|
| |
PiperOrigin-RevId: 212920113
|
|
|
|
|
|
|
|
|
| |
Also puts a ".unoptimized" suffix on dumped HLO protobuf files
to avoid the unoptimized dumped HLO protobuf colliding with the
optimized dumped HLO protobufs when the same dump directory is
specified for both.
PiperOrigin-RevId: 212914100
|
|
|
|
| |
PiperOrigin-RevId: 212912615
|
|
|
|
| |
PiperOrigin-RevId: 212908218
|
|
|
|
| |
PiperOrigin-RevId: 212905536
|
|
|
|
|
|
|
| |
The original CL was rolled back due to op registration conflicts in the pip.
Resolve the issue by only including core:ops in the toco binary itself, not in intermediate libraries.
PiperOrigin-RevId: 212902838
|
|
|
|
|
|
| |
achieve the same effect.
PiperOrigin-RevId: 212901207
|
|
|
|
|
|
| |
`make_tf_record_dataset` default `prefetch` buffer size to auto-tune (from 1).
PiperOrigin-RevId: 212900920
|
|
|
|
|
|
| |
This supports pathlib and other non-string path types.
PiperOrigin-RevId: 212897666
|
|
|
|
|
|
|
| |
This prevents an undefined behavior with signed integer overflow in
decode.bmp.op.
PiperOrigin-RevId: 212897289
|
|
|
|
| |
PiperOrigin-RevId: 212896336
|
|
|
|
|
|
| |
8ff1374008259719b54a8cb128ef951c02da164c
PiperOrigin-RevId: 212893036
|
|
|
|
| |
PiperOrigin-RevId: 212892844
|
|
|
|
|
|
|
|
| |
self.maxDiff instead of self.maxDiffs:
"Diff is 2708 characters long. Set self.maxDiff to None to see it."
PiperOrigin-RevId: 212892831
|