Home
last modified time | relevance | path

Searched defs:IteratorSaveAndRestoreTestCase (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dsparse_tensor_slice_dataset_op_test.cc308 struct IteratorSaveAndRestoreTestCase { struct
309 T dataset_params;
310 std::vector<int> breakpoints;
311 std::vector<std::vector<Tensor>> expected_outputs;
H A Dshuffle_dataset_op_test.cc457 struct IteratorSaveAndRestoreTestCase { struct
458 T dataset_params;
459 std::vector<int> breakpoints;
460 std::vector<Tensor> expected_shuffle_outputs;
H A Dwindow_dataset_op_test.cc407 struct IteratorSaveAndRestoreTestCase { struct
408 T dataset_params;
409 std::vector<int> breakpoints;
410 std::vector<std::vector<Tensor>> expected_outputs;
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.h504 struct IteratorSaveAndRestoreTestCase { struct
508 : dataset_params(std::move(dataset_params)), in dataset_params()
513 T dataset_params;
514 std::vector<int> breakpoints;
515 std::vector<Tensor> expected_outputs;
516 bool compare_order;