diff options
author | 2018-10-01 14:14:32 -0700 | |
---|---|---|
committer | 2018-10-01 14:22:33 -0700 | |
commit | 094e1953b7df0bbb9bd4d0e3329b3b4611edf984 (patch) | |
tree | afa17a59a28a0ca91afcd68e1fc826c603b97b38 /tensorflow/python/ops | |
parent | d7edbeb8dcc81a9cabc922ae46f549fe6b498eb9 (diff) |
Fix benchmark regression.
PiperOrigin-RevId: 215278033
Diffstat (limited to 'tensorflow/python/ops')
-rw-r--r-- | tensorflow/python/ops/conv2d_benchmark.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/ops/conv2d_benchmark.py b/tensorflow/python/ops/conv2d_benchmark.py index 28111c2730..f40488afbe 100644 --- a/tensorflow/python/ops/conv2d_benchmark.py +++ b/tensorflow/python/ops/conv2d_benchmark.py @@ -63,9 +63,9 @@ def build_graph(device, dtype, data_format, input_shape, filter_shape, strides, An array of tensors to run() """ with ops.device("/%s:0" % device): - inp = variables.Variable( + inp = variables.VariableV1( random_ops.truncated_normal(input_shape, dtype=dtype)) - filt = variables.Variable( + filt = variables.VariableV1( random_ops.truncated_normal(filter_shape, dtype=dtype)) outputs = [] |