Searched refs:select_replica (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | input_lib_test.py | 77 [values.select_replica(r, next_element) for r in range(len(devices))])
85 [values.select_replica(r, next_element) for r in range(len(devices))])
93 [values.select_replica(r, next_element) for r in range(len(devices))])
|
| D | values_test.py | 225 values.select_replica(0, result))
227 values.select_replica(1, result))
258 values.select_replica(0, result))
260 values.select_replica(1, result))
284 result_0 = values.select_replica(0, result)
289 result_1 = values.select_replica(1, result)
301 values.select_replica(0, result))
345 values.select_replica(device_id,
|
| D | strategy_test_lib.py | 313 [values.select_replica(r, next_element) for r in range(len(devices))])
319 [values.select_replica(r, next_element) for r in range(len(devices))])
327 computed_value = evaluate([values.select_replica(r, next_element)
|
| D | collective_all_reduce_strategy_test.py | 328 computed_value = sess.run([values.select_replica(r, next_element)
334 sess.run([values.select_replica(r, next_element)
343 computed_value = sess.run([values.select_replica(r, next_element)
|
| D | parameter_server_strategy_test.py | 588 computed_value = sess.run([values.select_replica(r, next_element)
594 sess.run([values.select_replica(r, next_element)
603 computed_value = sess.run([values.select_replica(r, next_element)
|
| D | mirrored_strategy_multigpu_test.py | 1309 device_result = values.select_replica(r, result)
1310 device_expected_result = values.select_replica(r, expected_result)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 177 values.select_replica(i, args),
178 values.select_replica(i, kwargs)])
326 …select_replica = lambda x: values.select_replica(replica_id, x) # pylint: disable=cell-var-from-l… function
328 select_replica, per_replica_inputs),))
|
| D | mirrored_strategy.py | 121 values.select_replica(index, args),
122 values.select_replica(index, kwargs))
189 t.merge_result = values.select_replica(r, merge_result)
|
| D | values.py | 1385 def select_replica(replica_id, structured): function
|