diff options
| author |  Skye Wanderman-Milne <skyewm@google.com> | 2018-02-09 14:25:28 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-09 14:31:52 -0800 |
| commit | ed5f003cc2c542c3c545369f71d4b57429da33fc (patch) | |
| tree | f10da814c2e75a3c09b0759f25de189d8af51395 /tensorflow/python/saved_model | |
| parent | 68a767e0aaa6fb3f01f5ce8f4fe533f36fcf7f82 (diff) | |
Make import_graph_def add default attr values with the C API enabled.
It turns out that the original Python code modifies the graph_def argument
to add default attr values. I'm not sure if the behavior is covered by
our API guarantees since it's not documented, but let's keep the behavior
consistent for now.
PiperOrigin-RevId: 185193037
Diffstat (limited to 'tensorflow/python/saved_model')
| -rw-r--r-- | tensorflow/python/saved_model/saved_model_test.py | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/tensorflow/python/saved_model/saved_model_test.py b/tensorflow/python/saved_model/saved_model_test.py index d1f6bc27ef..d9d3168825 100644 --- a/tensorflow/python/saved_model/saved_model_test.py +++ b/tensorflow/python/saved_model/saved_model_test.py @@ -873,8 +873,6 @@ class SavedModelTest(test.TestCase): 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval()) def testStripDefaultAttrs(self): - if ops._USE_C_API: return # TODO(skyewm): get this working - export_dir = self._get_export_dir("test_strip_default_attrs") builder = saved_model_builder.SavedModelBuilder(export_dir) |