Commit Graph

85844 Commits

Author SHA1 Message Date
Jared Duke
1f530076d1 Fix TFLite builds on Windows/MacOS
Avoid using `--enable_platform_specific_config` when cross-compiling for
iOS/Android, as this pulls in host build flags, which may not be
appropriate (e.g., when cross-compiling for Android on a Windows host).

Also fix an issue when building tensorflowlite_c for iOS.

Fixes .

PiperOrigin-RevId: 311767770
Change-Id: I80b817fd89a6889dc78be50f1def8b899f091cb6
2020-05-15 11:26:35 -07:00
Pavithra Vijay
ff17316b19 Check for _metrics in case sublayer is resetting _metrics property.
PiperOrigin-RevId: 311767501
Change-Id: I1f97904314a0f1912c918b89f461edd1183f4604
2020-05-15 11:23:16 -07:00
Edward Loper
77245d07d1 Add dispatch support to more Python APIs.
PiperOrigin-RevId: 311763060
Change-Id: Ib35371483aa083e245996508a82fd13d8ac43131
2020-05-15 11:03:18 -07:00
Skye Wanderman-Milne
26104505b8 [XLA:Python] Expose new use_spmd_partitioning compile option.
PiperOrigin-RevId: 311761373
Change-Id: I1f696e0c082295dc0e6896f05d1e88525de7ce70
2020-05-15 10:59:38 -07:00
Xunkai Zhang
262e92804b [tfls.util] Remove tensorflow-lite-gpu from dependencies.
PiperOrigin-RevId: 311760392
Change-Id: Ia8fe0682cfda037589f7546f1e70974c1be439c5
2020-05-15 10:56:34 -07:00
A. Unique TensorFlower
bd010a095e String formatting in assertAllEqual() fails in Python 3 because bytestring may
be converted using %s. Use %r to fix this.

PiperOrigin-RevId: 311760220
Change-Id: Ia46073b51bc38b8e88016edab37bc34ceebd5d7f
2020-05-15 10:53:11 -07:00
Tomer Kaftan
d62a22a300 Extend Keras Lambda layers to work with functions of any signature rather than only functions that take one argument.
Any *args and **kwargs passed when calling the lambda layer will be forwarded directly to the underlying lambda.

PiperOrigin-RevId: 311759844
Change-Id: Ia5ffe17f2951e4fd42d9ee4020c7c8b35ef9122f
2020-05-15 10:48:28 -07:00
Yuanzhong Xu
2540d202b5 Fix TF2XLA's InitGraph for unused feeds.
If a feed is not used, previously it would prune the placeholders and cause crashes.

PiperOrigin-RevId: 311754319
Change-Id: Ie1ad67c21ffb83ba88aeabea94c416473df099a0
2020-05-15 10:23:18 -07:00
Yuanzhong Xu
53c634a6c1 [MLIR/XLA] Constant sinking to control flow regions.
This is necessary for exporting to XLA since functional control flow is expected.

PiperOrigin-RevId: 311753796
Change-Id: If4e50a3b2fa668f162c9b30cc80e2bf743a9b641
2020-05-15 10:19:41 -07:00
Benjamin Kramer
9957cb60a2 Bump open source llvm revision to 9d4b4f344d8ea917e082cf58d66b71c0171e1650
PiperOrigin-RevId: 311751290
Change-Id: Ie8366f82180116dd363c3ed7ece36f948196bf1b
2020-05-15 10:08:18 -07:00
A. Unique TensorFlower
02b5a6754b Integrate LLVM at https://github.com/llvm/llvm-project/commit/9d4b4f344d8e
PiperOrigin-RevId: 311744575
Change-Id: Icdd7f018b188db8f8768f3b40d6411c2257547c1
2020-05-15 09:35:58 -07:00
A. Unique TensorFlower
c568e0dd7f Avoid overhead for creating executors if there is no change in execution mode.
PiperOrigin-RevId: 311742240
Change-Id: I8676ab711d3c3d9e64d4ec142e5d934f7c32ee73
2020-05-15 09:24:46 -07:00
Jose Baiocchi
6bddca85b3 Use fully-qualified std::string in TraceMe
PiperOrigin-RevId: 311741974
Change-Id: Ic9100c53ded4011b590651cbb5ca276b093a3fc2
2020-05-15 09:21:06 -07:00
Chao Mei
e8d51ef601 Remove the unnecessary address-returning operator and lamda expression.
PiperOrigin-RevId: 311737378
Change-Id: I55bf12bf66540ed32dd48d61da7f41bdf2ace5eb
2020-05-15 08:54:16 -07:00
Zhenyu Tan
ec0026c8c3 Update ops_history for bincount.
PiperOrigin-RevId: 311735482
Change-Id: I4bff5fdf6a840a5a5c692b5b906817815d41ba71
2020-05-15 08:41:47 -07:00
A. Unique TensorFlower
020a88ac12 Rollback to investigate failure
PiperOrigin-RevId: 311731132
Change-Id: I109ce87f13bb1b1c06b3e110bafbdf9c014c8258
2020-05-15 08:13:02 -07:00
Haoyu Zhang
8da4a14be3 Avoid overhead for creating executors if there is no change in execution mode.
PiperOrigin-RevId: 311726778
Change-Id: I33a1e5085e1740504181bd6096229b6df12b26f8
2020-05-15 07:37:26 -07:00
A. Unique TensorFlower
0c6f6f4776 Integrate LLVM at https://github.com/llvm/llvm-project/commit/854f5f332af4
PiperOrigin-RevId: 311704350
Change-Id: I2314bba4a4f72fdceda3a3439bac3e3fb96b811a
2020-05-15 04:03:03 -07:00
Stephan Herhut
df2c8d2823 Allow index typed values in hlo_scalars_to_dimension_tensor.
The limitation stems from a time where tensors with index element type were not allowed in MLIR. With this change, we can remove many `index_cast` ops.

