Searched refs:select_replica (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/python/distribute/ |
D | distribute_utils_test.py | 69 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(
|
D | mirrored_run.py | 160 distribute_utils.select_replica(index, args), 161 distribute_utils.select_replica(index, kwargs)) 230 t.merge_result = distribute_utils.select_replica(r, merge_result)
|
D | tpu_strategy.py | 1024 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)])
|
D | input_lib_test.py | 143 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"]))
|
D | distribute_utils.py | 130 def select_replica(replica_id, structured): function 149 return select_replica(replica_id, structured)
|
D | collective_all_reduce_strategy_test.py | 233 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)
|
D | strategy_test_lib.py | 382 [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
|
D | parameter_server_strategy_test.py | 542 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(
|
D | mirrored_strategy.py | 771 fn(v, *distribute_utils.select_replica(i, args), 772 **distribute_utils.select_replica(i, kwargs)))
|
D | mirrored_strategy_test.py | 1052 device_result = distribute_utils.select_replica(r, result) 1053 device_expected_result = distribute_utils.select_replica(
|
D | values_test.py | 484 self.assertIs(v, distribute_utils.select_replica(0, v))
|
/external/tensorflow/tensorflow/python/tpu/ |
D | tpu_embedding_v2_correctness_test.py | 436 def select_replica(x): function 441 return nest.map_structure(select_replica, structured)
|
D | tpu_embedding_v2.py | 1300 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,
|
D | tpu_embedding_v2_test.py | 1119 def select_replica(x): function 1124 return nest.map_structure(select_replica, structured)
|