Add custom registered graph optimizers run by MetaOptimizer.

PiperOrigin-RevId: 186837828

Checkpointable: Utility to gather initialization ops

A bit safer, since only variables which will be saved get initialized. Graph
building then raises an error when you've used one which won't be saved. Reduces
the need for the global collection.

Makes it a bit easier to deal with initialization when writing graph/eager
agnostic programs.

PiperOrigin-RevId: 186835744

eager: Change various examples to use tf.keras.Model instead of tfe.Network.

PiperOrigin-RevId: 186834891

Add test for bug in CUB that caused dynamic partition to fail on the GPU.

PiperOrigin-RevId: 186834668

Improvement to the eager device placement heuristic.

PiperOrigin-RevId: 186833677

Adds unit tests for mean op with uint8_t input data.

PiperOrigin-RevId: 186833364

Eager/C: Add a TF_Status argument to a couple of functions.

PiperOrigin-RevId: 186829318

Make tf.size() with optimize=True encode 0 if any dimension is 0.

PiperOrigin-RevId: 186824964

Make it easier to debug @assert_no_garbage_created unit test failures

Prints a bunch of information about each object.

PiperOrigin-RevId: 186823593

* CUB updated to 1.8.0
* updated ShuffleIndex because of API change

PiperOrigin-RevId: 186822637

Preserve user placement as much as possible when optimizing the graph

PiperOrigin-RevId: 186822511

Automated g4 rollback of changelist 185688704

PiperOrigin-RevId: 186820593

Respects some form of log_device_placement in eager.

PiperOrigin-RevId: 186820292

[XLA] Fix BF16 propagation pass to produce matching fusion root and output.

Previously, the propagation pass might produce different procision in the fused
computation's root than the fusion itself, when the fused root doesn't define a buffer.

Add explicit converts at such fusion roots.

PiperOrigin-RevId: 186812368

Add usage example to KMeans Estimator documentation.

PiperOrigin-RevId: 186805772

[TF:XLA] Fix a bug where executor's device_ordinal should be passed to AllocateShapedBuffer. Also enable C64 type for interpreter device.

PiperOrigin-RevId: 186805709

Cleanup for graph functions.

(1) Define constants for the names of Arg and Retval ops, and use them in various places.
(2) Change the signature and documentation for `BuildControlFlow` to reflect the fact that the supplied Graph is not mutated.
(3) Expose the FunctionLibraryRuntime's DeviceMgr, in preparation for multi-device functions.

PiperOrigin-RevId: 186804968

[XLA] Internal change.

PiperOrigin-RevId: 186802115

Remove redundant line which is almost a duplicate of one a few lines below in
datasets_quickstart document.

PiperOrigin-RevId: 186788306

Allow setting of OpMetadata for Send HLOs.

PiperOrigin-RevId: 186777369

Let variables initialized from checkpoints answer ".initialized_value()" correctly.

PiperOrigin-RevId: 186741832

Unify metropolis_hastings interface with HMC kernel.

PiperOrigin-RevId: 186716023

Enable constant propagation across Switch(x,x) by rewriting the two outputs as Const(false), Const(true) with appropriate control dependencies. This is a fairly common pattern when the graph contains assertions.

By rewriting the graph a bit, we can propagate the constants down the two output branches, and just use control dependencies to trigger the selected one at runtime. For example,
         +------+
     x-->|Switch|-->a
     x-->|      |-->b
         +------+

Is rewritten as

         +------+
     x-->|Switch|-->Identity--^>Const(false)-->a
     x-->|      |-->Identity--^>Const(true)-->b
         +------+

(In practice there may be multiple consumers of each output branch.)

PiperOrigin-RevId: 186714991

Pass 'mode' argument to input_fn if appropriate in TPUEstimator.

PiperOrigin-RevId: 186709373

Update ops-related pbtxt files.

PiperOrigin-RevId: 186707935

[XLA] Add SliceInDim to the local Python XLA client.
[XLA] Revise a comment regarding how the two-argument version of Reshape flattens dimensions (it's first-to-last, not major-to-minor since those terms apply to memory layout rather than logical dimensions, and the comment wasn't consistent about major-to-minor or minor-to-major).

PiperOrigin-RevId: 186707393

Add cost estimator tests for the BiasAdd, ReLU, and Conv2D operations.

PiperOrigin-RevId: 186705930

Fix fix Defun logic when returning a value captured from an outer scope.

Previously, the following would fail:

```python
c = tf.constant(...)
@Defun(...)
def Foo(...):
  return c
```

The fix involves ensuring that every output of the function is a member of the function graph, and invoking the capturing logic if it is not.

PiperOrigin-RevId: 186705800

Turn on strip_default_attrs by default during custom export.

PiperOrigin-RevId: 186704976

Moved LIBXSMM convolutions to a separate --define flag so that they are disabled by default.

PiperOrigin-RevId: 186700936