PiperOrigin-RevId: 311703219
Change-Id: I56c7dba29e43b3ee13a1066c0974f72e696600ab
2020-05-15 03:49:24 -07:00
A. Unique TensorFlower
bfe99ed9d6 Remove default initialization in tflite::StatefulNnApiDelegate::Data struct.
PiperOrigin-RevId: 311702143
Change-Id: I317473ef15a0ee8f31b1b99ee6e9f23f9f4f19cd
2020-05-15 03:36:23 -07:00
A. Unique TensorFlower
06c671cde8 [XLA] Verify statically shaped result type in xla_hlo.reshape operation
The result type of the `xla_hlo.reshape` operation must have a static shape.
This is now checked by the operation's verifier.

PiperOrigin-RevId: 311697582
Change-Id: I90e8e513d205d62bb052a6cbecd7ebd88614db6d
2020-05-15 02:49:51 -07:00
A. Unique TensorFlower
ab70af78dc compat: Update forward compatibility horizon to 2020-05-15
PiperOrigin-RevId: 311693261
Change-Id: Id490a7dbe95ff4e493b3490d71c92a9b41f2b484
2020-05-15 02:10:10 -07:00
A. Unique TensorFlower
f6e2a28158 Update GraphDef version to 402.
PiperOrigin-RevId: 311693254
Change-Id: Ieb9a5f81784013ad6bb4a6b1fd1f119b34c68604
2020-05-15 02:06:28 -07:00
A. Unique TensorFlower
cca62cc73b Add a new DelegateUtils::InvokeWithCPUFallback API for handling delegate failures automatically in Invoke(). This is especially useful for NNAPI usage, as drivers may evict clients.
The fallback is suitable when callers:
- Use CPU buffers only (no AHardwareBuffer or GL buffers)
- Do not store pointers to tensor data across Invoke() calls, as they are invalidated
- Do not have tensor state across multiple Invoke() calls

PiperOrigin-RevId: 311687567
Change-Id: Ib3b6fd3fb6c4e85c9512d939d8e87efea03ac4fe
2020-05-15 01:05:28 -07:00
A. Unique TensorFlower
2b2e441205 Integrate LLVM at https://github.com/llvm/llvm-project/commit/dad2e92eaf53
PiperOrigin-RevId: 311683742
Change-Id: I6177dd65ae548b719c656201ed1a7f9829acd745
2020-05-15 00:31:16 -07:00
Hye Soo Yang
24c75ce501 Addresses https://github.com/tensorflow/tensorflow/issues/38694 by ensuring TensorShape v1/v2 backwards compatibility.
PiperOrigin-RevId: 311670326
Change-Id: I0e7045ff4eb19cb0096d6fa41a494c8f9a6b85c5
2020-05-14 22:23:54 -07:00
Renjie Liu
28899d991f Optimize int8 broadcast min.
PiperOrigin-RevId: 311665392
Change-Id: I566547f44975d3d88cb7a17e8c6418a4a186ccda
2020-05-14 21:33:02 -07:00
Yujing Zhang
5cf4311435 Fix a memory leak.
PiperOrigin-RevId: 311662668
Change-Id: I59f9c9cdb8baed7a9828bb818ce1d293d185e6b6
2020-05-14 21:03:46 -07:00
Frank Chen
97f2fffe7e This is an internal change not visible to the public.
PiperOrigin-RevId: 311661259
Change-Id: I8443f6037e3cabca1dca72ce1748eea8fd71770c
2020-05-14 20:49:25 -07:00
Lu Wang
a9c0ce87a6 Check the minimum metadata parser version in the MetadataExtractor Java library.
PiperOrigin-RevId: 311657605
Change-Id: I39169392214b8a70d5882c5ec4af93021480ce23
2020-05-14 20:12:29 -07:00
A. Unique TensorFlower
9489fbca67 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 311655487
Change-Id: Ia4b492dc27139b316a3f8a5b90d68582c05efe4a
2020-05-14 19:52:36 -07:00
Jonathan Hseu
a98948acf8 Use CompactTextString instead of String for generating ops.
PiperOrigin-RevId: 311655146
Change-Id: I57e5c595522b47dd9badbf0720569ffef69fed66
2020-05-14 19:48:45 -07:00
Renjie Liu
3b225a9776 Rework kernel check for fully_connected.
PiperOrigin-RevId: 311655034
Change-Id: Ic82fd9a9350cac89043db85d1ba1d4ec480435e5
2020-05-14 19:45:18 -07:00
David Rim
37df93331e Adds utility methods for storing SignatureDefs in the metadata table in the flatbuffer
PiperOrigin-RevId: 311652937
Change-Id: I397c7ce6fad843cff789dedb583d6df44545db3f
2020-05-14 19:22:55 -07:00
A. Unique TensorFlower
efa3fb28d9 Split index_lookup into string_lookup and integer_lookup.
PiperOrigin-RevId: 311651579
Change-Id: Ie033727dbe1026a7c7a88e4b31653840a17ac3d1
2020-05-14 19:12:15 -07:00
Taehee Jeong
0de7edf8b1 Generate separate pod for Core ML delegate
PiperOrigin-RevId: 311651255
Change-Id: I7ba8755d447674fa8d20935b0c9815a5406b879f
2020-05-14 19:07:57 -07:00
TensorFlower Gardener
93a30c941c Merge pull request from yongtang:39381-protobuf-java
PiperOrigin-RevId: 311649749
Change-Id: I690d1353dfed2f3e29de53c78749fd038008192f
2020-05-14 18:54:55 -07:00
A. Unique TensorFlower
a5267f056f Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 311648388
Change-Id: Id9f6f7c4de82be3a405377e722e740fd0dfee80d
2020-05-14 18:43:58 -07:00
Renjie Liu
377612c026 Fix hardswish test for ubsan.
PiperOrigin-RevId: 311645688
Change-Id: Id9f3b31da09355c9997f3f2cc95dca5954c956ec
2020-05-14 18:23:20 -07:00
Smit Hinsu
4662933489 Build DynamicSlice and DynamicUpdateSlice ops with MlirHloBuilder
Whitelist XlaDynamicSlice and XlaDynamicUpdateSlice for testing

