Commit Graph

113215 Commits

Author SHA1 Message Date
Mihai Maruseac
a5317d67e6 Merge pull request #50498 from penpornk/cherrypicks_0N2LD
r2.6 cherry-pick request: Add the missing kernels_experimental header file to pip package
v2.6.0-rc0
2021-06-28 17:23:36 -07:00
Mihai Maruseac
1bb2ca624f Merge pull request #50496 from tensorflow/geetachavan1-patch-1
Update setup.py
2021-06-28 17:23:24 -07:00
Mihai Maruseac
410fd069c8 Update setup.py 2021-06-28 16:36:55 -07:00
Mihai Maruseac
7d88138c9a Update tensorflow/tools/pip_package/setup.py 2021-06-28 16:36:31 -07:00
Mihai Maruseac
8c82372aa7 Update tensorflow/tools/pip_package/setup.py 2021-06-28 16:36:21 -07:00
geetachavan1
8ec8bc53a9 Update setup.py 2021-06-28 16:30:22 -07:00
geetachavan1
84b9363899 Update setup.py 2021-06-28 16:28:08 -07:00
Mihai Maruseac
6ee7d01a22 Update tensorflow/c/BUILD 2021-06-28 16:09:36 -07:00
Mihai Maruseac
73fb51374d Merge pull request #50500 from geetachavan1/cherrypicks_0Q30D
[CherryPick:r2.6]Rollback breaking change
2021-06-28 16:08:27 -07:00
Mihai Maruseac
9cb751f56b Merge pull request #50499 from geetachavan1/cherrypicks_KQPKS
[CherryPick:r2.6]Disable test on macos and debug.
2021-06-28 16:07:53 -07:00
Mihai Maruseac
eb244d4375 Rollback breaking change
PiperOrigin-RevId: 381578821
Change-Id: I08d1c7896d76a19c6e693d8e178b89f5fb49af15
2021-06-28 15:37:56 -07:00
Yunlu Li
139646b15d Disable test on macos and debug.
PiperOrigin-RevId: 381563828
Change-Id: I8199401a23494d9c0f08e4818198ef07c51cf969
2021-06-28 15:33:41 -07:00
Mihai Maruseac
5770209bf5 Merge pull request #50495 from tensorflow-jenkins/version-numbers-2.6.0rc0-28554
Update version numbers for TensorFlow 2.6.0-rc0
2021-06-28 14:56:36 -07:00
Kulin Seth
7408cb8462 Add the newly added kernels_experimental header file to pip package. 2021-06-28 14:34:16 -07:00
geetachavan1
bbd799ce69 Update setup.py 2021-06-28 12:47:39 -07:00
TensorFlow Release Automation
efb23ee0f6 Update version numbers to 2.6.0-rc0 2021-06-28 12:30:29 -07:00
Peter Hawkins
539bd7cc39 [XLA:Python] Try once more to fix nonpip builds (MacOS, Ubuntu).
PiperOrigin-RevId: 381507252
Change-Id: Ifd958337d134181c708da4d619f91d476eabc06c
2021-06-25 11:08:52 -07:00
Geeta Chavan
82d094a101 Disable a test
PiperOrigin-RevId: 381503610
Change-Id: I7cd6de450286d1a961b814d9680e827af03566b5
2021-06-25 10:54:41 -07:00
Yunlu Li
21c1f02dbc Revert: Add integration test for training cancellation support.
PiperOrigin-RevId: 381503409
Change-Id: I850783294157e0cf93b59407123aec699340f02f
2021-06-25 10:49:53 -07:00
Rick Chao
81457e8917 ClusterCoordinator: Avoid wrapping wait_on_failure on resource rebuilding to prevent worker thread's deadlock, and handle InputError by looking at it's original exception.
PiperOrigin-RevId: 381497459
Change-Id: I5b14ef14ffe1363d1af63f0db8de929cb2eda8b6
2021-06-25 10:25:37 -07:00
Smit Hinsu
c886d996c5 Add supported functional ops by TPU to mark for outside compilation pass
PiperOrigin-RevId: 381493138
Change-Id: Idc4116b5c608ecda522a5190e2a074843b17aadd
2021-06-25 10:06:14 -07:00
Yang Chen
c990301177 [tf.data service] Remove unused class LocalCredentialsFactory.
PiperOrigin-RevId: 381489554
Change-Id: I481c7c5447671670d5a228581025360d8b522966
2021-06-25 09:53:58 -07:00
Dan Moldovan
5f6c0e0a50 Minor doc update.
PiperOrigin-RevId: 381478401
Change-Id: Icae8bb3d463073e3a7525b396b149f715e463a7c
2021-06-25 08:51:23 -07:00
A. Unique TensorFlower
1ba2eb7b31 Update TFRT dependency to use revision
b570a1921c.

