Lines Matching defs:ins
49 string RunInferShapes(const string& op_name, const string& ins,
52 const int num_inputs = 1 + std::count(ins.begin(), ins.end(), ';');
62 return ShapeInferenceTestutil::InferShapes(op, ins, expected_outs)
144 const string ins = "[0,1,?];[2];[1]";
145 EXPECT_CONTAINS(RunInferShapes(op, ins, "[?,2,?,d2_0]", fn),
147 EXPECT_CONTAINS(RunInferShapes(op, ins, "[0,2,?,d2_0]", fn),
150 RunInferShapes(op, ins, "[d0_0,2,?,d2_0]", fn),
152 EXPECT_CONTAINS(RunInferShapes(op, ins, "[x,2,?,d2_0]", fn),
155 EXPECT_CONTAINS(RunInferShapes(op, ins, "[d0_0|d0_2,2,?,d2_0]", fn),
158 EXPECT_CONTAINS(RunInferShapes(op, ins, "[d0_1,?,?,d0_0|d2_0]", fn),
163 RunInferShapes(op, ins, "[d0_1,2,8,d0_0|d2_0]", fn));
164 EXPECT_CONTAINS(RunInferShapes(op, ins, "[d0_1,2,d0_1|d2_0,d0_0|d2_0]", fn),
167 RunInferShapes(op, ins, "[d0_1,2,?,d0_0|d2_0]", fn));
