experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
98,479 Commits 49 Branches 133 Tags 692 MiB
cherrypick_339370490
Commit Graph

220 Commits

Author SHA1 Message Date
Haoyu Zhang
6d4f1d5c09 Use Unavailable error for non-existing server context since this indicates the remote server has restarted. 
PiperOrigin-RevId: 338367526
Change-Id: If9a9a09fd09a55a3f4734518a90905bf0764b669
 2020-10-21 17:32:30 -07:00
Anna R
f5af1da4f8 Clean up _pywrap_tfe deps. pywrap_tfe is a separate shared library and should not include tensorflow sources that are included in the main shared library. 
PiperOrigin-RevId: 337546286
Change-Id: I0841c250637dc7bf43c30ddab825cca38bb05722
 2020-10-16 11:42:44 -07:00
Russell Power
5b5aab7f63 Internal change 
PiperOrigin-RevId: 335147548
Change-Id: Ib445cfbcb28421b4eb522d4d9524e4a64fe631df
 2020-10-02 20:33:42 -07:00
A. Unique TensorFlower
4eb05c3014 Use macro helpers for TPU builds and clean up define flags. 
PiperOrigin-RevId: 334732331
Change-Id: Ice5d240cf785d64d11d4f634ff8955933da26b4d
 2020-09-30 20:13:49 -07:00
Russell Power
45d693198d Use macro helpers for TPU builds and clean up define flags. 
PiperOrigin-RevId: 334725778
Change-Id: Ib0c04366bd9e460329775075d82e7cfd47ed6d4e
 2020-09-30 19:10:21 -07:00
Saurabh Saxena
534cb9ab79 Replace C++ gradient helpers with TapeContext in mnist_gradients_testutil. 
PiperOrigin-RevId: 334255849
Change-Id: I065dc828ec6485822116bbad088827c6c6b3ef46
 2020-09-28 16:03:48 -07:00
Saurabh Saxena
868305f995 Implement python binding for TapeContext. 
Remove usages of tape helpers from mnist_gradients_testutil in the pybind wrappers.

PiperOrigin-RevId: 334245217
Change-Id: I84dec9c255f2c98b1b02ce9aed58ff61e1ff2a67
 2020-09-28 15:05:07 -07:00
Saurabh Saxena
7ec70d54b2 - Integrate C++ tape with op building APIs via TapeContext and TapeOperation which delegate calls to a parent execution context and record operations on the tape. Please see gradients_test.cc for usage. 
- This will replace the helper functions in gradients_internal.h. I will clean that up in a followup CL.
- Also drop ForwardOperation::ctx since that is unused right now. We can add it later if we need.

PiperOrigin-RevId: 333390787
Change-Id: I80f2c460a9538a1a14ed1497c59f7b37a633a633
 2020-09-23 16:12:39 -07:00
Saurabh Saxena
6e8b7c3033 [Cleanup] Add a tracing::MaybeSetOpName utility to reduce the verbosity of C++ ops. 
PiperOrigin-RevId: 333303477
Change-Id: I60d2c3df5d36102ec5e68064c8e309422c22510a
 2020-09-23 09:18:26 -07:00
Wenhao Jia
8f1362de18 Add target environment constraints to a subset of TensorFlow packages and targets. 
PiperOrigin-RevId: 332884872
Change-Id: I65691fa2021c065e6c2ab57815d5a2b342d30ee2
 2020-09-21 10:57:01 -07:00
Brian Zhao
00a5ef689b Support savedmodels that reference assets. 
PiperOrigin-RevId: 332523998
Change-Id: Ia0f7cd75f79020c1b385ee6c02323c9d431ff86d
 2020-09-18 14:29:39 -07:00
Saurabh Saxena
3d8020e442 Rename SparseSoftmaxCrossEntropyLoss to SparseSoftmaxCrossEntropyWithLogits to match up with the op name. 
PiperOrigin-RevId: 332353265
Change-Id: I76cf1dec99438a1661707006a9496aadd8226fc5
 2020-09-17 17:53:59 -07:00
TensorFlower Gardener
dde5f6ce98 Merge pull request from amturati:grad_check_v1 
PiperOrigin-RevId: 332341463
Change-Id: Ie11a45d373ececb711998506067e60e04fa5bc6b
 2020-09-17 16:43:08 -07:00
Haoyu Zhang
9821980f32 Reenable c_api_distributed_test in OSS. 
PiperOrigin-RevId: 332263445
Change-Id: Icd442ea377e917faa2b795e50e829af9ba06ae64
 2020-09-17 10:49:03 -07:00
amturati
81a4ce1a02 adding gradient files to srcs in BUILD (retry to fix format) 2020-09-17 01:53:37 +00:00
Saurabh Saxena
c11debf86c 1. Add helper for casting EagerTensor* to AbstractTensorHandle*. This allows us to directly work with EagerTensors and will also allow using tf.data with unified APIs. 
2. Enable unified_api_test to run with TFRT.

PiperOrigin-RevId: 331662847
Change-Id: I78e53f4eb32d40b6fc6672af3caf2a5c3c74a22c
 2020-09-14 17:40:33 -07:00
