Update api_def for UniqueWithCounts/UniqueWithCountsV2

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

Add test cases for unique_with_counts_v2

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

Add python wrapper for unique_with_counts

to call gen_array_ops._unique_with_counts

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

Hide UniqueWithCounts and UniqueWithCountsV2 in hidden_ops.txt

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

Enable multi-dimensional and axis support for tf.unique_with_counts

This fix tries to address the issue raised in 16499 to bring
multi-dimensional and axis support for `unique_with_counts`.

When `UniqueV2` kernel was added in 12952, it actually supports
multi-dimensional and axis support for `unique_with_counts` as well,
just not registered.

This fix:
1. Register `UniqueWithCountsV2` kernel to have axis support.
2. Hide both `UniqueWithCounts` and `UniqueWithCountsV2`
3. Add python unique_with_counts wrapper to call `gen_array_ops._unique_with_counts`
4. If APi review passes and the PR merges, `unique_with_counts` will switch to `gen_array_ops._unique_with_counts_v2` (in 3 weeks).

This fix fixes 16499.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>

Merge pull request #17134 from tensorflow/timeseries-head

TFTS: Enhancements to timeseries head

Merge pull request #17148 from halhorn/beam_search_decoder_for_float16

fix: Error on we feed float16 values into BeamSearchDecoder

Merge pull request #17152 from facaiy/BLD/ci_allow_bad_user_name

docker ci allows bad name user

Merge pull request #17224 from cclauss/patch-1

TF Lite: from six.moves import xrange for Python 3

Merge pull request #17139 from deroneriksson/operation_semantics_typos

Fix typos in Operation Semantics docs

Merge pull request #17111 from carlthome/patch-9

Add cast functions for complex64 and complex128

Merge pull request #17171 from ilya-biryukov/fix_cuda_compilation

Fix compiler error with cuda-clang

Merge pull request #17226 from yifeif/branch_186777369

Branch 186777369

Remove repeated defination due to auto-merge.

Merge commit for internal changes

from six.moves import xrange for Python 3

Lines 1785 and 1818 contain calls to the Python 2-only builtin function __xrange()__ which was removed in Python 3 in favor of __range()__.  This PR adds the line [__from six.moves import xrange__](https://pythonhosted.org/six/#module-six.moves) for compatibility with both Python 2 and Python 3.

Allow setting of OpMetadata for Send HLOs.

PiperOrigin-RevId: 186777369

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

PiperOrigin-RevId: 186741832

Merge pull request #17202 from yifeif/branch_186662441

Branch 186662441

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

Merge commit for internal changes

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

Fix typo (#17193)

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

Merge pull request #17058 from rajendraarora16/master

Documentation api reference badge added in Readme.md and added new header for contribution guidelines

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

Make configure script runnable from external workspace. (#17172)

To run from external workspace, you should now be able to invoke
script like the following. This will generate some TensorFlow specfic
bazel options and import them into your project's .bazelrc.

$(bazel info output_base)/external/org_tensorflow/configure.py --workspace=$(PWD)

Fix markdown nit (#17163)

Without a leading blank line, it doesn't render properly in https://www.tensorflow.org/programmers_guide/variables#variable_collections.

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