Merge commit for internal changes

Avoid recursion in ExpandDomain() as stack is not happy.

PiperOrigin-RevId: 198629366

Add kwargs support for tpu.outside_compilation

PiperOrigin-RevId: 198625799

Move RemoveInvolution optimization to optimizer stage.

PiperOrigin-RevId: 198624394

Add GCS configure ops.

PiperOrigin-RevId: 198624285

Merge pull request #19633 from yifeif/branch_198434814

Branch 198434814

Add `fill_triangular_inverse`, which flattens a triangular matrix in a way such that:
# Lower triangular matrix
x = tf.matrix_band_part(x, -1, 0)
x == fill_triangular(fill_triangular_inverse(x))
Code by srvasude@ which I'm submitting on his behalf.

PiperOrigin-RevId: 198623887

Add control dependencies to the correct graph when simplifying packing ops.

PiperOrigin-RevId: 198622727

Add `tf.contrib.distributions.bijectors.MatrixInverseTriL`: Bijector that inverts a lower-triangular matrix.

PiperOrigin-RevId: 198622553

Add include file which provides the proper std::string mapping.

PiperOrigin-RevId: 198620715

Fix CreateWrapperCPPFromBuffer properly by aselle@.

Skip errors in function optimizer if optimized graph was not modified before error happened.

Currently error can happen if function can't be instantiated as GrapplerFunctionItem.

PiperOrigin-RevId: 198595096

[tf.data] change batch dataset op test size to large to prevent timeout

PiperOrigin-RevId: 198592202

Let the swig wrapped builder to return the HloModuleProto.

PiperOrigin-RevId: 198588920

Add an option to propagate Status in GraphOptimizerStagePipelines.

PiperOrigin-RevId: 198585886

Internal change

PiperOrigin-RevId: 198582954

Disable flaky fused_rnn_cell_test

PiperOrigin-RevId: 198582181

