Searched refs:get_dataset_from_tensor_slices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_input_test.py | 45 def get_dataset_from_tensor_slices(inp_array): function
127 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
145 dataset = get_dataset_from_tensor_slices(data).batch(2)
189 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
206 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
226 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
252 dataset = get_dataset_from_tensor_slices([0, 1, 2, 3]).batch(2)
281 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
325 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
391 lambda _: get_dataset_from_tensor_slices(data)))
[all …]
|
| D | custom_training_loop_gradient_test.py | 34 def get_dataset_from_tensor_slices(inp_array): function
74 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
100 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
|