| Commit message (Collapse) | Author | Age |
... | |
|
|
|
| |
PiperOrigin-RevId: 177516249
|
|
|
|
|
|
| |
This fixes a TODO.
PiperOrigin-RevId: 177508258
|
|
|
|
|
|
| |
These WriteTextProto() calls are purely for diagnostics (and are usually called within IF_VLOG_IS_ON(1) guards), but if they fail to write to a file, they'll take down the entire calling process. Which makes debugging difficult, and seems rather astonishing.
PiperOrigin-RevId: 177506379
|
|
|
|
| |
PiperOrigin-RevId: 177506166
|
|
|
|
| |
PiperOrigin-RevId: 177505909
|
|
|
|
| |
PiperOrigin-RevId: 177503567
|
|
|
|
| |
PiperOrigin-RevId: 177502497
|
|
|
|
| |
PiperOrigin-RevId: 177502259
|
|
|
|
|
|
| |
connectivity check code.
PiperOrigin-RevId: 177499365
|
|
|
|
|
|
| |
accumulator.
PiperOrigin-RevId: 177497240
|
|
|
|
|
|
|
|
| |
absorbing elements for addition and multiplication with Identity.
Get rid of a gratuitous copy of the entire graph in the main optimizer loop.
PiperOrigin-RevId: 177491247
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
Some of the functionality has bled into the generic XlaCompiledCpuFunction, but
there still remains a fair amount of work to do before the AOT side of things
start working.
This CL also fixes a bug I introduced in a previous CL -- when I changed
IrEmitter::hlo_to_profile_idx_ to a value, I changed the signature of the
generated function to always have the "profile_counters" argument when the AOT
client code expects the signature to not have that argument. In practice this
wasn't an issue for the standard x86 calling convention, but it could easily
have been problematic on other architectures and calling conventions. After this
change the mismatch is no longer present.
PiperOrigin-RevId: 177481998
|
|
|
|
|
|
| |
contracting and batch dimensions.
PiperOrigin-RevId: 177481231
|
|
|
|
|
|
| |
algebraic manipulation instead of slices and pads that are more difficult to fuse.
PiperOrigin-RevId: 177480353
|
|
|
|
| |
PiperOrigin-RevId: 177478106
|
|
|
|
| |
PiperOrigin-RevId: 177478003
|
|
|
|
| |
PiperOrigin-RevId: 177474943
|
|
|
|
| |
PiperOrigin-RevId: 177473833
|
|
|
|
|
|
| |
Rolls back the rollback with some swiggery to get python3 to work.
PiperOrigin-RevId: 177470328
|
|
|
|
|
|
|
|
| |
which support mixed precision training.
Until the necessary support for mixed precision fused batch norm is added to XLA, implement by casting to a common type.
PiperOrigin-RevId: 177468202
|
|
|
|
| |
PiperOrigin-RevId: 177467740
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
Prior to this change, functions that closed over external tensors in a
while loop would cause a segfault at runtime. This is because the
external tensors are temporarily represented as placeholders in the
function body before being replaced by input parameters, and the
placeholders would be created directly in the while loop body. This
would eventually lead to using the input tensor in the while loop body
without an enter node. This wasn't caught by the runtime check because
it isn't applied to function bodies.
This change adds tests for capturing tensors in a while loop body and
in a cond context. Note that the cond test passed without this fix.
PiperOrigin-RevId: 177464541
|
|
|
|
|
|
| |
with C API enabled.
PiperOrigin-RevId: 177462553
|
|
|
|
| |
PiperOrigin-RevId: 177444775
|
|
|
|
| |
PiperOrigin-RevId: 177423981
|
|
|
|
| |
PiperOrigin-RevId: 177419730
|
|
|
|
| |
PiperOrigin-RevId: 177418947
|
|
|
|
|
|
| |
conservatively.
PiperOrigin-RevId: 177391746
|
|
|
|
| |
PiperOrigin-RevId: 177382431
|
|
|
|
|
|
| |
rather than `ds`, `bs`.
PiperOrigin-RevId: 177381853
|
|
|
|
|
|
|
| |
- Parameter names consistent in function declarations and definitions
- Class members naming
PiperOrigin-RevId: 177379085
|
|
|
|
| |
PiperOrigin-RevId: 177375237
|
|
|
|
| |
PiperOrigin-RevId: 177371177
|
|
|
|
| |
PiperOrigin-RevId: 177367024
|
|
|
|
| |
PiperOrigin-RevId: 177362829
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
With current tensorflow code, if user modifies
some operation after session.run() was called, this modification will
never make it to the C++ runtime and no errors will be raised leading
to silent wrong results.
This change adds checks for such cases when C API is enabled. We don't
change the code path for C API being disabled because C API should
be enabled by default soon.
PiperOrigin-RevId: 177359630
|
|
|
|
|
|
| |
A layer on top of the batcher could use this interface to pre-split large tasks that exceed the max batch size.
PiperOrigin-RevId: 177359263
|
|
|
|
| |
PiperOrigin-RevId: 177357416
|
|
|
|
| |
PiperOrigin-RevId: 177356621
|
|
|
|
| |
PiperOrigin-RevId: 177354924
|
|
|
|
| |
PiperOrigin-RevId: 177354350
|
|
|
|
| |
PiperOrigin-RevId: 177353959
|
|
|
|
| |
PiperOrigin-RevId: 177351409
|
|
|
|
| |
PiperOrigin-RevId: 177350575
|
|
|
|
|
|
|
| |
For example toolchains with clang may set -Oz which is more analogous to gcc's
-Os. -Os for clang is closer to -O2.
PiperOrigin-RevId: 177347371
|
|
|
|
| |
PiperOrigin-RevId: 177344207
|
|
|
|
| |
PiperOrigin-RevId: 177343020
|
|
|
|
|
|
| |
Add tf.contrib.data.shuffle_and_repeat which reshuffles its input on each epoch. Going forward, this will replace reshuffle_each_iteration=true.
PiperOrigin-RevId: 177339570
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
This change partially undoes my previous commit
(https://github.com/tensorflow/tensorflow/commit/f4c18a0eb05e21bae397c9c16527ff8080cae6b8).
Without this change, if an op is added that has invalid input shapes
and also requires a kernel label, the op will be added to the graph
before shape inference is run, but then the shape inference error will
prevent the kernel label from being applied. The placer will then
complain about the missing label when the graph is run.
This is only a problem with the C API disabled. With the C API
enabled, shape inference is performed when the TF_Operation is created
in Operation.__init__. Thus we can and should move the _add_op call
back to Operation.__init__ once the _USE_C_API flag is removed.
PiperOrigin-RevId: 177338123
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
By respecting cgo rules on pointers.
Without the change to graph.go, the newly added test would fail with:
panic: runtime error: cgo argument has Go pointer to Go pointer
in the call to the C function TF_SetAttrShapeList.
Fixes #14891
PiperOrigin-RevId: 177336663
|