Merge pull request #18972 from bstriner/fix_mklroot

Use MKLROOT

Merge pull request #18987 from eliben/update-xla-readme

Update xla README

Merge branch 'update-xla-readme' of github.com:eliben/tensorflow into update-xla-readme

Add XLA logo and beef up the README

Reorder

Center-align the logo image and set size

Add XLA logo and beef up the README

Merge pull request #18983 from caisq/branch_194768567

Branch 194768567

Merge pull request #18882 from nfelt/tb-nightly-dep

Update tb-nightly dep to >= 1.9.0a0, < 1.10.0a0

Merge pull request #18825 from saginadir/patch-1

Update README.md

Merge pull request #18965 from rmanyari/feature_columns_fix

add missing equality

autograph: Update README (#18981)

* Update README.md

fix typo (#18957)

Merge commit for internal changes

Cleaning up tracing code.

PiperOrigin-RevId: 194768567

Use MKLROOT

Keras: Supply `maximum_iterations` to the TF backend when possible.
PiperOrigin-RevId: 194723199

Add support for a clean checkpoint and shutdown in response to a termination notice.

PiperOrigin-RevId: 194722985

add missing equality

Added del_hparam(), the counter part of add_hparam.

PiperOrigin-RevId: 194711291

Update the Swift for TensorFlow community page.

PiperOrigin-RevId: 194687897

[TF:XLA]
- Require a module config when creating an HloModule.
- All tests using HloTestBase create a module using CreateNewModule.

PiperOrigin-RevId: 194684585

Internally rewrite RevBlock to use @custom_gradient

PiperOrigin-RevId: 194679657

Pass dtype to constructor in LSTMCell (#18178)

* Use float32 in case the dtype is not set in the constructor

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test case for 16228.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test case where dype is passed explicitly.

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

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Replace strings to objects to address review feedback.

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

Fix functions in CRF when sequence_lengths contains zero (#18487)

* Fix computation of crf_log_norm when sequence length is zero

* fix _single_seq_fn in crf when sequence_lengths contain zero

Emphasis any `Estimator` (#18793)

Add uint32 and uint64 support with tf.train.batch (#18805)

* Add uint32 and uint64 support with tf.train.batch

This fix tries to address the issue raised in 18586
to have uint32 and uint64 support with tf.train.batch.

This fix add uint32 and uint64 to `CopyElementToSlice`
for the support.

This fix fixes 18586.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test case for uint32 with tf.train.batch

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add uint64 test case

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

Fix typo in CMakeLists.txt (#18833)

Fix link to original LSTM paper (#18876)

Java: Release 1.8.0

PiperOrigin-RevId: 194663800

[tf.data] Use core::ScopedUnref to avoid resource leakage.

If for whatever reason iterator_resource->set_iterator did not return Status::OK(), we would leak a reference on the iterator_resource. With this change, we won't leak the resource.

PiperOrigin-RevId: 194662412

Allow not specifying eval_spec when evaluation is not necessarily run.

PiperOrigin-RevId: 194661814

This is Part 1 of Swift<->TF sends/recvs: support sending tensors from TF to
Swift via direct session.

The changes are:

1. Added an experimental TF C API TF_DequeueNamedTensor() to consume the queued
tensors from a dequeue op. One use case is for the Swift host program to consume
tensors sent by TF, where the queue is a Fifo queue managed by TF.

Enqueuing tensors are done by running an enqueue op in a graph. The queued
tensors are not persisted, and will be lost if the process/machine dies. The
queue has a bounded capacity, to prevent producer from being unboundedly ahead
of consumer.

while caller of TF_DequeueNamedTensor() could have run the Fifo dequeue op
directly, the extra level of indirection provided by this API allows us to more
easily switch the queuing impl to another mechanism. If and once we stabilize on
the Fifo queue based impl, we can remove this API.

2. Added a new S4TF runtime API _TFCReceiveTensorHandle() that receives a tensor
  via TF_DequeueNamedTensor().

3. To support tensor receives in host program, taught PartitionCloner in
  TFPartition to insert SIL code to call _TFCReceiveTensorHandle().

4. To support tensor sends in accelerator program, taught TFGraphLowering in
  generate QueueEnqueueV2 nodes in the TF graphs, with appropriate control
  dependence to make sure these nodes get executed.

a) The enqueue produces no output tensor, and is executed only for its side
effect. To ensure it is executed properly, control dependence is wired up. The
general design is: before a TF_Function (can be a top level function or the body
function of a while op) produces an output tensor OT, make OT control dependent
on the enqueue op, so that enqueue gets run before the function returns.

b) If tensor send occurs in a while loop body, the body logic currently gets
lowered in 3 places: the while op cond function, the while op body function, and
the ops at the same level as the while op itself (for running the last loop
iteration). In this case, the correct TFGraph lowering is to run the enqueue in
the last 2 out of the 3 places above.

After this CL, the dual versions of the above (dequeuing via an op, and
enqueuing via C API) will be added.

PiperOrigin-RevId: 194658511

Removing hidden_ops.txt file.

PiperOrigin-RevId: 194637892

Fix kernel creation bug, due to constant folding always use CPU.

PiperOrigin-RevId: 194636076

Merge pull request #18944 from tensorflow/martinwicke-patch-3

Fix docs rendering in placeholder docs page.

Fix docs rendering in placeholder docs page.

Add test case on compiling dense layer node with XLA.

PiperOrigin-RevId: 194634563

Properly export recurrent in contrib.

The following symbols are available:
- tf.contrib.recurrent.bidirectional_functional_rnn
- tf.contrib.recurrent.functional_rnn
- tf.contrib.recurrent.Recurrent

PiperOrigin-RevId: 194632138

HLO profiling for tfcompile.

This CL extends the --xla_hlo_profile knob to tfcompile.  tf_library rules can
now set enable_xla_hlo_profiling to True to:

 - Have the generated code update per-HLO profile counters as it executes.
 - Have tfcompile generate and serialize an instance HloProfilePrinterData with
   a compiled model that can be used to pretty-print the collected profile
   counters.

PiperOrigin-RevId: 194627272

  Add internal uint b stats to TfOpStats.

PiperOrigin-RevId: 194625155

Split up ElementaIrEmitter::MakeElementGenerator into smaller functions; NFC

PiperOrigin-RevId: 194622198

[XLA] Fix bug in ShapeUtil::StripDegenerateDimensions

PiperOrigin-RevId: 194621163

Make RetryingFileSystem a template.

PiperOrigin-RevId: 194614877

Revised roadmap (#18939)

Revised roadmap

Minor fix to SDCAOptimizer documentation.

PiperOrigin-RevId: 194609850

[TF:XLA:INTERPRETER] implement bfloat16 comparisons

PiperOrigin-RevId: 194608854

Fix broken ElementWiseFusionTest.

PiperOrigin-RevId: 194602336

Minor eager service proto clarification.

PiperOrigin-RevId: 194596337

[XLA:CPU] Implement fusion for the Gather HLO

PiperOrigin-RevId: 194594759

If two identical functions are given different grad func,
they should be named differently. Otherwise, tf.gradients
gets confused.

PiperOrigin-RevId: 194593519

Remove whitespaces from tags in saved_model_cli.

This currently causes tags mismatch because a leading whitespace is added
within the saved_model_cli when doing ', '.join(tag_set).

PiperOrigin-RevId: 194590154

[TF:XLA] Bump open source llvm revision to r330950

PiperOrigin-RevId: 194588403

Roll forward the custom optimizers change (#18742)

Remove scope name from bfloat16

PiperOrigin-RevId: 194580957

eager: Improve error message when GradientTape is used incorrectly.

PiperOrigin-RevId: 194580654

Replace GrapplerFunctionItem input with a constant.

PiperOrigin-RevId: 194579253

Merge pull request #18937 from lamberta/fix-typo

Docs: fix typo

Merge pull request #18935 from drpngx/branch_194551042

Branch 194551042

Docs: fix typo

Internally rewrite @recompute_grad to use @custom_gradient

PiperOrigin-RevId: 194571125

Add DeviceSet to Cluster (#18838)

* Add DeviceSet to Cluster so we can access memory allocators during grappler optimizations

* Fix review comments

* Add missing dependency to :virtual_cluster and fix clang format.

Fix merge conflict manual merge error.

Merge commit for internal changes

Merge pull request #18727 from freedomtan/label_image_tflite_profiling_new

[tflite] add profiling to label_image for tflite

Added string conversion operator to tensorflow::StringPiece.
Marked ToString method as deprecated.

This will allow tensorflow::StringPiece to be replaced with absl::string_view (once the deprecated method is removed) as absl::string_view does not contain the ToString method.

PiperOrigin-RevId: 194551042

Fix bug in @custom_gradient in Eager mode with numpy inputs

PiperOrigin-RevId: 194538828

[TF:XLA] Bump open source llvm revision to r330926

PiperOrigin-RevId: 194530610

Automated g4 rollback of changelist 194306629

PiperOrigin-RevId: 194507274

Reenable factorization_ops_test on ASAN after adding shard_count = 4. Tests now
finish with these stats: "max = 150.6s, min = 27.4s, avg = 66.3s, dev = 19.5s" over 1000 runs and this runtime distribution should be fairly safe for deadline of 300s.

PiperOrigin-RevId: 194500204

Implement floor operator

PiperOrigin-RevId: 194490433

Automated g4 rollback of changelist 194442428

PiperOrigin-RevId: 194485227

Free scratch memory in ~BaseGPUDevice.

PiperOrigin-RevId: 194483351

Implements linear no-offset (aka symmetric) quantizer.

PiperOrigin-RevId: 194482547

Split out SaveableObjects into their own file

Pulls a couple build rules out of tensorflow/python:training. I'd like to use a SaveableObject in :checkpointable (for saving some Python state by default), which means the file with SaveableObject has to be essientially dependency-free.

PiperOrigin-RevId: 194473987

More informative error message when loading a graph_def which uses unknown ops.
Fixes #17014

PiperOrigin-RevId: 194472083

Fix some flakiness in test.

PiperOrigin-RevId: 194470125

Internal change.

PiperOrigin-RevId: 194468535

Merge pull request #18846 from yongtang/04252018-FloorDiv-int8

Enable int8 support for FloorDiv

Handle variations in scoping of batch norms for correct unfused batch norm folding.

PiperOrigin-RevId: 194465704

Remove the inter-op thread pool

Forgot about this in cl/194299356.  However, when I checked cl/194299356, I
found that we actually (incorrectly?) used the *intra* op thread pool in the
parallel CPU executable?  Does that mean the inter op thread pool was always
unused?

PiperOrigin-RevId: 194464734

Merge pull request #18881 from ManHyuk/fix_typo

fix typo

Merge pull request #18907 from yongtang/18363-mpi

Fix build error with MPI support

[tf.data] Adds support for adding scalar value to `StatsAggregator`.

PiperOrigin-RevId: 194463407

Merge pull request #18896 from KikaTech/fix_lite_topk

Fix some issues in official tf.nn.topk() in lite

[tf.data] Changes description for `bytes_produced_stats` and `latency_stats` in accordance with the breaking changes in cl/193432590.

PiperOrigin-RevId: 194461964

- Default values of cov and inv variables are now 0.  Zero-debiasing (as in Adam) is used for the cov matrices.  Note this this requires that cov variables, then inv variables, are all updated before the first training update is made.  All examples have been modified to do this. NOTE: you *may* have to increase the damping value you use at the start of optimization after this change (or throughout, if you are using a constant value).
- Changed the initial default approximation used for generic registrations to "diagonal"
- Convenience properties for ops and thunks have all been removed, along with "make_ops_and_vars". User should only interface with "make_vars_and_create_op_thunks" (or maybe "create_ops_and_vars_thunks").

PiperOrigin-RevId: 194461623

Add support for variables in tf.custom_gradient

PiperOrigin-RevId: 194460752

Disable triangular_solve_test on ASAN due to flaky time outs.

PiperOrigin-RevId: 194460641

Edit tensorflow.org/community/swift page.

PiperOrigin-RevId: 194459754

Adds optimization to convert division of sqrt to multiplication of rsqrt

PiperOrigin-RevId: 194459152

Disable densenet_test on MSAN due to flaky time outs.

PiperOrigin-RevId: 194458270

Merge branch 'master' into label_image_tflite_profiling_new

Remove unnecessary TF_NEED_GCP from build scripts.

PiperOrigin-RevId: 194448612

Fix cmake build issues with GPU on Linux (#18775)

* Update include file for cmake

While attempting to build tensorflow with cmake for gpu the following
build error happens:
```
/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.h:26:45: fatal error: cuda/extras/CUPTI/include/cupti.h: No such file or directory
compilation terminated.
CMakeFiles/tf_core_cpu.dir/build.make:3302: recipe for target 'CMakeFiles/tf_core_cpu.dir/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.cc.o' failed
make[2]: *** [CMakeFiles/tf_core_cpu.dir/home/ubuntu/tensorflow/tensorflow/core/platform/default/gpu/cupti_wrapper.cc.o] Error 1
CMakeFiles/Makefile2:2402: recipe for target 'CMakeFiles/tf_core_cpu.dir/all' failed
make[1]: *** [CMakeFiles/tf_core_cpu.dir/all] Error 2
Makefile:127: recipe for target 'all' failed
make: *** [all] Error 2
```
This fix tries to fix the build error above for cmake.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add xla dependency to stream_executor

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Remove nccl files temporarily

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add gomp library to libcusolver.so

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

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

Merge pull request #17602 from joeyearsley/patch-1

Update fold_old_batch_norms.cc to accommodate 'NCHW' format.

Adding some slightly more exhaustive strided_slice test parameters.

PiperOrigin-RevId: 194446000

[TF:XLA] Add INTEL MKL_DNN Conv2d method to XLA/CPU backend

The INTEL MKL_DNN provides 32-bit Conv2d method. With INTEL_MKL flag set,
XLA backend emits runtime call to MKL_DNN Conv2d instead of Eigen.

PiperOrigin-RevId: 194445212

Run 2 passes of rewrites by default

PiperOrigin-RevId: 194443770

Fix build error with MPI support

This fix tries to fix the issue raised in 18363 where
the bazel build with MPI support fails as a header is missing
in the include.

This fix fixes the issue. The fix is verified locally
with MPI+CUDA on Ubuntu 16.04.

This fix fixes 18363.

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