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

Make testMultiGPU pass when on CPU only

Browse Source
This commit is contained in:
drebain 2020-08-24 12:12:55 -07:00
parent be418f2918
commit b857488021
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/kernel_tests/dynamic_partition_op_test.py
View File

@ -358,7 +358,8 @@ class DynamicPartitionTest(test.TestCase):
partitions = constant_op.constant(np.arange(1000, dtype=np.int32) % 10) partitions = constant_op.constant(np.arange(1000, dtype=np.int32) % 10)
result = data_flow_ops.dynamic_partition(data, partitions, 10) result = data_flow_ops.dynamic_partition(data, partitions, 10)
results.append(self.evaluate(result)) results.append(self.evaluate(result))
self.assertAllEqual(results, np.zeros((len(device_list), 10, 100))) if device_list:
self.assertAllEqual(results, np.zeros((len(device_list), 10, 100)))
if __name__ == "__main__": if __name__ == "__main__":