Add "python 3 only" support for bazel build (#19443)

* Add python 3 only support for bazel build

When building tensorflow in a python3 system (i.e.,
only `/usr/bin/python3`, no `/usr/bin/python`), even
though `/usr/bin/python3` is specified during the configure
the following failure still occur:
```
SUBCOMMAND: # //tensorflow/core:version_info_gen [action 'Executing genrule //tensorflow/core:version_info_gen [for host]']
(cd /home/ubuntu/.cache/bazel/_bazel_ubuntu/ad1e09741bb4109fbc70ef8216b59ee2/execroot/org_tensorflow && \
  exec env - \
    PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin \
  /bin/bash -c 'source external/bazel_tools/tools/genrule/genrule-setup.sh; tensorflow/tools/git/gen_git_source.py --generate external/local_config_git/gen/spec.json external/local_config_git/gen/head external/local_config_git/gen/branch_ref "bazel-out/host/genfiles/tensorflow/core/util/version_info.cc" --git_tag_override=${GIT_TAG_OVERRIDE:-}')
ERROR: /home/ubuntu/tensorflow/tensorflow/core/BUILD:2015:1: Executing genrule //tensorflow/core:version_info_gen failed (Exit 127)
/usr/bin/env: 'python': No such file or directory
Target //tensorflow/tools/pip_package:build_pip_package failed to build
Use --verbose_failures to see the command lines of failed build steps.
ERROR: /home/ubuntu/tensorflow/tensorflow/tools/api/generator/BUILD:27:1 Executing genrule //tensorflow/core:version_info_gen failed (Exit 127)
INFO: Elapsed time: 391.217s, Critical Path: 46.74s
INFO: 656 processes, local.
FAILED: Build did NOT complete successfully
```

This fix adds the necessary `${PYTHON_BIN_PATH}` (configured already in `./configure`)
into related commands so that bazel build works even in python 3 only environment.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Fix build failure for api generator in python 3 only environment

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add `${PYTHON_BIN_PATH}` to cython build

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Optionally add PYTHON_BIN_PATH for non-windows only

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

KL divergence for two Dirichlet distributions.

PiperOrigin-RevId: 198573236

Adding tf.name_scope blocks to make the TensorBoard graph visualization usable.

PiperOrigin-RevId: 198569786

[tf.data] Adding a concurrency stress test for `map_and_batch`.

PiperOrigin-RevId: 198566777

internal

PiperOrigin-RevId: 198560342

beautify test output file name.

PiperOrigin-RevId: 198555383

Remove unused Make variables from tf_py_wrap_cc.

PiperOrigin-RevId: 198518885

Adds GPU kernel registration for igamma, igammac.

Switches use_gpu=True to force_gpu=True for cwise_ops_test.

PiperOrigin-RevId: 198515293

Improve error message when a missing feature name is passed as a unicode string.

PiperOrigin-RevId: 198514621

Fix GPU build on windows

PiperOrigin-RevId: 198513480

internal cleanup

PiperOrigin-RevId: 198504528

Introduced kDomain HLO instruction set isolation to bound connected sets of instructions with similar attributes (ie, sharding).
This CL simply adds the infrastructure, but leaves the wire-on to a separate CL.

PiperOrigin-RevId: 198503625

Add features to HloRunner for running while leaving buffers on the device and add option to test_utils for generating more-boring data much faster.

PiperOrigin-RevId: 198502753

Do not allow cross computation instruction lookups in HLO parser.

PiperOrigin-RevId: 198496653

Fix for Raspberry Pi wheel architecture tags (#19637)

[XLA] Minor HloSharding cleanups.

Delete dead code in HloSharding::ToString(), and add and use proper
hasher struct.

PiperOrigin-RevId: 198493972

Remove unnecessary shape registration fn from cudnn rnn ops.

The registered ones are the same as default.

PiperOrigin-RevId: 198489529

Making RPC op handle the case where cancellation manager is not initialized in OpKernelContext.

Fixes #19496

PiperOrigin-RevId: 198488860

Convenience functions to create TensorProto directly from data (std::vector).

PiperOrigin-RevId: 198486802

Fix tflite error:
TypeError: in method 'InterpreterWrapper_CreateWrapperCPPFromBuffer', argument 1 of type 'char const *'

Use absolute indexing in `fill_triangular`.

PiperOrigin-RevId: 198485926

Fixing a bug in `map_and_batch_fusion` and improving test coverage.

PiperOrigin-RevId: 198481898

Add tf.keras programmer's guide.

PiperOrigin-RevId: 198480159

Fix documented numpy equivalent of matrix_triangular_solve.

PiperOrigin-RevId: 198478933

Automated g4 rollback of changelist 198137414

PiperOrigin-RevId: 198477942

Adjust TPUEstimator timeout for worker shutdown to 60 seconds.

PiperOrigin-RevId: 198477309

Wraps the FinalOp exection with a user-friendly error mssage.

PiperOrigin-RevId: 198476911

Update setup.py with project description and development status.

PiperOrigin-RevId: 198475440

Add microbenchmarks for the executor.

PiperOrigin-RevId: 198475385

  streaming trace viewer need to filter host.

PiperOrigin-RevId: 198471853

Adding a check in eager metrics to make sure that the shapes of labels and predictions are exactly the same. The issue is that math_ops.equal would do broadcasting and so even if the shapes weren't entirely equal it'll produce an output which would be incorrect rather that reporting an error.

PiperOrigin-RevId: 198468251

In TPUEstimator.export_savedmodel(), if saving TPU metegraph fails, issue a warning instead so that user can still use the CPU metagraph.

PiperOrigin-RevId: 198458571

[tf.data] better benchmarking code in tests for measuring improvements to csv parsing

PiperOrigin-RevId: 198457501

Fix more merge error.

[TF:XLA] Implement Bucketize.

PiperOrigin-RevId: 198452289

Make the quantize_and_dequantize op use the full quantized range when possible.

PiperOrigin-RevId: 198450816

Fix merge error.

Automated g4 rollback of changelist 198421828

PiperOrigin-RevId: 198444757

Merge commit for internal changes

Increase tolerances in depthwise_conv_op_test.

PiperOrigin-RevId: 198434814

Use target machine features to determine max vectorization width for GEMM

PiperOrigin-RevId: 198434296

L2HMC trained with strongly correlated Gaussian. Simple testing
and benchmark with eager and graph mode execution.

PiperOrigin-RevId: 198433911

Extracts the 'remove split or splitv nodes' optimization into its own method.

PiperOrigin-RevId: 198432976

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

PiperOrigin-RevId: 198432337

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

Add note to bijector_impl.py explaining that `log_det_jacobian` is `log(*abs*(det(Jacobian)))`.

PiperOrigin-RevId: 198428995

[XLA] Clarify that options passed to LocalService override legacy_flags values.

PiperOrigin-RevId: 198426696

Update ops-related pbtxt files.

PiperOrigin-RevId: 198426617

Tile on the M dimension in GEBP.

After this change the inner reduction loop in GEBP multiplies a tile from the
LHS and a tile from the RHS to get a result tile.

PiperOrigin-RevId: 198425769

  [tpu:profiler] Add a new field to the statistics collected at each step.

PiperOrigin-RevId: 198425588

Extracts the 'remove shuffle or transpose node' optimization into its own method.

PiperOrigin-RevId: 198425354

Add security notices for recently discovered and patched vulnerabilities.

PiperOrigin-RevId: 198422244

[TF/XLA] Add validation to find ops incompatible with the given device type at the beginning of graph compilation.

PiperOrigin-RevId: 198421828

  Support NHWC_VECT_W in MakeShapeFromFormat.

PiperOrigin-RevId: 198421617

Update python_configure.bzl (#19614)

Add BAZEL_SH environment variable as a dependency of python_configure repository rule

Add AnonymousIteratorHandleOp for non-shared Iterator resources

Fixes Iterator cleanup when executing eagerly. DestroyResourceOp will now remove the last reference from the Iterator resource when it runs (after the last Python reference to an EagerIterator is removed).

Previously EagerIterator used IteratorHandleOp to create resource handles, which used one kernel per (unique) shared name since the shared name was an attribute. These kernels each held a reference to their resource, which kept it alive indefinitely.

Fixes #19499.

PiperOrigin-RevId: 198417997

Clarify argument types and relationships in docstrings of statistical_testing.py.

PiperOrigin-RevId: 198414898

Add mirror for nasm (#19349)

Eliminate self.test_session in favor of self.evaluate in statistical_testing_test.py.

PiperOrigin-RevId: 198410306

Clean up: handle the hidden additional clause on for loops in a way that's clearer about what it does.

PiperOrigin-RevId: 198409797

Merge pull request #19574 from braincodercn/patch-1

Fix build error

Clean up: remove useless super delegation.

PiperOrigin-RevId: 198405670

 Fixed memory leak with py_func (#18292) (#19085)

* Fixing memory leak with py_func (#18292)

* Fixed memory leak with py_func (#18292)

Allow assignment to subscripts in static analysis.

Move the handling of syntactic unpackings to a generic helper function since the pattern is used in multiple places. Update the type info analyzer to correctly process function arguments.

PiperOrigin-RevId: 198401368

Generalize assert_true_mean_equal and assert_true_mean_equal_two_sample to assert_true_mean_in_interval.

PiperOrigin-RevId: 198400265

Re-apply #18192.

PiperOrigin-RevId: 198358055

Fix an incorrect precondition check in IndexedArrayAnalysis

PiperOrigin-RevId: 198354001

Pass HloOpcode instead of HloInstruction; NFC

Minor code cleanup change.

PiperOrigin-RevId: 198351045

Make IndexedArrayAnalysis behave well around StatusOr

PiperOrigin-RevId: 198348355

python_configure.bzl: Find bash binary path through BAZEL_SH env var. (#19598)

* python_configure.bzl: Find bash binary path through BAZEL_SH env var.

This helps avoid invoking the wrong bash binary when "Bash on Ubuntu on Windows"
is installed.

Fixed https://github.com/tensorflow/tensorflow/issues/11735

* Readability modifications.

Windows: fail gracefully when undname.exe is not found (#19600)

* Windows: fail gracefully when undname.exe is not found

* Edit error message.

Merge pull request #19539 from AD-530/master

Add missing deps for simd_armv8a in jpeg.BUILD

Merge pull request #19577 from tensorflow/terrytangyuan-patch-1

Fixed typo in exporter.py

Merge pull request #19580 from ManHyuk/fix_typo

Fix typo

Adding tf.data optimization for rewriting `map(...).batch(...)` to `map_and_batch(...)`.

PiperOrigin-RevId: 198310806

Sort variables in C++ instead of Python.

PiperOrigin-RevId: 198298103

Relax compatibility checks for Protocol Buffer classes, to not rely on the exact layout of the generated classes. The definition of the message is checked instead.

PiperOrigin-RevId: 198292780

[XLA] Don't display metadata inline in HLO graph dump.

We only want to display it in the tooltip.

PiperOrigin-RevId: 198235268

TPUEstimator.export_savedmodel() saves a SavedModel with both TPU and CPU graphs.

PiperOrigin-RevId: 198229550

fix typo

Merge branch 'master' into fix_typo

Fixed typo in exporter.py

Fix build error

Fix build error: 'function' in namespace 'std' does not name a template type