Searched refs:set_input_specs (Results 1 – 2 of 2) sorted by relevance
65 f.set_input_specs({create_test_input_spec({size})}); in TEST()85 f.set_input_specs({create_test_input_spec({1, 3, 224, 224})}); in TEST()124 f.set_input_specs({create_test_input_spec({size})}); in TEST()138 f.set_input_specs({create_test_input_spec({size})}); in TEST()
151 void set_input_specs(const std::vector<InputSpec>& input_specs) { in set_input_specs() function