Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dmirrored_function_strategy.py126 return_values.append(fn(*values.select_replica(index, args),
127 **values.select_replica(index, kwargs)))
129 return_values.append(fn(*values.select_replica(index, args),
130 **values.select_replica(index, kwargs)))
Dinput_lib_test.py179 [values.select_replica(r,
188 [values.select_replica(r,
200 [values.select_replica(r,
210 [values.select_replica(r, x) for r in range(len(devices))])
562 values.select_replica(0, per_replica_batch["dense"]),
565 values.select_replica(1, per_replica_batch["dense"]),
570 values.select_replica(i, per_replica_batch["ragged"]).values, 6)
572 values.select_replica(i, per_replica_batch["ragged"]).to_tensor(),
573 values.select_replica(i, per_replica_batch["dense"]))
575 values.select_replica(i, per_replica_batch["sparse"]).indices, 6)
[all …]
Dtpu_strategy.py419 …select_replica = lambda x: values.select_replica(replica_id, x) # pylint: disable=cell-var-from-l… function
421 select_replica, per_replica_inputs),))
791 values.select_replica(i, args),
792 values.select_replica(i, kwargs)])
Dvalues_test.py206 values.select_replica(0, result))
208 values.select_replica(1, result))
247 values.select_replica(0, result))
249 values.select_replica(1, result))
279 result_0 = values.select_replica(0, result)
284 result_1 = values.select_replica(1, result)
295 values.select_replica(0, result))
350 values.select_replica(device_id,
723 self.assertIs(v, values.select_replica(0, v))
Dcollective_all_reduce_strategy_test.py365 computed_value = sess.run([values.select_replica(r, next_element)
374 sess.run([values.select_replica(r, next_element)
383 computed_value = sess.run([values.select_replica(r, next_element)
Dmirrored_strategy.py129 values.select_replica(index, args),
130 values.select_replica(index, kwargs))
197 t.merge_result = values.select_replica(r, merge_result)
Dstrategy_test_lib.py352 [values.select_replica(r, next_element) for r in range(len(devices))])
361 [values.select_replica(r, next_element) for r in range(len(devices))])
370 values.select_replica(r, next_element) for r in range(len(devices))
Dparameter_server_strategy_test.py535 computed_value = sess.run([values.select_replica(r, next_element)
544 sess.run([values.select_replica(r, next_element)
553 computed_value = sess.run([values.select_replica(r, next_element)
Dmirrored_strategy_test.py983 device_result = values.select_replica(r, result)
984 device_expected_result = values.select_replica(r, expected_result)
Dvalues.py1246 def select_replica(replica_id, structured): function