Searched refs:get_input_datasets (Results 1 – 1 of 1) sorted by relevance
30 def get_input_datasets(use_bfloat16=False): function111 train_ds, eval_ds, input_shape = get_input_datasets()