(self)
| 991 | class CreateOpTest(test_util.TensorFlowTestCase): |
| 992 | |
| 993 | def testNodeDefArgs(self): |
| 994 | g = ops.Graph() |
| 995 | op1 = g.create_op("FloatOutput", [], [dtypes.float32], None, name="myop1") |
| 996 | with g.device("/device:GPU:0"): |
| 997 | op2 = g.create_op( |
| 998 | "FloatOutputStringOutput", [], [dtypes.float32, dtypes.string], None, |
| 999 | name="myop2") |
| 1000 | op3 = g.create_op( |
| 1001 | "Foo3", |
| 1002 | [list(op1.values())[0], list(op2.values())[1], list(op2.values())[0]], |
| 1003 | [dtypes.float32, dtypes.int32], |
| 1004 | None, |
| 1005 | name="myop3") |
| 1006 | self.assertDeviceEqual(None, op1.device) |
| 1007 | self.assertDeviceEqual("/device:GPU:0", op2.device) |
| 1008 | self.assertDeviceEqual(None, op3.device) |
| 1009 | self.assertProtoEquals("name:'myop1' op:'FloatOutput'", op1.node_def) |
| 1010 | self.assertProtoEquals( |
| 1011 | "name:'myop2' op:'FloatOutputStringOutput' device:'/device:GPU:0'", |
| 1012 | op2.node_def) |
| 1013 | self.assertProtoEquals( |
| 1014 | "name:'myop3' input:'myop1' input:'myop2:1' input:'myop2' op:'Foo3'", |
| 1015 | op3.node_def) |
| 1016 | |
| 1017 | def testReferenceInput(self): |
| 1018 | g = ops.Graph() |
nothing calls this directly
no test coverage detected