one_host_numpy_dataset (reference) in projects: aosp_15_r20

Searched refs:one_host_numpy_dataset (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A D	numpy_dataset.py	`72 def one_host_numpy_dataset(numpy_input, colocate_with, session): function`
H A D	one_device_strategy.py	`303 return numpy_dataset.one_host_numpy_dataset(`
H A D	parameter_server_strategy.py	`376 return numpy_dataset.one_host_numpy_dataset(`
H A D	mirrored_strategy.py	`580 return numpy_dataset.one_host_numpy_dataset(`
H A D	tpu_strategy.py	`963 return numpy_dataset.one_host_numpy_dataset(`