[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

[XLA:CPU] Don't hard-code lane width in horizontal sum routine

The bulk of change is actually in dot_operation_test to get it to the point
where it would have caught this bug.  I made the following changes:

 - Moved some tests under a "no layout assignment pass runs" mode.  This lets
   us test the layout specific aspects of the dot operation more thoroughly.
   Unfortunately not many tests can be run in this mode -- for instance dot
   tests that calls into Eigen won't work here because they need a specific
   layout assignment for correctness.  Tests that runs with layout assignment
   enabled but with non-default layouts are essentially integration tests that
   check that the layout assignment pass + the dot lowering work correctly.

 - Changed the matrix-vector dot tests to check all 4 layouts and the
   matrix-vector Ax+b fusion tests to check all 8 layouts.

 - Duplicated some of the F32 tests to run for F64.

 - Added some new test shapes.

PiperOrigin-RevId: 186511289

Internal change.

PiperOrigin-RevId: 186510594

Activity analysis annotation on FunctionDef

PiperOrigin-RevId: 186510035

In VirtualScheduler, if there is a Recv without a Send, handle the Recv as an
initially ready node.

PiperOrigin-RevId: 186509851

K-FAC: LM algorithm for adapting damping, Example to train MNIST autoencoder model using variable size training data and update damping parameter, add KFACOptimizer.{update_damping}.

PiperOrigin-RevId: 186509305

Made tf.HParams compatible with https://github.com/google/pytype.

PiperOrigin-RevId: 186508693

Add test that checks all core ops have shape functions.

This is meant to be a replacement for the current Python code that
checks that core ops have shape functions registered. Some ops were
missing a shape function, so I added UnknownShape.

This also adds an OpRegistry::GetOpRegistrationData() method for
fetching all the shape functions.

PiperOrigin-RevId: 186508356

Change node to Identity operation for shuffle/reverse operations on scalar values, but not
directly removing those nodes from the graph.

PiperOrigin-RevId: 186505857

Ensure that final layer of networks (which doesn't have an activation) get correctly quantized.

PiperOrigin-RevId: 186505814

Fix a bug in tf.metrics.mean_tensor for case that the weights are very small.

We have renamed metrics_test.MeanTensorTest.testWeighted1d as metrics_test.MeanTensorTest.testBinaryWeighted1d, since the weights on the instances are zeros and ones.

We have added a new metrics_test.MeanTensorTest.testWeighted1d that has small weights. It was failing for the previous implementation, but passes now.

Now the code for mean_tensor() and mean() now use the same _safe_div method. Previously, mean_tensor() used a different means to ensure that we don't divide by zero. This set the denominator to max(1., sum(weights)), which was inaccurate when sum(weights) is non-zero, but less than one.

PiperOrigin-RevId: 186503714

Allow non-integer values for Poisson CDF/PMF.

PiperOrigin-RevId: 186502845

Merge tf.layers.GraphNetwork into tf.keras.Network and remove it.

PiperOrigin-RevId: 186502476

eager/mnist: Point to the example in tensorflow/models instead.

PiperOrigin-RevId: 186502375

framework/ops.py: Stricter check for use of the C API for graph construction.

Fixes #16913

Didn't add an explicit test for this since arguably use of the C API for graph
construction will soon become the default, so I figured testing that
tfe.defun's use of _use_c_api_hack() would only add code that should be deleted
soon.

PiperOrigin-RevId: 186502140

Automatically add a tf.name_scope of the function or class name as the first line of each function.  For functions inside a class, the class name is also prepended to the name scope.  Also adds a decorator param to turn this off (it defaults to True.)

PiperOrigin-RevId: 186499660

Adds inverse_link_fn argument in regression_head. This is used in generalized regression.

PiperOrigin-RevId: 186497656

Add bidirectional sequence LSTM to TFLite Ops.

PiperOrigin-RevId: 186497571

Adding support for dilated convolution.

PiperOrigin-RevId: 186496353

Temporarily disabled part of a test that fails on MacOS

PiperOrigin-RevId: 186494795

Fast-path for losses code.

PiperOrigin-RevId: 186494736

Add S64 add/subtract test.

PiperOrigin-RevId: 186494344

Add link to SECURITY.md from doc describing model loading.

PiperOrigin-RevId: 186489041

Internal change.

PiperOrigin-RevId: 186487354

[XLA] Add FindInstruction and FindComputation helpers to HloTestBase.

These are useful for tests that create HLOs and then search for a
particular computation/instruction.

While we're at it, add a c_find_if utility and fix up the (lack of)
perfect forwarding in some of our other c_foo utilities.

PiperOrigin-RevId: 186482111

Create fast path for common case of finite values in CheckNumericsOp

PiperOrigin-RevId: 186474851

Internal Change

PiperOrigin-RevId: 186472818

Internal updates.

PiperOrigin-RevId: 186472487

In the arithmetic optimizer enqueue the fanout of optimized nodes in a
deterministic order

PiperOrigin-RevId: 186468633

memory_size should be expressed in Bytes, but port::AvailableRam() returns kB.

PiperOrigin-RevId: 186468461

fix not_covered anchor

PiperOrigin-RevId: 186467828

Modify optimized quantized LSTM implementation so that it only needs one instantiation of fixed-point Tanh, for 3 integer bits, regardless of the value of StateIntegerBits

PiperOrigin-RevId: 186464604

Adding some beginner advice to the README.

PiperOrigin-RevId: 186461145

Minor corrections in feature_columns doc

PiperOrigin-RevId: 186449350

[XLA:CPU] Add FP32<->FP16 conversion routines

LLVM generates calls to these functions when lowering some fp16 operations on
certain architectures.  These symbols are defined in compiler-rt but we don't
always link to compiler-rt so these symbols are sometimes absent.

This change adds __gnu_f2h_ieee and __gnu_h2f_ieee as weak symbols.  Making them
weak ensures that we are able to build successfully even when linking to a
compiler-rt that defines these symbols.

PiperOrigin-RevId: 186416684

Add a small test to ensure that rewrites are idempotent.

PiperOrigin-RevId: 186410356

Clarify the shape of convolution arguments.

Replace n-d with size n, as n-d often means rank n, while the arguments here are arrays of size n.

PiperOrigin-RevId: 186406143

TFLite: Define a DELEGATE op type.
PiperOrigin-RevId: 186405366

TFLite Conv2D: Create temporary tensors in Prepare phase.

PiperOrigin-RevId: 186402268

[TF2XLA] Account for input edge of predicate.

PiperOrigin-RevId: 186397549

Make sure the nodes that are refered to by a collection are preserved during an
optimization

PiperOrigin-RevId: 186394467

Automated g4 rollback of changelist 186260342

PiperOrigin-RevId: 186393300

Update decorators transformer with additional clarifications in the tests and handling of the more rare cases when multiple decorators are applied together, as well as when decorators are used with local functions.

PiperOrigin-RevId: 186390564

Add a utility that generalizes getcallargs to non-function callables like constructors and __call__ operators.

PiperOrigin-RevId: 186390545

Fill the new `custom_initial_data(_size)?` fields in TfLiteNode.

PiperOrigin-RevId: 186389819

[XLA] Emit saturating shifts on CPU, GPU and interpreter

With this change shifting out >= bitwidth for shift left and logical shift right
produces 0, and shifting out >= bitwidth for arithmetic shift right produces -1
if the LHS is negative and 0 otherwise.  Before this we were invoking undefined
behavior for these out-of-bounds shifts in LLVM and the HLO evaluator.

PiperOrigin-RevId: 186379160

Shorten grappler per-node report by default with previous behavior behind --verbose.

PiperOrigin-RevId: 186369380

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

Fix a typo in the comment (TFLite)

PiperOrigin-RevId: 186363449

Update contrib/quantize docs to add description of how to use rewrites.

PiperOrigin-RevId: 186362791

Update ops-related pbtxt files.

PiperOrigin-RevId: 186361455

[tf.data] Fix memory leak when not all elements of a `Dataset.from_generator()` are consumed.

This change introduces a new C++ Dataset implementation
(`GeneratorDataset`) that takes three functions:

1. An initialization function that is called before the first use.
2. A "get next" function that is called to produce the elements, until a call
   raises the OutOfRange error.
3. A finalization function that is called before the iterator is destroyed.

Previously, the generator state would only be cleaned up if the caller
consumed *every* element of the generator. In the new version, the
finalization function ensures that the Python-side state of the
generator is released regardless of how the iterator is disposed.

Fixes #16163.

PiperOrigin-RevId: 186360401

Fix a memory corruption issue in boosted trees as the iterators become invalid after an Add.

PiperOrigin-RevId: 186360144

Add documentation to contrib/quantization to reduce confusion with contrib/quantize.

PiperOrigin-RevId: 186358131

Basic LogSoftmax support

PiperOrigin-RevId: 186357933

Add an inspection helper module for related routines not found in the core inspect.

PiperOrigin-RevId: 186357270

Turn on swapping heuristic by default to better manage memory usage on GPU

PiperOrigin-RevId: 186356358

Java: Fix #17130

PiperOrigin-RevId: 186354700

[TF:XLA] Tiny fixes.
Add missing compile-time constant input annotation to BatchToSpaceND.
Make definition of Acosh slightly more accurate. Addition/subtraction of numbers with similar magnitudes is more accurate, and x^2 likely will be further from 1.0 than x due to the doubling of the exponent caused by squaring.

PiperOrigin-RevId: 186353472

Clarify GpuDeviceInfo struct

PiperOrigin-RevId: 186352333