get_distributed_datasets_from_function (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A D	input_util.py	`78 def get_distributed_datasets_from_function(dataset_fn, function`
H A D	one_device_strategy.py	`334 return input_util.get_distributed_datasets_from_function(`
H A D	parameter_server_strategy_v2.py	`899 return input_util.get_distributed_datasets_from_function(`
H A D	parameter_server_strategy.py	`394 return input_util.get_distributed_datasets_from_function(`
H A D	mirrored_strategy.py	`593 return input_util.get_distributed_datasets_from_function(`
H A D	collective_all_reduce_strategy.py	`676 return input_util.get_distributed_datasets_from_function(`
H A D	input_lib_test.py	`1858 dist_dataset = input_util.get_distributed_datasets_from_function( 1922 dist_dataset = input_util.get_distributed_datasets_from_function(`
H A D	tpu_strategy.py	`1025 distributed_dataset = input_util.get_distributed_datasets_from_function(`