Scott Zhu
97d5010a8f
Move keras related eager memory test to keras
...
PiperOrigin-RevId: 305504849
Change-Id: If783553d443ad4c900eebbb72970ea582cf801a5
2020-04-08 10:30:35 -07:00
Brian Atkinson
1800ace98f
Small adjustments on import spacing.
...
This is mostly the result of an internal cleanup and formatting pass.
PiperOrigin-RevId: 286600606
Change-Id: Ibcc5b919701fb71a17dd1c29d6d473cddb48464d
2019-12-20 10:28:04 -08:00
Brian Atkinson
13088cc7ba
Move additional_deps to deps for cuda_py_test.
...
PiperOrigin-RevId: 285289102
Change-Id: Ic1ca096f27689fda42c9fc98505adbe5b7489993
2019-12-12 15:53:59 -08:00
Kibeom Kim
cafc2b641f
Fix _DelayedRewriteGradientFunctions memory leak.
...
Root cause:
_DelayedRewriteGradientFunctions was always registered to global gradient registry that's never deleted.
Fix:
Don't use global gradient registry for _DelayedRewriteGradientFunctions. Instead, keep it as op._gradient_function. It has two benefits:
1. Lifetime management is simple and intuitive since _DelayedRewriteGradientFunctions is used by op.
2. Make the code simpler by not using global registry.
PiperOrigin-RevId: 273858076
2019-10-09 22:35:58 -07:00
Akshay Modi
eda96d073f
Use the SendTensor op by default
...
PiperOrigin-RevId: 270381552
2019-09-20 17:17:58 -07:00
Gunhan Gulsoy
2417ede33d
Disable flaky memory_test everywhere.
...
PiperOrigin-RevId: 268081532
2019-09-09 15:28:18 -07:00
Mehdi Amini
abaf7ecd91
Exclude python/eager/memory_tests:memory_tests_gpu from continuous integration: it is flaky at the moment
...
PiperOrigin-RevId: 268053311
2019-09-09 13:19:55 -07:00
Yifei Feng
a42c0ed8d2
Disable flaky test //third_party/tensorflow/python/eager/memory_tests:remote_memory_test_xla_gpu
...
PiperOrigin-RevId: 267453483
2019-09-05 14:50:02 -07:00
Akshay Modi
b74fe7f4b1
Add test main to the memory test file
...
PiperOrigin-RevId: 265569142
2019-08-26 17:02:01 -07:00
Akshay Modi
a8c33ceae3
Add memory_tests/memory_test to the eager_pip package since it is imported by
...
some other package.
PiperOrigin-RevId: 256000620
2019-07-01 14:07:08 -07:00
Akshay Modi
38f0926c4f
Disable remote_memory_test in OSS to unbreak builds
...
PiperOrigin-RevId: 255480060
2019-06-27 18:37:40 -07:00
Akshay Modi
bdadd0a4fe
Stop sharding the memory tests, use different files instead.
...
Keeping all tests in a single file led to timeouts.
PiperOrigin-RevId: 255234617
2019-06-26 16:34:37 -07:00
