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

Use integer division to compute number of updates.

Browse Source 
Change: 144675800
This commit is contained in:
Patrick Nguyen 2017-01-16 23:24:28 -08:00 committed by TensorFlower Gardener
parent 97c893338c
commit 0b56eab8de
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/kernel_tests/scatter_nd_ops_test.py
View File

@ -52,7 +52,7 @@ def _FlatOuterDims(tensor, ndims=2):
def _NumpyScatterNd(ref, indices, updates, op): def _NumpyScatterNd(ref, indices, updates, op):
ixdim = indices.shape[-1] ixdim = indices.shape[-1]
num_updates = indices.size / ixdim num_updates = indices.size // ixdim
total_nd = len(ref.shape) total_nd = len(ref.shape)
slice_size = 1 slice_size = 1
for i in range(ixdim, total_nd): for i in range(ixdim, total_nd):