| 128 | } |
| 129 | |
| 130 | Output Combine(const Scope& scope, const string &device, Input a, Input b) { |
| 131 | return ConstructOp(scope, "Combine", device, {a, b}); |
| 132 | } |
| 133 | |
| 134 | TEST_F(DistGraphPartitionTest, testSplitSubGraph) { |
| 135 | string worker_device = "/job:worker/replica:0/task:0/cpu:0"; |