Home
last modified time | relevance | path

Searched refs:create_per_worker_dataset (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dcluster_coordinator_test.py552 distributed_dataset = self.coordinator.create_per_worker_dataset(input_fn)
581 distributed_dataset = self.coordinator.create_per_worker_dataset(input_fn)
626 distributed_dataset = self.coordinator.create_per_worker_dataset(input_fn)
628 distributed_dataset = self.coordinator.create_per_worker_dataset(
643 self.coordinator.create_per_worker_dataset(input_fn)
657 distributed_dataset = self.coordinator.create_per_worker_dataset(input_fn)
659 distributed_dataset = self.coordinator.create_per_worker_dataset(
750 per_worker_dataset = self.coordinator.create_per_worker_dataset(input_fn)
765 per_worker_dataset = self.coordinator.create_per_worker_dataset(dataset)
781 per_worker_dataset1 = self.coordinator.create_per_worker_dataset(input_fn1)
[all …]
H A Dfault_tolerance_test.py84 self.cluster_coord.create_per_worker_dataset(distribute_dataset_fn))
86 self.cluster_coord.create_per_worker_dataset(distribute_dataset_fn))
92 self.cluster_coord.create_per_worker_dataset(dataset))
94 self.cluster_coord.create_per_worker_dataset(dataset))
H A Dcluster_coordinator.py1284 def create_per_worker_dataset(self, dataset_fn): member in ClusterCoordinator
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistributed_table_test.py169 per_worker_dataset = coordinator.create_per_worker_dataset(
212 per_worker_dataset = coordinator.create_per_worker_dataset(
253 per_worker_dataset = coordinator.create_per_worker_dataset(
297 per_worker_dataset = coordinator.create_per_worker_dataset(
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.distribute.coordinator.-cluster-coordinator.pbtxt14 name: "create_per_worker_dataset"
H A Dtensorflow.distribute.experimental.coordinator.-cluster-coordinator.pbtxt14 name: "create_per_worker_dataset"
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Ddata_adapter.py1378 …self._dataset = self._model._cluster_coordinator.create_per_worker_dataset( # pylint: disable=pro…