LSTM support: Quantized types, quantization params for 16-bit unfused LSTMs.

PiperOrigin-RevId: 186697357

Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 186696903

Clarify ownership story of TfLiteIntArray* nodes_to_replace

PiperOrigin-RevId: 186696787

Support degenerate strided slicing under XLA.

For example, when start, end, and stride are all positive, but `start` is
greater than `end`, tf2xla used to raise an error instead of returning a
tensor with that dimension size being 0. Latter is the regular
tensorflow and python behavior. This change makes XLA behave the same
way.

PiperOrigin-RevId: 186695842

Internal change.

PiperOrigin-RevId: 186695534

[XLA] Enable most xla/tests for interpreter.

PiperOrigin-RevId: 186695486

Add to disabled tests the date they were last ran and failed.

This is to address the comments in cl/186664459.

PiperOrigin-RevId: 186695081

Add integration tests for remote build execution.

PiperOrigin-RevId: 186694734

New Mutex operations for a distributed-happy and Function-less CriticalSection.

Original idea by Alex Passos; impl and cancellation handling by Eugene Brevdo with help from Alex.

PiperOrigin-RevId: 186692306

Add a regression test for virtual_scheduler.

PiperOrigin-RevId: 186691392

Implement the logic to parse TensorProto (the tensor value for input or filter shape info) in op_level_cost_estimator.

PiperOrigin-RevId: 186685409

[tf.data] Handle a function-raised OutOfRange error correctly in ParallelMapDataset.

PiperOrigin-RevId: 186680982

Merge changes from github.

PiperOrigin-RevId: 186674197

Internal change.

PiperOrigin-RevId: 186673561

Adds support for identifying dilated convolution emulation.

Dilated convolution can be emulated by a chain of SpaceToBatchND -> Conv2D -> BatchtoSpaceND. This change adds a graph transformation that identifies this pattern, and variations of it, and substitutes a true dilated convolution instead.

PiperOrigin-RevId: 186669260

Enable int64 outfeed.

PiperOrigin-RevId: 186667574

[XLA] Enable F16 convolution test for CPU.

Remove TODO(b/72509305) as the issue has been fixed.

PiperOrigin-RevId: 186664459

Add min_check_interval_secs with 5s as default to ValidationMonitor to avoid checking for existance of a new checkpoint overly frequent that can lead to severe performance issues on remote filesystems.

PiperOrigin-RevId: 186662441

Relax one of the error conditions to allow modeling graphs without explicit set of feed nodes.

PiperOrigin-RevId: 186661729

Remove a bit of misleading documentation (we no longer do Graph containers)

PiperOrigin-RevId: 186660057

Internal change.

PiperOrigin-RevId: 186658974

[XLA] HLO BF16 propagation pass.

Using BFloat16Support provided by the backend to determine what precision is needed for
each HloInstruction. If the implementation of some HLOs already reduces input precision to BF16, this pass can enable BF16 on more ops without affecting the result.

PiperOrigin-RevId: 186656378

Generate example for basic lstm cell in tflite

PiperOrigin-RevId: 186656247

Fix BaseGPUDevice, let it report the actual memory limit of the allocator. Also added a helper method to reset ProcessState.

PiperOrigin-RevId: 186655996

Measure the performance of the original placement to ensure that we preserve it
in case the placer isn't given enough time to find a better solution.

PiperOrigin-RevId: 186655094

Add eager support for unit tests for most Keras layers.
A few minor layers were left out:
- noise layers (apparent issue with tf.random_normal)
- bidirectional wrapper
- conv recurrent layers (impending refactor)

PiperOrigin-RevId: 186654795

Checkpointable: Re-use the Saver's SaveableObject infrastructure to create
restore ops, cache them.

The basic infrastructure is usable after this CL in graph and eager, but I still need to make a bunch of objects Checkpointable and make some other usability fixes.

Also sets some of the groundwork for feeding and fetching Python values during save/restore (as in, save has a feed dict now; gathering feeds and placeholders from Checkpointable objects is still to do but should be relatively straightforward).

PiperOrigin-RevId: 186652696

Update bazel toolchains dependency.

PiperOrigin-RevId: 186650360

Add basic support for quantized unfused LSTMs.

PiperOrigin-RevId: 186650338

[TF:XLA] Improve readability of HLO graphs when rendered via Tensorboard.
Add operator metadata around the computation arguments and retvals, so they are grouped together.
Teach the batchnorm expander pass to propagate the operator metadata from the original batch norm operators.

PiperOrigin-RevId: 186648547

Add a test only method to reset ProcessState.

PiperOrigin-RevId: 186647005

Change the MomentumOptimzer lambda so it has the same named argument (learning_rate)
as the MomentumOptimzer constructor.

PiperOrigin-RevId: 186642325

Added a regression test to make sure we deal with large constants properly

PiperOrigin-RevId: 186639709

Change warning message for case where Python detects a colocation that conflicts with a device assignment.

