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

correct pylint formattings

Browse Source
This commit is contained in:
kushanam 2020-11-10 16:01:21 -08:00 committed by Geeta Chavan
parent 309350e9de
commit d3e2a16524
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/distribute/input_lib.py
View File

@ -2003,9 +2003,9 @@ def _get_dataset_attributes(dataset):
def _should_use_multi_device_iterator(options):
"""Determine whether to use multi_device_iterator_ops.OwnedMultiDeviceIterator"""
if (options is None
or options.experimental_replication_mode == InputReplicationMode.PER_WORKER
or (options.experimental_replication_mode == InputReplicationMode.PER_REPLICA
and options.experimental_prefetch_to_device)):
or options.experimental_replication_mode == InputReplicationMode.PER_WORKER
or (options.experimental_replication_mode == InputReplicationMode.PER_REPLICA
and options.experimental_prefetch_to_device)):
return True
return False

8
tensorflow/python/distribute/input_lib_type_spec_test.py
View File

@ -460,7 +460,7 @@ class InputTypeSpecTest(test.TestCase, parameterized.TestCase):
strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
],
enable_get_next_as_optional=[True, False],
experimental_place_dataset_on_device=[True,False],
experimental_place_dataset_on_device=[True, False],
experimental_prefetch_to_device=[True, False],))
def testFromFunctionInputSignatureForPerReplicaValuesWithOptions(self, distribution,
enable_get_next_as_optional,
@ -486,9 +486,9 @@ class InputTypeSpecTest(test.TestCase, parameterized.TestCase):
input_context.get_per_replica_batch_size(4))
options = distribute_lib.InputOptions(
experimental_place_dataset_on_device = experimental_place_dataset_on_device,
experimental_prefetch_to_device = experimental_prefetch_to_device,
experimental_replication_mode = distribute_lib.InputReplicationMode.PER_REPLICA)
experimental_place_dataset_on_device=experimental_place_dataset_on_device,
experimental_prefetch_to_device=experimental_prefetch_to_device,
experimental_replication_mode=distribute_lib.InputReplicationMode.PER_REPLICA)
distribution.extended.experimental_enable_get_next_as_optional = (
enable_get_next_as_optional)