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

Fix style errors

Browse Source
This commit is contained in:
drebain 2020-08-25 15:31:41 -07:00
parent b857488021
commit 9225122330
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/kernel_tests/BUILD
View File

@ -2048,10 +2048,10 @@ cuda_py_test(
name = "dynamic_partition_op_test", name = "dynamic_partition_op_test",
size = "medium", size = "medium",
srcs = ["dynamic_partition_op_test.py"], srcs = ["dynamic_partition_op_test.py"],
tfrt_enabled = True,
tags = [ tags = [
"multi_and_single_gpu", "multi_and_single_gpu",
], ],
tfrt_enabled = True,
deps = [ deps = [
"//tensorflow/python:array_ops", "//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",

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

@ -354,7 +354,7 @@ class DynamicPartitionTest(test.TestCase):
results = [] results = []
for device in device_list: for device in device_list:
with ops.device(device.name): with ops.device(device.name):
data = constant_op.constant(np.zeros((1000, ))) data = constant_op.constant(np.zeros((1000,)))
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))