Searched refs:num_tensors_per_slice (Results 1 – 2 of 2) sorted by relevance
135 size_t num_tensors_per_slice = components.size(); in TEST_F() local147 for (int i = 0; i < num_tensors_per_slice; ++i) { in TEST_F()178 EXPECT_LT(i + num_tensors_per_slice * cur_slice, in TEST_F()186 expected_outputs[i + num_tensors_per_slice * cur_slice] in TEST_F()193 expected_outputs[i + num_tensors_per_slice * cur_slice])); in TEST_F()234 size_t num_tensors_per_slice = components.size(); in TEST_F() local246 for (int i = 0; i < num_tensors_per_slice; ++i) { in TEST_F()265 EXPECT_EQ(produced_output_dtypes.size(), num_tensors_per_slice); in TEST_F()266 for (int i = 0; i < num_tensors_per_slice; ++i) { in TEST_F()277 size_t num_tensors_per_slice = components.size(); in TEST_F() local[all …]
133 int num_tensors_per_slice = test_case.input_range_dataset_params.size(); in CreateDatasetFromTestCase() local135 {{num_tensors_per_slice}}, in CreateDatasetFromTestCase()202 int num_tensors_per_slice = test_case.input_range_dataset_params.size(); in TEST_P() local206 expected_output_dtypes.reserve(num_tensors_per_slice); in TEST_P()207 for (int i = 0; i < num_tensors_per_slice; ++i) { in TEST_P()220 int num_tensors_per_slice = test_case.input_range_dataset_params.size(); in TEST_P() local224 expected_output_shapes.reserve(num_tensors_per_slice); in TEST_P()225 for (int i = 0; i < num_tensors_per_slice; ++i) { in TEST_P()238 int num_tensors_per_slice = test_case.input_range_dataset_params.size(); in TEST_P() local242 test_case.expected_outputs.size() / num_tensors_per_slice); in TEST_P()[all …]