get_distributed_dataset (reference) in projects: aosp_15_r20

Searched refs:get_distributed_dataset (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A D	input_util.py	`22 def get_distributed_dataset(dataset, function`
H A D	one_device_strategy.py	`320 return input_util.get_distributed_dataset(`
H A D	input_lib_test.py	`299 dist_dataset = input_util.get_distributed_dataset( 499 dist_dataset = input_util.get_distributed_dataset(dataset, input_workers, 1800 dist_dataset = input_util.get_distributed_dataset(dataset, input_workers,`
H A D	parameter_server_strategy_v2.py	`876 return input_util.get_distributed_dataset(`
H A D	parameter_server_strategy.py	`340 return input_util.get_distributed_dataset(`
H A D	mirrored_strategy.py	`572 return input_util.get_distributed_dataset(`
H A D	collective_all_reduce_strategy.py	`659 return input_util.get_distributed_dataset(`
H A D	tpu_strategy.py	`1001 return input_util.get_distributed_dataset(`