Commit Graph

117151 Commits

Author SHA1 Message Date
Fan Du
0c825e6705
Add CITATION.cff 2021-09-06 10:26:23 -05:00
A. Unique TensorFlower
ddc99304b9 Integrate LLVM at llvm/llvm-project@ac312a9d7c
Updates LLVM usage to match
[ac312a9d7c03](https://github.com/llvm/llvm-project/commit/ac312a9d7c03)

PiperOrigin-RevId: 395099564
Change-Id: Ie59ee9ee169a7b95d60ab0bf6b3235ae7a1556c5
2021-09-06 08:09:53 -07:00
TensorFlower Gardener
409e6a835f Merge pull request #51782 from KumaTea:master
PiperOrigin-RevId: 395093386
Change-Id: I6ef7de35d7327a81d4bf95169a91717706e5e94d
2021-09-06 07:16:04 -07:00
Faizan Muhammad
36a8dba382 Avoid losing payloads at Status recreation
PiperOrigin-RevId: 395088620
Change-Id: Ib17e2942a04351a98533c51a978f3c5227fd68d4
2021-09-06 06:45:54 -07:00
A. Unique TensorFlower
83959a6a52 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 395080520
Change-Id: I6cdde4ac8bdfc69f96d8d155f6291567e2cc28f5
2021-09-06 05:52:47 -07:00
TensorFlower Gardener
ef2500338e Merge pull request #51851 from Tessil:toupstream/mlir_fill_op_force_same_scale
PiperOrigin-RevId: 395073754
Change-Id: Ib452f8fcd651cd535560606c898a4a76f58065c0
2021-09-06 04:55:00 -07:00
A. Unique TensorFlower
845b0b05db Use the raw C API for ShardedDeviceArray.
It's similar than PyBuffer.

PiperOrigin-RevId: 395071943
Change-Id: Id0e664ab781ae79e2108c31cb29da4b7386a89be
2021-09-06 04:39:52 -07:00
A. Unique TensorFlower
04cdde6394 Integrate LLVM at llvm/llvm-project@a0a5964499
Updates LLVM usage to match
[a0a596449981](https://github.com/llvm/llvm-project/commit/a0a596449981)

PiperOrigin-RevId: 395061955
Change-Id: I34566bf6ef562fe092e4f5909f70ec43c796401b
2021-09-06 03:28:17 -07:00
Thibaut Goetghebuer-Planchon
272faebcc8 Force the MLIR quantizer to assign the same scale for the quantized operand and result of the FILL op 2021-09-06 11:03:51 +01:00
Jean-Baptiste Lespiau
a4f691af27 Use the raw C API for ShardedDeviceArray.
It's similar than PyBuffer.

PiperOrigin-RevId: 395058367
Change-Id: Ib9b9431ece0bea3ff1310105cb335bff4c88725a
2021-09-06 03:02:34 -07:00
A. Unique TensorFlower
279287fdf6 Update GraphDef version to 881.
PiperOrigin-RevId: 395051343
Change-Id: I562a76c7e7b7a2f35a2127779bea88b633939e4c
2021-09-06 02:16:57 -07:00
A. Unique TensorFlower
cd7f23375c compat: Update forward compatibility horizon to 2021-09-06
PiperOrigin-RevId: 395051329
Change-Id: If2f674a883fc95302e40f848731a1b0d3b48880a
2021-09-06 02:09:09 -07:00
A. Unique TensorFlower
7f942dd7b3 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 395049234
Change-Id: I97fe88eae84b6eb47b7c298975c3ead06470e4ed
2021-09-06 01:50:01 -07:00
A. Unique TensorFlower
b3ade93659 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 395024120
Change-Id: I5b7582b46633973f9b69ad34efe20ca0ee1783fd
2021-09-05 21:49:47 -07:00
A. Unique TensorFlower
fdfd1e0989 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394969790
Change-Id: I3301205df32d0647fbabd312dd07080bc4595185
2021-09-05 09:51:37 -07:00
A. Unique TensorFlower
b7e4172f09 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394960695
Change-Id: I1ef2cbf268e1ed30165343f08b7fd28e0a9cfdca
2021-09-05 07:52:12 -07:00
TensorFlower Gardener
17bf3ec089 Merge pull request #51808 from Intel-tensorflow:amin/bug-fix-pattern-matcher
PiperOrigin-RevId: 394944110
Change-Id: Ic64011d8bb44b70ffb117e1ae4a0903eef29c7f3
2021-09-05 04:06:41 -07:00
Renjie Liu
c38e12c163 Support uint32 cast for tflite
PiperOrigin-RevId: 394943535
Change-Id: I98cc78ad52898b2a5704a24811da25bd4c288e87
2021-09-05 03:59:00 -07:00
A. Unique TensorFlower
cfcb091c4e Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394942946
Change-Id: I159a87eca8aa23b01192f9b58d514f46a8d7e7a0
2021-09-05 03:49:57 -07:00
A. Unique TensorFlower
7f07ae89c0 compat: Update forward compatibility horizon to 2021-09-05
PiperOrigin-RevId: 394934387
Change-Id: Ic493c63510b608265c3d083ed612ff5f147a58a3
2021-09-05 02:13:14 -07:00
A. Unique TensorFlower
803f38706c Update GraphDef version to 880.
PiperOrigin-RevId: 394934384
Change-Id: I16a62ed3e1aab1adef1a522f21e08f033cdeed03
2021-09-05 02:07:12 -07:00
A. Unique TensorFlower
1fd0e34628 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394933188
Change-Id: I0c3b3831111652e0465de35c48c16b9b555820d8
2021-09-05 01:52:17 -07:00
A. Unique TensorFlower
b926c1a0c3 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394923620
Change-Id: Ie423cd42d55d332bebbf1b991d53333771702f29
2021-09-04 23:51:00 -07:00
A. Unique TensorFlower
d01e20681f Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394906646
Change-Id: I1739e02f76f752a06cccc4a605d6827b6af02111
2021-09-04 19:51:44 -07:00
A. Unique TensorFlower
d62c6069bd Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394898598
Change-Id: I85dec8fffb46cd98087b699e4904c3b2e170bcd6
2021-09-04 17:51:35 -07:00
Yash Katariya
9dfa5fce87 Add a copybara rule for grpc
PiperOrigin-RevId: 394865627
Change-Id: I26c7fe35f012d082fad4573274e4c4c9104de161
2021-09-04 10:01:44 -07:00
A. Unique TensorFlower
210a17d830 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394865077
Change-Id: Iaf52d2c77058a0d66df7809782f2c8038fa39f46
2021-09-04 09:52:54 -07:00
CJ Carey
2089bbd6a1 Use a sized-container detection strategy supported by MSVC as well as clang.
PiperOrigin-RevId: 394846952
Change-Id: I5306092359fbfcdd7c5f9168578b7582c2692676
2021-09-04 05:57:02 -07:00
A. Unique TensorFlower
f2216110f8 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394846800
Change-Id: Ie65619932b67d217df462cce8d2b364cc1ee05d1
2021-09-04 05:51:26 -07:00
A. Unique TensorFlower
03aa35d555 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394837338
Change-Id: Ia97b7daf6504119ad96a9720dd97e42a1f4e5a6a
2021-09-04 03:52:12 -07:00
A. Unique TensorFlower
454e99de46 Integrate LLVM at llvm/llvm-project@1ece14b652
Updates LLVM usage to match
[1ece14b65275](https://github.com/llvm/llvm-project/commit/1ece14b65275)

PiperOrigin-RevId: 394830427
Change-Id: I9844010352d12d41608899a7b85f0f9f1d4039a4
2021-09-04 02:27:39 -07:00
A. Unique TensorFlower
2bd7addfcc Update GraphDef version to 879.
PiperOrigin-RevId: 394829050
Change-Id: I6da36933f087813a410fe29286835cbd074443bd
2021-09-04 02:12:32 -07:00
A. Unique TensorFlower
6d3c4948d1 compat: Update forward compatibility horizon to 2021-09-04
PiperOrigin-RevId: 394829036
Change-Id: I0e93012e492bb6c49d99c3756c2be565222cd326
2021-09-04 02:07:06 -07:00
Jay Shi
d18c738d23 [tf.data] Implement interleave_depth value for different iterators, which counts the number of ParallelInterleaveDatasets in the path from root node to the node in the input pipeline tree, not including the node itself.
PiperOrigin-RevId: 394826250
Change-Id: I2aa7d2a56db10e9a1e78573d5318be0d36e1e181
2021-09-04 01:26:01 -07:00
A. Unique TensorFlower
0088a1ada5 Internal build cleanup
PiperOrigin-RevId: 394812335
Change-Id: I7a16ccfd3cb4d4db9331ce600f34d3c4309f4fb7
2021-09-03 22:25:59 -07:00
Xunkai Zhang
b04ad22413 #tflite-maven-reorg Remove DataType's dependency to native runtime.
PiperOrigin-RevId: 394800634
Change-Id: I413734976e43ca728039e6adadaa2d35e6eddd08
2021-09-03 20:01:33 -07:00
Jiri Simsa
caf7d98937 [tf.data] Fix a memory leak.
PiperOrigin-RevId: 394800264
Change-Id: I22ad6309f447bd74f1fb19247a1acd8856c1ac09
2021-09-03 19:55:59 -07:00
A. Unique TensorFlower
bddd0adbd6 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 394800124
Change-Id: I3e0057c20d443e514db81a38279af9dc4942e826
2021-09-03 19:51:06 -07:00
A. Unique TensorFlower
1706cb0371 Internal change
PiperOrigin-RevId: 394799281
Change-Id: Ie3758998f29c5dd2ed8873706b4fae22a6bee491
2021-09-03 19:39:07 -07:00
A. Unique TensorFlower
0d4a9ba888 Verify the nullity of operands.
Operation to be verified may in an invalid form. Be conservative while accessing the values of the operation.

PiperOrigin-RevId: 394795868
Change-Id: I3ba1a0b213f6718b81795b1c595f9fefc98a176f
2021-09-03 19:00:11 -07:00
Mark McDonald
b029cbd2a1 Convert TensorFlow Lite API navigation to use accordion style.
This is in preparation for importing the new Lite Java APIs, which are quite
lengthy. Using accordion sections keeps things tidy when the sections are
large.

The `platform` additions are purely semantic.

PiperOrigin-RevId: 394793465
Change-Id: Iba238cee60b17e0dc9f5bb17f739923a2a4af27f
2021-09-03 18:45:42 -07:00
A. Unique TensorFlower
fae88228a7 TPU embedding: introduce moving average term into the accumulator computation
for the Adagrad momentum optimizer.

PiperOrigin-RevId: 394792505
Change-Id: I25c449fbbf606a0548466f4fae193aac5c8a86ef
2021-09-03 18:39:01 -07:00
Reed Wanderman-Milne
fc91e5caf3 Add make_deterministic pass.
This pass is enabled when deterministic ops is enabled. The pass rewrites ParallelInterleave ops into non-parallel Interleave ops if the ParallelInterleave is potentially nondeterministic due to the use of stateful ops within the interleave function.

PiperOrigin-RevId: 394791760
Change-Id: I1f3a81e715b412d5addc6d60a4a46abeff7282d4
2021-09-03 18:32:47 -07:00
Katherine Wu
e6a4711994 Refactor SavedModel SaverDef implementation so that the save/restore functions are generated in the Eager context.
This change prepares the SavedModel code to allow users to register advanced checkpoint saving/restore functions.

The SaverDef is a SavedModel field that allows TF2 SavedModels to be compatible with TF1. Before this CL, the SaverDef is generated by copying all objects and resource tensors from the eager context/default graph to the SavedModel exported graph.

The object-copying will not work well when we start introducing custom objects and classes, so this change defines the save and restore functions in same context that the user is saving in.

The extra arguments in `GraphView.frozen_saveables` are removed, and there is a slight reshuffling of code to make things the functionality partitions clearer.

PiperOrigin-RevId: 394791282
Change-Id: I94d049e4119e4963f5f4469e88aaf60aa1bafcb3
2021-09-03 18:26:51 -07:00
Yuanzhong Xu
894d827547 [XLA:SPMD] Fix simple manual propagation
We should not call GetShardingFromUser if it's purely manual.

Also fix minimum tile count for merging so that it is properly marked as unchanged.

PiperOrigin-RevId: 394789513
Change-Id: I725fc0eaedf77ad9c95c244bd079d08d237ddc72
2021-09-03 18:10:06 -07:00
Reed Wanderman-Milne
51ee415540 Rollback of https://github.com/tensorflow/tensorflow/pull/51392 (for the second time).
Causes an internal performance regression.

PiperOrigin-RevId: 394788152
Change-Id: I702fb4ec245823b96ce82f58b1b0d6c505b674c3
2021-09-03 17:57:45 -07:00
Faizan Muhammad
7bf435719b Add CreateWithUpdatedMessage helper function in errors.h
PiperOrigin-RevId: 394785752
Change-Id: I1b210d785e05fc70d4611aedb60166de09595b6f
2021-09-03 17:33:18 -07:00
TensorFlower Gardener
fdf55ec918 Merge pull request #51788 from lipracer:master
PiperOrigin-RevId: 394781317
Change-Id: I34ea7d269faf4d305cef6911565729344f3ce1f2
2021-09-03 16:59:40 -07:00
Austin Anderson
4bfa9f1f88 Handle old git versions
Somehow the version of Git that all our sizetracker jobs use reverted to one
that did not support this method of %(trailer:...) formatting in git log
--pretty=format:...
PiperOrigin-RevId: 394780028
Change-Id: Ifaa680bec8c194200ecd07826943f6f257d2ed07
2021-09-03 16:53:03 -07:00
Kuangyuan Chen
50ffddcc09 Deprecate force_bef_function_async
force_bef_function_async is not useful and even confusing. Though it launches the first BEF function call inline, it cannot control whether the graph is executed in a single thread. A more appropriate approach is to control the threadpool implementation to control threading behavior.

PiperOrigin-RevId: 394775487
Change-Id: I1018a0e684b006e9388bac9f8fff354f2c5bf446
2021-09-03 16:26:23 -07:00