Searched refs:tf_input_sizes (Results 1 – 1 of 1) sorted by relevance
4658 std::vector<int> tf_input_sizes = ok_params[i].expected_output_dims; in TEST_F() local4660 tf_input_sizes.insert(tf_input_sizes.begin(), in TEST_F()4662 QCHECK_EQ(4, tf_input_sizes.size()); in TEST_F()4663 AddTestWeights<int>("input_sizes", {4}, tf_input_sizes); in TEST_F()4667 tf_input_sizes.pop_back(); in TEST_F()4669 tf_input_sizes.erase(tf_input_sizes.begin()); in TEST_F()4671 QCHECK_EQ(2, tf_input_sizes.size()); in TEST_F()4672 AddTestWeights<int>("input_sizes", {2}, tf_input_sizes); in TEST_F()