amturati
e51c27f54b deleted entry in BUILD 2020-09-13 19:21:32 +00:00
amturati
3b2480f118
Merge branch 'master' into grad_check_v1 2020-09-11 18:39:26 -05:00
Akshay Modi
34b7331467 Disable c_api_distributed_test in OSS since it seems flaky 
PiperOrigin-RevId: 331204141
Change-Id: Idaae784b31758631a0e32e5aec29161bb7d38bb5
 2020-09-11 13:12:22 -07:00
amturati
102242943e updated mnist_gradients_test to use gradients_util 2020-09-10 19:52:20 +00:00
amturati
a82765f58b deleted whitespace on line 386 of BUILD 2020-09-10 19:52:19 +00:00
amturati
610f1599bf fixed BUILD formatting 2020-09-10 19:52:19 +00:00
amturati
4ffd477305 got rid of is_scalar_out, we sum if the number of dims > 0 2020-09-10 19:52:19 +00:00
amturati
a9f647ebdf made is_scalar_out default to true 2020-09-10 19:52:19 +00:00
amturati
a17c8e3f99 took out test-only functions from test_util and put them in gradients_util 2020-09-10 19:52:18 +00:00
amturati
0fdb700766 added gradients_testutil files and fixed Status 2020-09-10 19:52:18 +00:00
amturati
c3594ab934 updated style and fixed BUILD 2020-09-10 19:52:17 +00:00
amturati
713bed7cee crude gradient check working, need to decompose functionality 2020-09-10 19:52:16 +00:00
Saurabh Saxena
896635c9cb [C++ gradients] Add experimental python bindings for unified API's Tape. 
See unified_api_test for usage.

PiperOrigin-RevId: 330992900
Change-Id: I1e7b2fc3ce6ab7d53c8f05a9807e7a03f6e59715
 2020-09-10 12:39:44 -07:00
Saurabh Saxena
7519f88d21 Enable unified_api_test in OSS. 
Needed to make sure that the pybind targets have header-only dependencies.

PiperOrigin-RevId: 330925715
Change-Id: I9b63049cd5a02af73a764837c954586ab7da8af0
 2020-09-10 06:42:07 -07:00
Saurabh Saxena
6cbf1daf2d 1. Do not cache the tracing impl string in c_api_unified_experimental instead actively lookup the factory fn and store a pointer to that in default_factory. This avoid keeping a copy of the string passed in from python. As a result SetDefaultTracingEngine now returns a Status which needed to be plumbed through. 
2. In `GraphContext` keep a copy of the string passed in from python since the original pointer may be prematurely destroyed.

PiperOrigin-RevId: 330647473
Change-Id: I00bf7243b30c3b6b3eb2dd1bb5ba1cbc8b46c008
 2020-09-08 21:12:23 -07:00
Saurabh Saxena
ffc9bef49c Fix segfault in mnist_gradients_test and enable it on TAP. 
PiperOrigin-RevId: 328814584
Change-Id: Icfb075332005988d1aa1ee2cc52b4a6be5b94f47
 2020-08-27 14:21:03 -07:00
George Karpenkov
fbb7dab9fd [TF2XLA] Remove unused and broken call to enable XLA, remove unneeded build macro 
Additionally, "common" bits of the XLA compiler are now linked in by default,
so we don't even need the build option to check for linkage.

PiperOrigin-RevId: 328776929
Change-Id: Ie59796a8c0f0bb400b38397b49853ed23912bdaf
 2020-08-27 11:25:21 -07:00
Saurabh Saxena
d293b02e35 Fix gradients_test and disable mnist_gradients_test so that TAP is unblocked. 
PiperOrigin-RevId: 328206539
Change-Id: I0b208dc20fa454acff3b892b5f0833536e62fc9f
 2020-08-24 14:22:29 -07:00
amturati
65ecf2486c updated mnist_util BUILD entry 2020-08-21 18:24:15 +00:00
amturati
4a770307ba tried moving nn_ops up in the BUILD file 2020-08-20 19:05:38 +00:00
amturati
fe40506296 fixed Build files manually 2020-08-20 16:48:50 +00:00
amturati
126f523167 renamed util files and formatted Build 2020-08-20 15:01:09 +00:00
amturati
7b79154d57 fixed nits and updated RunModel to handle nullptr issue 2020-08-17 22:57:37 +00:00
amturati
2136cd8a3a separated nn_grads and ran clang on all 2020-08-17 22:57:36 +00:00
amturati
fc3cd94b3c rebase update 2020-08-17 22:57:35 +00:00
amturati
efcb8679eb separated ops into correct files & directories 2020-08-17 22:57:34 +00:00
amturati
601254efb3 updated ops and grads directories, no longer need mnist_gradients.h 2020-08-17 22:57:30 +00:00
amturati
ef413e4503 commiting progress so we can update grad signature 2020-08-17 22:57:30 +00:00
amturati
0926d6f238 updating rebase 2020-08-17 22:57:30 +00:00
amturati
22cd9e87cc separated gradient implementations into their own files 2020-08-17 22:57:29 +00:00
amturati
b5f97dc190 created util files for better file decomposition 2020-08-17 22:57:29 +00:00
amturati
aa08b2755a MNIST Forward scores passing 2020-08-17 22:57:28 +00:00
amturati
950378868e separated gradient implementations into their own files 2020-08-17 22:57:26 +00:00
amturati
361d8b4e93 created util files for better file decomposition 2020-08-17 22:57:26 +00:00