(self)
| 1502 | """, gd) |
| 1503 | |
| 1504 | def testNestingString(self): |
| 1505 | g = ops.Graph() |
| 1506 | with g.device("/job:worker/replica:2"): |
| 1507 | g.create_op("FloatOutput", [], [dtypes.float32]) |
| 1508 | with g.device("/job:worker/replica:3/task:0"): |
| 1509 | g.create_op("FloatOutput", [], [dtypes.float32]) |
| 1510 | g.create_op("FloatOutput", [], [dtypes.float32]) |
| 1511 | gd = g.as_graph_def() |
| 1512 | self.assertProtoEqualsVersion(""" |
| 1513 | node { name: "FloatOutput" op: "FloatOutput" |
| 1514 | device: "/job:worker/replica:2" } |
| 1515 | node { name: "FloatOutput_1" op: "FloatOutput" |
| 1516 | device: "/job:worker/replica:3/task:0" } |
| 1517 | node { name: "FloatOutput_2" op: "FloatOutput" |
| 1518 | device: "/job:worker/replica:2" } |
| 1519 | """, gd) |
| 1520 | |
| 1521 | def testNestingOverrideGpuCpu(self): |
| 1522 | g = ops.Graph() |
nothing calls this directly
no test coverage detected