PiperOrigin-RevId: 381464971
Change-Id: I7e86fbf0e49ad8a6fe8fb7c46eb50d597038879c
2021-06-25 07:15:43 -07:00
TensorFlower Gardener
2420695580 Merge pull request #50360 from linearhit:disc_dev2
PiperOrigin-RevId: 381462080
Change-Id: I5d507aff2a38f142104a57c7c3b9a155d536f3f6
2021-06-25 06:56:29 -07:00
Peter Hawkins
fade81b13f [XLA:Python] Fix build failure on Mac OS.
PiperOrigin-RevId: 381460038
Change-Id: I49d1a2962e95af1e228bb638f13c6e2100636353
2021-06-25 06:47:51 -07:00
A. Unique TensorFlower
80c714662d Integrate LLVM at llvm/llvm-project@f7b1fa6f5e
Updates LLVM usage to match
[f7b1fa6f5ebe](https://github.com/llvm/llvm-project/commit/f7b1fa6f5ebe)

PiperOrigin-RevId: 381459627
Change-Id: I4d28ef7b4a326fbbc66b36f93c53f7150c1731ed
2021-06-25 06:41:50 -07:00
Stefano Galarraga
ae5ad0d02b Add events to track failures initializing mini-benchmark
PiperOrigin-RevId: 381449632
Change-Id: I1b29a3dfb9186988129b594704f433e02bf6bee6
2021-06-25 05:28:49 -07:00
A. Unique TensorFlower
4b67de7963 Update TFRT dependency to use revision
d3bd15c978.

PiperOrigin-RevId: 381449212
Change-Id: Ib86d0161c085ca8b4c6cf559716cb28439a60924
2021-06-25 05:21:23 -07:00
A. Unique TensorFlower
bffcd01d3f Integrate LLVM at llvm/llvm-project@d07f43641f
Updates LLVM usage to match
[d07f43641f98](https://github.com/llvm/llvm-project/commit/d07f43641f98)

PiperOrigin-RevId: 381446296
Change-Id: I0d3413738d1412e11b7917858623437fbd28d8f0
2021-06-25 04:56:28 -07:00
Thomas Köppe
c0d892e6b6 Fix type of "bytes_accessed" (from float to int64): Every interaction with this field is with an int64, never with a float. It stands to reason that this was meant to be an int64 field.
PiperOrigin-RevId: 381443185
Change-Id: I74426f6cb9d13c293000338e37d6331639ddc055
2021-06-25 04:28:12 -07:00
A. Unique TensorFlower
935dfb5eb5 Update GraphDef version to 808.
PiperOrigin-RevId: 381426426
Change-Id: I7bbe63a867b3a79f4b7c100218b123f59bc9038d
2021-06-25 02:13:24 -07:00
A. Unique TensorFlower
a3da6c0a52 compat: Update forward compatibility horizon to 2021-06-25
PiperOrigin-RevId: 381426365
Change-Id: I3e2779c68dbd7f205fde16d03b98332e2e199434
2021-06-25 02:05:59 -07:00
Taehee Jeong
f2ca8d51c4 Improve error message during TFR op generation
PiperOrigin-RevId: 381409746
Change-Id: Ieb6e6bb3df54f14876b00fa744e131170fabca92
2021-06-24 23:43:40 -07:00
A. Unique TensorFlower
bc5eddee3d Add Ragged support to tf.math.sigmoid operation.
PiperOrigin-RevId: 381407240
Change-Id: Ic64d0603d85da24b436ab190a9b7c0ba34c9412c
2021-06-24 23:15:27 -07:00
A. Unique TensorFlower
0730663768 Use optimized dequantizer for detection postprocess.
PiperOrigin-RevId: 381405890
Change-Id: Ic1ca8c438a93693e16d6cc13f18357747c8b3b4a
2021-06-24 23:02:54 -07:00
Ruoxin Sang
32634e9041 Add a flag in distributed_tpu_rewrite_pass to control adding locking for multicore TPU computations so that the same program on different TPU cores are enqueued atomically. This avoids deadlocks or incorrect results when the computations aren't otherwise sequenced.
PiperOrigin-RevId: 381401618
Change-Id: I349deb754145560a309475e337da0b8b6b330974
2021-06-24 22:19:14 -07:00
A. Unique TensorFlower
f1bde5d92e Update TFRT dependency to use revision
4b20d65ea5.

PiperOrigin-RevId: 381398940
Change-Id: I6180c1e35064de1181acb67f0de718c48484d46a
2021-06-24 21:51:31 -07:00
A. Unique TensorFlower
0c8078d480 Integrate LLVM at llvm/llvm-project@dcccb2f594
Updates LLVM usage to match
[dcccb2f59401](https://github.com/llvm/llvm-project/commit/dcccb2f59401)

PiperOrigin-RevId: 381396260
Change-Id: Iae980fae57058c591add734e44339c0eaf644343
2021-06-24 21:22:26 -07:00
A. Unique TensorFlower
0a935589b9 Use optimized ArgMax to replace partial ArgSort when the number of category is one.
PiperOrigin-RevId: 381393443
Change-Id: I2f0a47184d374543124b137b81741857e783bdeb
2021-06-24 20:56:07 -07:00
A. Unique TensorFlower
909938a177 [tf.data] Replace port::MaxParallelism() with port::NumSchedulableCPUs() to reflect the available hardware CPUs.
PiperOrigin-RevId: 381391150
Change-Id: I088327395809679ea91c60b8cbcebc04e7681d3c
2021-06-24 20:31:26 -07:00
David Majnemer
0d3996f57c Fixing the MSVC issue.
PiperOrigin-RevId: 381387766
Change-Id: If4a2cb20205046646229ced8855ad0817d8294be
2021-06-24 19:57:17 -07:00
Thai Nguyen
c6d88a1778 Add Roll as a TFLite custom op
MLIR converter support lowering for TF_RollOp with constant shift and axis. However, if that assumption is not true, a custom op is required.

PiperOrigin-RevId: 381383127
Change-Id: I0f76859c728413f3abb197d2e4986e97a15475fd
2021-06-24 19:05:06 -07:00
Jaesung Chung
6403cdf411 Remove unsupported int64 type from TFLite Split op legalization
Fixes #50445

PiperOrigin-RevId: 381380985
Change-Id: I10c3919753f62e337cc1f3030b5e0741661c073b
2021-06-24 18:46:16 -07:00
Peter Hawkins
1a6fb54c2a Revert:
[PJRT] Simplify transpose planning code.

We already have a level of indirection (the loop_order), so we don't need to muck with the order of the input dimensions to handle striding.

Cleanup only, no functional changes intended.

PiperOrigin-RevId: 381376782
Change-Id: Ib816a46f33b01c5fc69faf300f04414be3116f20
2021-06-24 18:27:36 -07:00
TensorFlower Gardener
12dd44c010 Merge pull request #50311 from armkevincheng:tosa/pr_007
PiperOrigin-RevId: 381375154
2021-06-24 18:21:28 -07:00
A. Unique TensorFlower
e8253e1138 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 381374492
Change-Id: Ie94eb36485d8835b7400759db4d15f0e7fa442d6
2021-06-24 18:12:46 -07:00
TensorFlower Gardener
4cdece0345 Merge pull request #50444 from tensorflow:jvishnuvardhan-patch-6
PiperOrigin-RevId: 381372887
Change-Id: I91a11411db3de22a753029e8a643429fefda62b0
2021-06-24 18:08:15 -07:00
Andrew Audibert
945ab698fd [tf.data service] Support concatenate in distributed_epoch mode.
PiperOrigin-RevId: 381371085
Change-Id: If4e8252ebaf80301bb57afce032c00b21f35a1cb
2021-06-24 18:02:09 -07:00
A. Unique TensorFlower
00a29e825c Variable-to-constant conversion method for ConcreteFunction in Graph mode
PiperOrigin-RevId: 381370619
Change-Id: I9fbe3a7b392dd1974c30cc74d089b4467304aadd
2021-06-24 17:56:37 -07:00