Home
last modified time | relevance | path

Searched refs:num_tensors_per_slice (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc105 size_t num_tensors_per_slice = input_names.size(); in TEST_P() local
114 EXPECT_LT(i + num_tensors_per_slice * cur_slice, in TEST_P()
121 test_case.expected_outputs[i + num_tensors_per_slice * cur_slice] in TEST_P()
128 test_case.expected_outputs[i + num_tensors_per_slice * cur_slice])); in TEST_P()
273 size_t num_tensors_per_slice = params.num_tensors_per_slice(); in TEST_P() local
293 num_tensors_per_slice * (cur_iteration - 1)] in TEST_P()
300 test_case.expected_outputs[i + num_tensors_per_slice * in TEST_P()
Ddataset_test_base.h300 size_t num_tensors_per_slice() const { return components_.size(); } in num_tensors_per_slice() function