| 22 | namespace tensorflow { |
| 23 | |
| 24 | TEST(InputPipelineOpsTest, ObtainNext_InvalidNumberOfInputs) { |
| 25 | ShapeInferenceTestOp op("ObtainNext"); |
| 26 | op.input_tensors.resize(3); |
| 27 | INFER_ERROR("Wrong number of inputs passed", op, "?;?;?"); |
| 28 | } |
| 29 | |
| 30 | TEST(InputPipelineOpsTest, ObtainNext) { |
| 31 | ShapeInferenceTestOp op("ObtainNext"); |