Home
last modified time | relevance | path

Searched refs:experimental_distribute_values_from_function (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dstrategy_gather_test.py72 distributed_values = strategy.experimental_distribute_values_from_function(
130 distributed_values = strategy.experimental_distribute_values_from_function(
152 distributed_values = strategy.experimental_distribute_values_from_function(
180 distributed_values = strategy.experimental_distribute_values_from_function(
227 distributed_values = strategy.experimental_distribute_values_from_function(
272 per_replica_value = strategy.experimental_distribute_values_from_function(
367 per_replica_value = strategy.experimental_distribute_values_from_function(
395 per_replica_value = strategy.experimental_distribute_values_from_function(
424 per_replica_value = strategy.experimental_distribute_values_from_function(
492 per_replica_value = strategy.experimental_distribute_values_from_function(
[all …]
H A Dvalues_test.py85 distribution.experimental_distribute_values_from_function(value_fn))
105 distribution.experimental_distribute_values_from_function(value_fn))
124 distribution.experimental_distribute_values_from_function(value_fn))
149 distribution.experimental_distribute_values_from_function(value_fn))
174 distribution.experimental_distribute_values_from_function(value_fn))
194 distribution.experimental_distribute_values_from_function(value_fn))
215 distribution.experimental_distribute_values_from_function(value_fn))
248 distribution.experimental_distribute_values_from_function(value_fn))
278 distribution.experimental_distribute_values_from_function(value_fn))
304 distribution.experimental_distribute_values_from_function(value_fn))
H A Dstrategy_common_test.py225 per_replica_value = strategy.experimental_distribute_values_from_function(
H A Ddistribute_lib.py1680 def experimental_distribute_values_from_function(self, value_fn): member in Strategy
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt45 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.-mirrored-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.-one-device-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.-strategy.pbtxt43 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.-parameter-server-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt44 name: "experimental_distribute_values_from_function"
H A Dtensorflow.distribute.-t-p-u-strategy.pbtxt48 name: "experimental_distribute_values_from_function"
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v2_valid_input_test.py152 data = strategy.experimental_distribute_values_from_function(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/
H A Dsaved_model_test.py359 value = strategy.experimental_distribute_values_from_function(