Searched refs:partial_input_shape (Results 1 – 1 of 1) sorted by relevance
2239 std::vector<int> partial_input_shape; in TEST_P() local2243 partial_input_shape.resize(4, -1); in TEST_P()2244 partial_input_shape[1] = node_input[i].dims[1]; in TEST_P()2247 node_input[i].val, partial_input_shape, in TEST_P()4546 std::vector<int> partial_input_shape; in TEST_P() local4549 partial_input_shape.resize(ok_params[i].input_dims.size(), -1); in TEST_P()4551 partial_input_shape[channel_id] = ok_params[i].input_dims[channel_id]; in TEST_P()4555 ok_params[i].input, partial_input_shape); in TEST_P()