emailweixu-patch-1
Yang Yu 7 years ago
parent b397febcb5
commit 0f8489c42e

@ -175,8 +175,10 @@ class ParallelOpTest(BaseParallelForTest):
class ParallelOpTestMultipleInput(BaseParallelForTest): class ParallelOpTestMultipleInput(BaseParallelForTest):
@staticmethod @staticmethod
def __network__(): def __network__():
x = fluid.layers.data(shape=[784], dtype='float32', name='img1', stop_gradient=False) x = fluid.layers.data(
y = fluid.layers.data(shape=[784], dtype='float32', name='img2', stop_gradient=False) shape=[784], dtype='float32', name='img1', stop_gradient=False)
y = fluid.layers.data(
shape=[784], dtype='float32', name='img2', stop_gradient=False)
yield [x, y] yield [x, y]
x = x + y x = x + y
hidden1 = fluid.layers.fc(input=x, size=200, param_attr='fc1.w') hidden1 = fluid.layers.fc(input=x, size=200, param_attr='fc1.w')

Loading…
Cancel
Save