Fix style errors
This commit is contained in:
parent
b857488021
commit
9225122330
tensorflow/python/kernel_tests
@ -2048,10 +2048,10 @@ cuda_py_test(
|
||||
name = "dynamic_partition_op_test",
|
||||
size = "medium",
|
||||
srcs = ["dynamic_partition_op_test.py"],
|
||||
tfrt_enabled = True,
|
||||
tags = [
|
||||
"multi_and_single_gpu",
|
||||
],
|
||||
tfrt_enabled = True,
|
||||
deps = [
|
||||
"//tensorflow/python:array_ops",
|
||||
"//tensorflow/python:client_testlib",
|
||||
|
@ -354,7 +354,7 @@ class DynamicPartitionTest(test.TestCase):
|
||||
results = []
|
||||
for device in device_list:
|
||||
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)
|
||||
result = data_flow_ops.dynamic_partition(data, partitions, 10)
|
||||
results.append(self.evaluate(result))
|
||||
|
Loading…
Reference in New Issue
Block a user