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 | 47 def get_dataset_from_tensor_slices(inp_array): function
129 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
147 dataset = get_dataset_from_tensor_slices(data).batch(2)
191 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
208 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
228 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
254 dataset = get_dataset_from_tensor_slices([0, 1, 2, 3]).batch(2)
283 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
327 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
393 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)
|