commit e826feabce35024cd00ce27d76db6196f4628970
author A. Unique TensorFlower <gardener@tensorflow.org> Tue Mar 08 15:55:15 2022 -0800
committer Mihai Maruseac <mihaimaruseac@google.com> Tue Mar 08 16:10:18 2022 -0800
tree c1f3f41ee6ca3c32612ffc3a9a2cd85872c83b69
parent c3ecdfa4f186d43c061a508d5f891e082ceb6640

Merged commit includes the following changes:
433332091  by A. Unique TensorFlower<gardener@tensorflow.org>:

    [tf.data] Ramp up `initial_parallelism_value` experiment to 100%

--
433332037  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Return conversion error if `StatefulPartitionedCall` or `ParitionedCall` is in the result model.

    Neither `StatefulPartitionedCall` nor `ParitionedCall` can be executed by TF Lite, so we should raise a conversion error if such an op appears in the result model.

--
433331676  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Add ContainerMap forward type inference function

    Helper for ops with semantics of applying some transformation to the elements of a container, e.g. changing the batch size of a dataset.

    `<t>[PRODUCT[T1, ..., Tn]] -> <t>[PRODUCT[U1, ..., Un]]`

    where Ui is obtained by applying a map T -> U. Both <t> and the "map"
    function are parameterized by this factory.

    Note that when the "map" function is the identity function, this becomes equivalent to ReplicateInput.

--
433331338  by A. Unique TensorFlower<gardener@tensorflow.org>:

    [NNAPI] Allow dequantizing fp16 scalar input

--
433327321  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Go: Update generated wrapper functions for TensorFlow ops.

--
433324664  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Update tf.py_function and tf.numpy_function docstring.

--
433324643  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Implement try_generalizing_trace_type for optimize_tracing_count

--
433323662  by A. Unique TensorFlower<gardener@tensorflow.org>:

    [XNNPACK] Rename concatenation tests to include number of ouputs in name

    Also introduce a helper to create shapes that match in all dimensions except the axis.

--
433316824  by A. Unique TensorFlower<gardener@tensorflow.org>:

    [XNNPACK] Change ConcatenationTester to be able to take multiple input shapes

    This will allow us to test concatenation with more than 2 inputs.

--
433316507  by A. Unique TensorFlower<gardener@tensorflow.org>:

    PR #53796: Add linux_aarch64 target to AOT.

    Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/53796

    The lack of a target for `linux_aarch64` on AOT breaks tests under `//tensorflow/python/tools`.
    Copybara import of the project:

    --
    e50a9a79b73b1a5b5ef258fc28bde3d6545043e1 by Everton Constantino <everton.constantino@linaro.org>:

    Add linux_aarch64 target to AOT.

--
433314278  by A. Unique TensorFlower<gardener@tensorflow.org>:

    mutate xplane for kernel launch flows, not enabled in OSS, because missing event and json support.

--
433313910  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Export Tracing Protocol

--
433305742  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Add XlaOptimizationBarrier tensorflow op.

    See https://www.tensorflow.org/xla/operation_semantics#optimizationbarrier.

    The use case for this is to support the new XLA remat op in jax2tf.

--
433303419  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Add missing Copyright notice in the new files

--
433290108  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Go: Update generated wrapper functions for TensorFlow ops.

--
433289730  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Change call sites from deprecated `mlir::parseSourceFile()` to `mlir::parseSourceFile<::mlir::ModuleOp>()`.

--
433285889  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Implement parser/printer as member methods.

    See https://reviews.llvm.org/D121090.

--
433284791  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Update TFRT dependency to use revision
    http://github.com/tensorflow/runtime/commit/d7dd5e431e9319a690d898f81193d09e87bc8643.

--
433283527  by A. Unique TensorFlower<gardener@tensorflow.org>:

    [XLIR] Support bf16 type for gemm.

--
433280976  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Set print_operand_shape=false for HloPrintOptions::Fingerprint(). This reduces fingerprint computation time by 10-100x.

--
433280889  by A. Unique TensorFlower<gardener@tensorflow.org>:

    Add Remapper pass for TFG.

    Implements ReplaceSigmoidMulWithSwish as a beginning pattern, which is migrated from grappler

--

PiperOrigin-RevId: 433332091