PiperOrigin-RevId: 311642899
Change-Id: Icbf009cf69d3b183d0c83c10925a5fbaa3c49f1f
2020-05-14 18:04:39 -07:00
A. Unique TensorFlower
a2ef8b5a06 Update svd_op_test to run (non-gradient) tests in eager as well as graph mode.
PiperOrigin-RevId: 311640894
Change-Id: I39b4666c461c64ffe3f33992bb536961a266abd7
2020-05-14 17:51:37 -07:00
Haoyu Zhang
d5e0f468cd Report remote target in error messages for gRPC eager service requests.
PiperOrigin-RevId: 311634462
Change-Id: Ib0550c172e419ea17dac9ffa28c18b9e1a03b3cc
2020-05-14 17:09:29 -07:00
Lucy Fox
90077f8c7c Instrument the number of times the MLIR-based TF Bridge is enabled.
PiperOrigin-RevId: 311633792
Change-Id: Iba286e1c82900833b5cf9f69a697a312e51f3156
2020-05-14 17:04:52 -07:00
Anna R
e6c2a5a212 Change more libraries in third_party/tensorflow/c/BUILD to depend on portable_tensorflow_lib_lite_no_runtime to support effort to reduce dependencies on mobile.
PiperOrigin-RevId: 311632630
Change-Id: I5061b458f894bccb9c0e23791d265f6ee95bdd38
2020-05-14 16:56:40 -07:00
TensorFlower Gardener
7834b9d9ac Merge pull request from andrewxcav:master
PiperOrigin-RevId: 311627750
Change-Id: Ia492517dcedf6d76f9a29aedf256d155f8e1b255
2020-05-14 16:31:42 -07:00
Jared Duke
7d40f2c389 Fix bazel TFLM compilation w/ TF_LITE_STATIC_MEMORY
Ensure dynamic string utils aren't compiled when this build define is
present.

PiperOrigin-RevId: 311626904
Change-Id: Ica229bf337019f0f446fdb94aaf42c6b7e7c749e
2020-05-14 16:27:45 -07:00
TensorFlower Gardener
098f9176b6 Merge pull request from yongtang:mlir_graph_optimization_pass-header
PiperOrigin-RevId: 311625764
Change-Id: I1369f7cd19c383529cea81fe68546e1f75fcecd1
2020-05-14 16:23:02 -07:00
A. Unique TensorFlower
c628246c31 Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 311625551
Change-Id: I3205d380573ed326d5b55cdc089577f34433f1f1
2020-05-14 16:18:58 -07:00
TensorFlower Gardener
8e7a2d65fe Merge pull request from JRMeyer:wav-error-msg
PiperOrigin-RevId: 311624646
Change-Id: I40087b6f423a01ea34cefb113bb8926a7a645ecf
2020-05-14 16:14:26 -07:00
A. Unique TensorFlower
6abea04db7 Added BUILD rules for the micro-frontend TF op so that we can use it as a C++ library.
PiperOrigin-RevId: 311621054
Change-Id: I54e9932fe54f7cc94f5863f7924d85853d24e48e
2020-05-14 15:54:09 -07:00