PiperOrigin-RevId: 186637887

Update LLVM for API changes in r325725

PiperOrigin-RevId: 186624266

Add node name to placer device-placement/kernel-assignment error messages .

PiperOrigin-RevId: 186622923

Add S64 add/subtract test and convert tests.

* Fixed bugs in convert from U32 to S64.
END_PUBLIC

*** Original change description ***

BEGIN_PUBLIC
Automated g4 rollback of changelist 186494344

PiperOrigin-RevId: 186616875

[tf.data] Add experimental ability to override the function threadpool.

The purpose of this feature is to enable experimentation with differentiating
the CPU resources available to different stages of a `tf.data` pipeline. As a
concrete example, we might use the new feature to move all input-related work
from the inter-op threadpool onto a separate threadpool, leaving the inter-op
threadpool free to execute higher priority work (such as dispatching ops that
send tensors to an accelerator). The current implementation only allows users
to create fixed-size `tensorflow::ThreadPool` resources, but we could imagine
opening up this API to allow custom threadpools as well.

PiperOrigin-RevId: 186614315

Optionally have persistent make_vjp

PiperOrigin-RevId: 186610572

Adapt TensorFlow to LLVM API change from r325725

PiperOrigin-RevId: 186604023

Fix compile errors by patching eigen locally.

PiperOrigin-RevId: 186592198

Disable flaky keras:metrics_test.

PiperOrigin-RevId: 186573303

Avoid creating large constants since protocol buffers are limited to 2GB in size.

PiperOrigin-RevId: 186567461

Disable flaky moving_average test.

PiperOrigin-RevId: 186563841

Started to open source the RL placer.

PiperOrigin-RevId: 186563773

Deleting test that checks that 2D convolution with NCHW format is not implemented on CPU. The tests fail with MKL because the operation is implemented.

PiperOrigin-RevId: 186558730

Do not set cudnn batch norm persistent mode when doing inference.

PiperOrigin-RevId: 186547439

Delete dot_operation_runtime_test -- it is identical to dot_operation_test.

PiperOrigin-RevId: 186546771

Fix control flow bug.

Without this change, the newly added test would fail with a "Reval[0]"
error. This was due to extra Enter nodes being added in the gradients
graph in order to enter the outer while contexts for a value that was
already in those contexts. The extra Enter nodes would cause the
frames of the execution to be messed up, which prevented the final
value from being propagated to the outermost Exit node.

A similar change is probably needed in WhileContext, although I don't
have a test case for this yet.

PiperOrigin-RevId: 186544961

Fix subtle race condition in ResourceVariable.is_initialized

PiperOrigin-RevId: 186544846

Add more strided R1 tests.

PiperOrigin-RevId: 186542219

Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 186542037

[XLA] Convert large constants of the same value into broadcasts.

PiperOrigin-RevId: 186539902

Update ops-related pbtxt files.

PiperOrigin-RevId: 186537602

Fix lint errors and improve docs in fully_connected_reader.py.

PiperOrigin-RevId: 186537109

Internal change.

PiperOrigin-RevId: 186534524

Move sorting of variables from the implicit_grad family of functions
up to GradientTape.watched_variables() so we also get deterministic
behavior when using the GradientTape as is done in
Optimizer.compute_gradients().

PiperOrigin-RevId: 186533323

Automated g4 rollback of changelist 186494344

PiperOrigin-RevId: 186530782

[tf.data] Move the `tf.contrib.data.unique()` C++ implementation to contrib.

PiperOrigin-RevId: 186530750

Internal-only change.

PiperOrigin-RevId: 186528023

Disable flaky test tensorflow/contrib/opt:moving_average_optimizer_test

PiperOrigin-RevId: 186527039

Add the ability to specify an explicit `training` argument when calling a Model (including Sequential).

PiperOrigin-RevId: 186526925

Don't assign device for the keras part of _saved_first_checkpoint. Fix #14504.

PiperOrigin-RevId: 186526175

Don't require shape functions when creating ops from Python using the C API.

There are many ops out there without shape functions, and it's very
onerous to add UnknownShape to all of them.

PiperOrigin-RevId: 186524294

Mark the `SerializeSparseOp<Variant>` kernel as inexpensive.

Since this op only performs a constant amount of work, and typically
executes in a few microseconds, it should be profitable to execute
this op inline, rather than scheduling it on a remote thread.

PiperOrigin-RevId: 186522885

Record gradient in C

PiperOrigin-RevId: 186522240

Improve error message.

PiperOrigin-RevId: 186521902

Ensured that the model pruner outputs the nodes of the optimized graph in a
deterministic order

PiperOrigin-RevId: 186520272

Make CPUAllocator VisitableAllocator, for better RDMA networking.

PiperOrigin-RevId: 186518037

Make with_c_api a no-op if the C API is already enabled.

This will help transition to turning the C API on by default by
preventing new tests from breaking.

PiperOrigin-RevId: 186516976