Searched refs:get_dataset_from_tensor_slices (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | custom_training_loop_input_test.py | 43 def get_dataset_from_tensor_slices(inp_array): function
125 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
143 dataset = get_dataset_from_tensor_slices(data).batch(2)
187 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
204 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
224 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
250 dataset = get_dataset_from_tensor_slices([0, 1, 2, 3]).batch(2)
279 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
323 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
389 lambda _: get_dataset_from_tensor_slices(data)))
[all …]
|
| H A D | custom_training_loop_gradient_test.py | 30 def get_dataset_from_tensor_slices(inp_array): function
70 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
96 dataset = get_dataset_from_tensor_slices([5., 6., 7., 8.]).batch(2)
|