Home
last modified time | relevance | path

Searched refs:experimental_fetch_to_device (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/tpu/tests/
Dtpu_embedding_v2_hd_valid_input_test.py41 experimental_fetch_to_device=False)))
48 experimental_fetch_to_device=False)))
55 experimental_fetch_to_device=False)))
84 experimental_fetch_to_device=False)))
95 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_valid_input_test.py51 experimental_fetch_to_device=False))))
102 experimental_fetch_to_device=False)))
107 experimental_fetch_to_device=False)))
131 experimental_fetch_to_device=False)))
188 experimental_fetch_to_device=False)))
273 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_enqueue_mode_test.py48 experimental_fetch_to_device=False)))
87 experimental_fetch_to_device=False)))
120 experimental_fetch_to_device=False)))
148 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_invalid_input_test.py84 experimental_fetch_to_device=False)))
108 experimental_fetch_to_device=False)))
113 experimental_fetch_to_device=False)))
152 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_sequence_feature_test.py43 experimental_fetch_to_device=False)))
74 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_hd_invalid_input_test.py45 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_correctness_dense_lookup_test.py39 options=distribute_lib.InputOptions(experimental_fetch_to_device=False))
Dtpu_embedding_v2_correctness_sequence_feature_test.py65 experimental_fetch_to_device=False))))
Dtpu_embedding_v2_correctness_base_test.py60 options=distribute_lib.InputOptions(experimental_fetch_to_device=False))
Dtpu_embedding_v1_correctness_test.py133 options=distribute_lib.InputOptions(experimental_fetch_to_device=False))
282 experimental_fetch_to_device=False)))
Dtpu_embedding_v2_optimizer_test.py90 options=distribute_lib.InputOptions(experimental_fetch_to_device=False))
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py267 experimental_fetch_to_device=[True, False],
271 experimental_place_dataset_on_device, experimental_fetch_to_device): argument
273 if experimental_place_dataset_on_device and experimental_fetch_to_device:
295 experimental_fetch_to_device=experimental_fetch_to_device,
469 experimental_fetch_to_device=[True, False]))
472 experimental_fetch_to_device): argument
480 experimental_fetch_to_device=experimental_fetch_to_device)
518 experimental_fetch_to_device=[True, False],
522 experimental_place_dataset_on_device, experimental_fetch_to_device): argument
524 if experimental_place_dataset_on_device and experimental_fetch_to_device:
[all …]
Done_device_strategy_test.py121 experimental_fetch_to_device=True)
142 experimental_fetch_to_device=False)
Dinput_lib_test.py1533 experimental_fetch_to_device=True,
1538 experimental_fetch_to_device=True,
1579 experimental_fetch_to_device=False,
1611 experimental_fetch_to_device=False,
1616 experimental_fetch_to_device=True,
1644 experimental_fetch_to_device=False,
1648 experimental_fetch_to_device=True,
1679 experimental_fetch_to_device=False,
1684 experimental_fetch_to_device=True,
1717 experimental_fetch_to_device=False,
[all …]
Dinput_util.py124 options.experimental_fetch_to_device and
Dtpu_strategy_test.py900 distribute_lib.InputOptions(experimental_fetch_to_device=False)))
950 distribute_lib.InputOptions(experimental_fetch_to_device=False)))
1006 experimental_fetch_to_device=False)))
1171 experimental_fetch_to_device=True)
1186 experimental_fetch_to_device=False)
Ddistribute_lib.py704 experimental_fetch_to_device=None, argument
708 if experimental_fetch_to_device is None:
709 experimental_fetch_to_device = True
712 cls).__new__(cls, experimental_fetch_to_device,
Done_device_strategy.py262 if not options or options.experimental_fetch_to_device:
Dtpu_strategy.py968 if not options or options.experimental_fetch_to_device:
998 if options is None or options.experimental_fetch_to_device:
1033 if options is None or options.experimental_fetch_to_device:
Dcollective_all_reduce_strategy_test.py318 experimental_fetch_to_device=prefetch_to_device)
337 experimental_fetch_to_device=False)
Dcustom_training_loop_input_test.py460 distribute_lib.InputOptions(experimental_fetch_to_device=False)))
479 distribute_lib.InputOptions(experimental_fetch_to_device=False)))
Dparameter_server_strategy_test.py781 experimental_fetch_to_device=prefetch_to_device)
804 experimental_fetch_to_device=False)
Dparameter_server_strategy.py324 if not options or options.experimental_fetch_to_device:
Dinput_lib.py1603 not self._options.experimental_fetch_to_device):
1949 and options.experimental_fetch_to_device)):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-input-options.pbtxt7 name: "experimental_fetch_to_device"

12