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

Tensor tracer python3 compatibility.

Browse Source 
PiperOrigin-RevId: 314863876
Change-Id: Iec44c5bf59eb8eac3615e083a72944a2bd06c354
This commit is contained in:
Mehmet Deveci 2020-06-04 21:55:32 -07:00 committed by TensorFlower Gardener
parent 75c40f6bff
commit 17834a43d1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/tpu/tensor_tracer.py
View File

@ -1844,7 +1844,7 @@ class TensorTracer(object):
if len(processed_tensors) != 1: if len(processed_tensors) != 1:
raise RuntimeError('Multiple stats are only allowed in compact ' raise RuntimeError('Multiple stats are only allowed in compact '
'mode.') 'mode.')
processed_out_tensor = processed_tensors.values()[0] processed_out_tensor = list(processed_tensors.values())[0]
# Store the whole tensor in a buffer. # Store the whole tensor in a buffer.
trace_op = self._snapshot_tensor(processed_out_tensor) trace_op = self._snapshot_tensor(processed_out_tensor)
else: else: