| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | custom_training_loop_input_test.py | 388 distribution.distribute_datasets_from_function(
403 distribution.distribute_datasets_from_function(
441 distribution.distribute_datasets_from_function(
477 distribution.distribute_datasets_from_function(
673 distribution.distribute_datasets_from_function(dataset_fn))
701 distribution.distribute_datasets_from_function(dataset_fn))
755 distribution.distribute_datasets_from_function(dataset_fn))
781 distribution.distribute_datasets_from_function(dataset_fn))
966 distribution.distribute_datasets_from_function(dataset))
|
| H A D | distributed_table_test.py | 167 return strategy.distribute_datasets_from_function(dataset_fn)
210 return strategy.distribute_datasets_from_function(dataset_fn)
251 return strategy.distribute_datasets_from_function(dataset_fn)
295 return strategy.distribute_datasets_from_function(dataset_fn)
|
| H A D | one_device_strategy.py | 110 def distribute_datasets_from_function( member in OneDeviceStrategy
152 self).distribute_datasets_from_function(dataset_fn, options)
|
| H A D | tpu_strategy_test.py | 898 strategy.distribute_datasets_from_function(
948 strategy.distribute_datasets_from_function(
1003 strategy.distribute_datasets_from_function(
1232 iter(strategy.distribute_datasets_from_function(dataset_fn))
1247 iter(strategy.distribute_datasets_from_function(dataset_fn))
|
| H A D | parameter_server_strategy.py | 131 def distribute_datasets_from_function(self, dataset_fn, options=None): member in ParameterServerStrategyV1
140 super(ParameterServerStrategyV1, self).distribute_datasets_from_function(
|
| H A D | distribute_lib_test.py | 556 default_strategy.distribute_datasets_from_function(
563 default_strategy.distribute_datasets_from_function(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | cluster_coordinator_test.py | 1308 return self.strategy.distribute_datasets_from_function(dataset_fn)
1335 distributed_dataset = self.strategy.distribute_datasets_from_function(
1366 return self.strategy.distribute_datasets_from_function(
1467 return self.strategy.distribute_datasets_from_function(
1485 return self.strategy.distribute_datasets_from_function(
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.distribute.-strategy.pbtxt | 31 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-one-device-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 36 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-mirrored-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 33 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-mirrored-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-one-device-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-strategy.pbtxt | 31 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-parameter-server-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| H A D | tensorflow.distribute.-t-p-u-strategy.pbtxt | 32 name: "distribute_datasets_from_function"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v2_optimizer_test.py | 88 dist = strategy.distribute_datasets_from_function(
|