Home
last modified time | relevance | path

Searched refs:select_replica (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_utils_test.py69 distribute_utils.select_replica(0, result))
71 distribute_utils.select_replica(1, result))
133 distribute_utils.select_replica(0, result))
135 distribute_utils.select_replica(1, result))
174 result_0 = distribute_utils.select_replica(0, result)
179 result_1 = distribute_utils.select_replica(1, result)
190 distribute_utils.select_replica(0, result))
235 distribute_utils.select_replica(
Dmirrored_run.py160 distribute_utils.select_replica(index, args),
161 distribute_utils.select_replica(index, kwargs))
230 t.merge_result = distribute_utils.select_replica(r, merge_result)
Dtpu_strategy.py1024 select_replica = lambda x: distribute_utils.select_replica( # pylint: disable=g-long-lambda function
1027 select_replica, per_replica_inputs),))
1296 fn(value, *distribute_utils.select_replica(i, args),
1297 **distribute_utils.select_replica(i, kwargs)))
1457 distribute_utils.select_replica(i, args),
1458 distribute_utils.select_replica(i, kwargs)])
Dinput_lib_test.py143 distribute_utils.select_replica(r, next_element)
156 [distribute_utils.select_replica(r, x) for r in range(len(devices))])
1068 distribute_utils.select_replica(0, per_replica_batch["dense"]),
1071 distribute_utils.select_replica(1, per_replica_batch["dense"]),
1076 distribute_utils.select_replica(i,
1080 distribute_utils.select_replica(
1082 distribute_utils.select_replica(i, per_replica_batch["dense"]))
1084 distribute_utils.select_replica(i,
1089 distribute_utils.select_replica(i, per_replica_batch["sparse"])),
1090 distribute_utils.select_replica(i, per_replica_batch["dense"]))
Ddistribute_utils.py130 def select_replica(replica_id, structured): function
149 return select_replica(replica_id, structured)
Dcollective_all_reduce_strategy_test.py233 computed_value = sess.run([distribute_utils.select_replica(
242 sess.run([distribute_utils.select_replica(r, next_element)
252 distribute_utils.select_replica(r, next_element)
Dstrategy_test_lib.py382 [distribute_utils.select_replica(r, next_element) for r in
392 [distribute_utils.select_replica(r, next_element) for r in
402 distribute_utils.select_replica(r, next_element) for r in
Dparameter_server_strategy_test.py542 computed_value = sess.run([distribute_utils.select_replica(
551 sess.run([distribute_utils.select_replica(r, next_element)
560 computed_value = sess.run([distribute_utils.select_replica(
Dmirrored_strategy.py771 fn(v, *distribute_utils.select_replica(i, args),
772 **distribute_utils.select_replica(i, kwargs)))
Dmirrored_strategy_test.py1052 device_result = distribute_utils.select_replica(r, result)
1053 device_expected_result = distribute_utils.select_replica(
Dvalues_test.py484 self.assertIs(v, distribute_utils.select_replica(0, v))
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_correctness_test.py436 def select_replica(x): function
441 return nest.map_structure(select_replica, structured)
Dtpu_embedding_v2.py1300 replica_inputs = distribute_utils.select_replica(replica_id,
1302 replica_weights = distribute_utils.select_replica(replica_id,
1345 tensor_list.append(distribute_utils.select_replica(replica_id,
Dtpu_embedding_v2_test.py1119 def select_replica(x): function
1124 return nest.map_structure(select_replica, structured)