experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix benchmark regression.

Browse Source 
PiperOrigin-RevId: 215278033
This commit is contained in:
Alexandre Passos 2018-10-01 14:14:32 -07:00 committed by TensorFlower Gardener
parent d7edbeb8dc
commit 094e1953b7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/ops/conv2d_benchmark.py
View File

@ -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 = []