Fix grappler to use CudaGpuId instead of TfGpuId to query device states.

PiperOrigin-RevId: 185233116

Add a test that exhaustively checks Log/Exp/Tanh

PiperOrigin-RevId: 185216684

import_graph_def: support "absolute" names with the C API enabled.

Passing a name with a trailing '/' to import_graph_def causes that
name to be used as-is (i.e. it is not appended to the existing name
scope and not de-duped with any existing name scopes. This is in order
to re-use an existing name scope). This didn't work with the C API
enabled because it was set to always have the C API uniquify the
prefix.

The fix is to not uniquify the prefix, since calling name_scope in
import_graph_def already has the logic to uniquify the prefix if
necessary. I'm not sure why I thought we needed the C API to do this
to being with.

In addition, this changes the graph_constructor.cc logic to uniquify
names if the prefix cannot be guaranteed unique (see the new test case
in graph_constructor_test.cc for why/when this is necessary).

PiperOrigin-RevId: 185215326

[XLA:Tool] Make Hlo parser report the location of the already defined instrucion/computation.

PiperOrigin-RevId: 185213461

[XLA] Implement GeneralDot semantics in HloEvaluator.

Also:
- add a general matmul test, enable interpreter to run dot_operation_test.
- remove now redundant/obselete CHECKS for HandleDot in HloEvaluator.
- improve documentation for DotGeneral a bit.
PiperOrigin-RevId: 185211512

Use x*x instead of x^2 to calculate square in huber loss implementation. The reason is d(x^y)/dy = x^y * log(x) and when x is zero, it becomes NaN. Even if y is constant, the check op would still report failure.

PiperOrigin-RevId: 185208180

Disable flaky spinn_test

PiperOrigin-RevId: 185207742

Add pylint check for W0622 redefined-builtin in ci_sanity.sh and fix existing pylint errors.

PiperOrigin-RevId: 185206494

Add delegate API to tflite.

- Context gets GetNodes, num_nodes and PartitionNodesIntoSubgraphs.
- TfLiteDelegate provides one function that need be implemented
- Delegates choose nodes and those nodes are all compacted into
  a new macro kernel.

PiperOrigin-RevId: 185204338

[XLA] Use a real priority queue in list scheduling

PiperOrigin-RevId: 185201882

Recursively creating directories in CreateSummaryFileWriter

PiperOrigin-RevId: 185199219

[tf.data] Remove deprecated `tf.contrib.data.Dataset` class.

This change removes the following class:
* `tf.contrib.data.Dataset`.

IF THIS BREAKS YOU: Replace `tf.contrib.data.Dataset` with `tf.data.Dataset`
when constructing a dataset. Note that you may have to modify downstream
transformations to use the core API. See "tensorflow/contrib/data/README.md" for
details of how to update your code to use the core API.

PiperOrigin-RevId: 185197005

Use TFLite CocoaPod in iOS Simple Demo app

PiperOrigin-RevId: 185196984

Remove header dependence on cuda_config.h to fix opensource custom op support.
Fixes #14454
Fixes #12860

PiperOrigin-RevId: 185194924

Fix read_variable_op GPU test

PiperOrigin-RevId: 185194768

Enabled XLA for TF C API.

Summary of changes:

1. Set MarkForCompilationPassFlags::tf_xla_cpu_global_jit default to true in
C_API unit test env when XLA-execute is intended. Together with setting session
config config.graph_options.optimizer_options.global_jit_level to > 0, this
turns on XLA for the entire graph (eligible nodes only, with _Arg and _RetVal
nodes excluded).

