|
5 | 5 | from tensorflow.python.framework import ops
|
6 | 6 | from tensorflow.python.ops import variables
|
7 | 7 | from tensorflow.python.training.optimizer import Optimizer, _DenseResourceVariableProcessor, \
|
8 |
| - _OptimizableVariable, _StreamingModelPortProcessor |
| 8 | + _OptimizableVariable |
9 | 9 | import tensorflow as tf
|
10 | 10 |
|
11 | 11 | from distutils.version import LooseVersion
|
|
16 | 16 | if LooseVersion(tf.__version__) < LooseVersion("1.8.0"):
|
17 | 17 | from tensorflow.python.training.optimizer import _get_variable_for
|
18 | 18 |
|
| 19 | +if LooseVersion(tf.__version__) < LooseVersion("1.9.0"): |
| 20 | + from tensorflow.python.training.optimizer import _StreamingModelPortProcessor |
| 21 | + |
19 | 22 | import collections
|
20 | 23 |
|
21 | 24 | from TATi.models.basetype import dds_basetype
|
@@ -69,8 +72,9 @@ def _get_processor(v):
|
69 | 72 | return _DenseResourceVariableProcessor(v)
|
70 | 73 | if isinstance(v, variables.Variable):
|
71 | 74 | return RefVariableWalkerEnsembleProcessor(v)
|
72 |
| - if v.op.type == "SubmodelPort": |
73 |
| - return _StreamingModelPortProcessor(v) |
| 75 | + if LooseVersion(tf.__version__) < LooseVersion("1.9.0"): |
| 76 | + if v.op.type == "SubmodelPort": |
| 77 | + return _StreamingModelPortProcessor(v) |
74 | 78 | if LooseVersion(tf.__version__) >= LooseVersion("1.7.0"):
|
75 | 79 | if isinstance(v, ops.Tensor):
|
76 | 80 | return _TensorProcessor(v)
|
|
0 commit comments