diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2016-12-05 10:50:58 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-05 11:06:55 -0800 |
| commit | 8a5dae396113655db2e4b16fe344dda4dca54994 (patch) | |
| tree | 86024e08cf159e88d020017655bbe6006fc50f65 /tensorflow/python/kernel_tests/variable_scope_test.py | |
| parent | 1e8ce51f43f5a3811740520d5a6205003e9665c2 (diff) | |
Adds VariableV2 with a sane shape_fn.
Change: 141071094
Diffstat (limited to 'tensorflow/python/kernel_tests/variable_scope_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/variable_scope_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/variable_scope_test.py b/tensorflow/python/kernel_tests/variable_scope_test.py index b46cc535cb..0c524a7f80 100644 --- a/tensorflow/python/kernel_tests/variable_scope_test.py +++ b/tensorflow/python/kernel_tests/variable_scope_test.py @@ -641,7 +641,7 @@ class VariableScopeTest(tf.test.TestCase): varname_type = [] def device_func(op): - if op.type == "Variable": + if op.type in ["Variable", "VariableV2"]: varname_type.append((op.name, op.get_attr("dtype"))) return "/gpu:0" |