We decided against defaulting MarkForCompilationPassFlags::tf_xla_cpu_global_jit
to true, due to performance concerns with the single-threaded nature of the XLA
CPU backend (see
https://www.tensorflow.org/performance/xla/jit#turning_on_jit_compilation).

2. In FindCompilationCandidates() during MarkForCompilationPass, skip compiling
any '_Arg'-typed nodes. This is necessary to avoid hitting a "Invalid argument
number" error during MarkForCompilationPass.

3. Extended C API based build rules to link in XLA libraries, and added unit
test "CAPI.Session_Min_XLA_CPU".

Also added some misc improvements and debugging aids.

PiperOrigin-RevId: 185193314

Make import_graph_def add default attr values with the C API enabled.

It turns out that the original Python code modifies the graph_def argument
to add default attr values. I'm not sure if the behavior is covered by
our API guarantees since it's not documented, but let's keep the behavior
consistent for now.

PiperOrigin-RevId: 185193037

TEST:  test of kernel_results added to hmc_test.py
PiperOrigin-RevId: 185191871

Removes const qualifier from ListWorkers

PiperOrigin-RevId: 185190346

Update llvm revision to r324720. This is needed because r324700 introduces an Orc API change.

PiperOrigin-RevId: 185185088

Fixes issue with tfe.make_template when function objects don't have names

PiperOrigin-RevId: 185181846

Split gpu_id.h and GpuIdManager out from build target //tensorflow/core:gpu_runtime, to reduce the size of dependencies, so when other lightweight libraries like grappler utils needs the TfToCudaGpuId translation function it doesn't need to depend on things like stream executor and cuda libraries.

PiperOrigin-RevId: 185175757

[XLA] Add a test for reduce window with large minor dimension.

PiperOrigin-RevId: 185175593

Add dynamic_rnn support for CloudTPU Magenta RNN

PiperOrigin-RevId: 185169370

Simplify and extend the management of input-conditional losses and updates.

Instead of keeping track of dependencies manually, we rely on the TF graph structure to find dependencies. The resulting implementation is cleaner and more robust.

This does not change any existing behavior. It extends the current behavior by allowing `get_updates_for(inputs)` and `get_losses_for(inputs)` to be called from *any* tensors upstream of the layer, not just the immediate layer's inputs.

PiperOrigin-RevId: 185168680

Fix lint warnings. NFC

PiperOrigin-RevId: 185167035

Add export calls for protos.

PiperOrigin-RevId: 185166764

Remove the unneeded "tf" argument for utility functions.

PiperOrigin-RevId: 185166507

Fix some doc strings that were accidentally removed.

PiperOrigin-RevId: 185164375

  regression_head accepts link_fn as parameter, so the use can use self defined link fn

PiperOrigin-RevId: 185161896

TFTS: Better handling of exogenous features

Adds (dummy) exogenous features to the LSTM model-building example, and adds some small methods needed to support that (fetching the shape of embedded exogenous features).

Also makes it more automatic to export a SavedModel with exogenous features (placeholder shapes will be inferred from the given FeatureColumns), which makes the LSTM example friendlier.

PiperOrigin-RevId: 185157085

TFTS: Add exporting to SavedModel to the LSTM example

Was previously not using a state manager, which among other issues prevented
exporting.

Fixes #16590.

PiperOrigin-RevId: 185150900

Update for LLVM API change r324700

PiperOrigin-RevId: 185149198

Comment update.

PiperOrigin-RevId: 185141668

Only convert format if input is of layout-agnostic type.

PiperOrigin-RevId: 185103227

Add tf_export decorators back to gen_*_ops.py files.

PiperOrigin-RevId: 185075262

Extended the Halton sequences to support randomization. Implemented the randomization scheme described in arXiv:1706.02808.

PiperOrigin-RevId: 185073515

Automated g4 rollback of changelist 185006374

PiperOrigin-RevId: 185072479

Fix bug in checkpointing DT_VARIANT tensors.
Also add an integration test that runs and checkpoints multiple input pipelines in one graph.

PiperOrigin-RevId: 185071397

Improve the flow to build TFLite iOS demo app

PiperOrigin-RevId: 185069356

Add our own "global step" called quantization_step to avoid needing to reset the global step before running the graph rewrite.

PiperOrigin-RevId: 185068631

Allow C64 infeeds.

PiperOrigin-RevId: 185068327

Make SavedModelTest.testStripDefaultAttrsInconsistentConsumerDefaults work with C API.

The test originally altered the Python version of the op registry,
which is not reflected in the C API. This changes the test to alter
the serialized node def instead of the op def, and renames the test to
testInconsistentConsumerDefaultAttrs.

PiperOrigin-RevId: 185067838

Changes behavior of optimizer.minimize(loss) when loss is a function.

Now supports both eager and graph mode, and loss is always a callable
with no arguments. Supports variables and non-variables in var_list.

PiperOrigin-RevId: 185063691

Adding tf_export decorators/calls to TensorFlow functions and constants.

PiperOrigin-RevId: 185062323

Update fake quant op to support bitwidths in the range 2 to 16, from 2 to 8.

PiperOrigin-RevId: 185061307

Object based saving prototype: create ResourceVariables directly by default.

This avoids variable reuse errors when building a graph. Where necessary for compatibility, we can still use get_variable.

PiperOrigin-RevId: 185060891

Add memory usage report to cost analyzer tool; run all default optimizations.

PiperOrigin-RevId: 185058999

Add more vlogging to op level estimator.

PiperOrigin-RevId: 185058989

[TF contrib RNN] Expose some rnn classes and functionality in contrib.

PiperOrigin-RevId: 185057994

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

Add a _TensorProcessor for computing gradients but not applying them.

PiperOrigin-RevId: 185056764

Updated API for quantize_graph: Programmable quant_delay to determine start of quantization in training.
Remove delay_requested parameter in _InsertQuantOp, presence or absence of quant_delay
is directly inferred from value of quant_delay. Any positive value causes insertion of
delayed quantization into the graph.

Experimental APIs for quantization with more programmability, including bitwidths for weights and activations.

PiperOrigin-RevId: 185056081

Adding tf_export decorators/calls to TensorFlow functions and constants.

PiperOrigin-RevId: 185055497

Update ops-related pbtxt files.

PiperOrigin-RevId: 185052000

Updates to fastpath execution code

- Add default value shape handling
- Add default value func handling (including adding a new function for this in
  the eager C API)
- Update callback execution interface to match record_gradient call

PiperOrigin-RevId: 185051447

Object-based saving prototype: allow only named edges

Simplifies the logic a bit, and removes one source of potential checkpoint
incompatibilities.

Classes like Sequential which need anonymous references to other objects may
generate string names containing numbers.

PiperOrigin-RevId: 185050714

Don't fail if control dependency is on an input of the function.

PiperOrigin-RevId: 185049319

Adding tf_export decorators/calls to TensorFlow functions and constants.

PiperOrigin-RevId: 185044705

Prototype object-based save/restore syntax sugar

- Overrides __setattr__ to allow implicit dependencies
- Supports any valid Python 2 identifier as a Checkpointable dependency name (was messing with underscore prefixes)

PiperOrigin-RevId: 185044022

Plumbs a variable from the batching function decorator to the queue depth control option in the underlying BatchingQueue. It defaults to 10 if not set, which was the original default value in the BatchingQueue options struct.

PiperOrigin-RevId: 185043844

Internal functional _If and _While ops.

PiperOrigin-RevId: 185042663

C API: Fixes #7394

Ideally, when TF_NewTensor is provided with invalid arguments it would provide
a detailed error message. However, for now, to keep the existing API, signal
failure by returning nullptr.

PiperOrigin-RevId: 185040858

[TF:XLA] Handle fusion instructions in HloInstruction equality comparison.

Two fusion instructions are equal iff the fusion kind matches and the fused subcomputations are structurally equal.

PiperOrigin-RevId: 185038129

Bug fix in buffer assignment for colocated buffers

PiperOrigin-RevId: 185034095

[XLA] Improve comment about why we can't unroll loops with static trip count of 1 if they contain side-effecting ops.

PiperOrigin-RevId: 185034032

Add effective_sample_size to tf.contrib.bayesflow.mcmc_diagnostics.
Also, start dealing with list args in a more regular manner.

PiperOrigin-RevId: 185032115

Expose python:platform build target.

PiperOrigin-RevId: 185030536

Adding tf_export decorators/calls to TensorFlow functions and constants.

PiperOrigin-RevId: 185026527

Changing ".lite" to ".tflite" in documents.

PiperOrigin-RevId: 185026484

Update llvm revision to r324405, which introduces an Orc API change.

PiperOrigin-RevId: 185025122

[tf.data] Remove deprecated `tf.contrib.data.Iterator` alias.

This change removes the following class:
* `tf.contrib.data.Iterator`.

IF THIS BREAKS YOU: Replace `tf.contrib.data.Iterator` with `tf.data.Iterator`
when explicitly constructing an iterator. The API for the resulting object is
identical.

PiperOrigin-RevId: 185024771

Preserving order when removing nodes.

PiperOrigin-RevId: 185023366

Update the wrapper gen code to call out to the fastpath function.

PiperOrigin-RevId: 185016979

Fix minor bug with tf.keras model calling (related to mask caching).

PiperOrigin-RevId: 185016910

[tf.data] Remove deprecated reader dataset classes in `tf.contrib.data`.

This change removes the following classes:
* `tf.contrib.data.FixedLengthRecordDataset`
* `tf.contrib.data.TextLineDataset`
* `tf.contrib.data.TFRecordDataset`

IF THIS BREAKS YOU: Replace `tf.contrib.data` with `tf.data` when constructing a
`FixedLengthRecordDataset`, `TextLineDataset`, or `TFRecordDataset`. Note that
you may have to modify downstream transformations to use the core API. See
"tensorflow/contrib/data/README.md" for details of how to update your code to use
the core API.

PiperOrigin-RevId: 185016587

Update for LLVM API change r324405

PiperOrigin-RevId: 185016276

Remove cudnn_type parameter from a few template member functions in class
CudnnSupport.

The cudnn_type value in those functions is implied by the device memory
type DeviceMemory<T> and is a compile-time constant.

PiperOrigin-RevId: 185013909

Make MklCpuAllocator a VisitableAllocator, instead of just an Allocator.
This allows it to work more efficiently with RDMA networking.

PiperOrigin-RevId: 185013628

Make quantization rewrites happen in place.

If no graph is provided, then the default graph is used.

PiperOrigin-RevId: 185007107

Enable algebraic optimizations for operations with neutral and absorbing elements by default - not only when feeds are absent or in aggressive mode.

PiperOrigin-RevId: 185006374

More updates:
 * add a verbose flag, useful for debugging
 * fix the canonicalization of old-style print statement
 * fix broken py_func wrapper
 * expand the conditional statement to return a dummy value if we cannot fine any return values, and call without an assignment so that the side effect guards catch it instead
 * streamline the converter tests a bit more
 * avoid aliasing "tf" and "self" in the side effect guards
 * improve the namer to generate shorter names

PiperOrigin-RevId: 185002802

Emitting type-dependent cond and while_loop code.

PiperOrigin-RevId: 184993166

Patch cub on download to fix compilation error with clang.

The same patch was sent via PR to cub upstream:
https://github.com/NVlabs/cub/pull/125

PiperOrigin-RevId: 184975304

Automated g4 rollback of changelist 184303789

PiperOrigin-RevId: 184970903

Update ops-related pbtxt files.

PiperOrigin-RevId: 184947099

Switch to using ResourceVariable in Keras for full Eager-mode compatibility.

PiperOrigin-RevId: 184945626

Fixing the anchor link for the Math ops guide on Segmentation (it's a capital S but our docs link to a lowercase s).

PiperOrigin-RevId: 184942994

Temporarily weaken Identity pruning in model_pruner while investigating test failure of //robotics/learning/sensor_predict:utils_multi_sensor_rnn_test.

PiperOrigin-RevId: 184942554

Move TPU doc to "using_tpu". Add short titles to some docs.

PiperOrigin-RevId: 184941101

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

[tf.data] Move C++ code backing `tf.contrib.data.ignore_errors()` to contrib.

This change moves the `OpKernel` and `DatasetBase` implementations to
"tensorflow/contrib/data/kernels", where they are packaged as a custom
op library. This demonstrates (and enforces by continuous integration)
the ability to build a C++ Dataset implementation in a custom op
library.

Other contrib Dataset implementations will move in subsequent changes.

PiperOrigin-RevId: 184938885

[XLA:CPU] Fix tfcompile's use of freeze_graph.py

Not sure if I am holding it wrong, but I could not make the existing
FLAGS.check_version work.

PiperOrigin-RevId: 184935374

Reduce the number of concats to avoid test timeout.

PiperOrigin-RevId: 184929151

Remove note about accumulator variables, since those are not added to TRAINABLE_VARIABLES.

PiperOrigin-RevId: 184922273

Add operation to forward log messages from remote workers to a local system.

PiperOrigin-RevId: 184921657

[XLA:CPU] Fix test case for vectorized Exp and Tanh to actually vectorize

I just noticed that the test case for ArrayElementwiseOpTest::ExpF32sVector and
possibly for ArrayElementwiseOpTest::ExpF32sVector does not actually vectorize
the intrinsic calls.  This is most likely a very recent regression because I
remember fixing at least one issue in the emitter demonstrated by the test.
Despite that I think the current approach is better since we have unit tests
that check that we at least vectorize the vector-of-F32's case.

PiperOrigin-RevId: 184918373

Update ops-related pbtxt files.

PiperOrigin-RevId: 184916250

Disable internally failing array_ops test.

PiperOrigin-RevId: 184915141

Remove tf.contrib.ndlstm as it is not maintained and barely used.

Users can find an external implementation by the original author at:

https://github.com/tmbarchive/tfndlstm

PiperOrigin-RevId: 184914822