def forward(self, x, out=None):
inputs = [x]
self.unify_devices(inputs)
outputs = [out] if out else [_ReferenceTensor(x)]
return self.run(inputs, outputs)
def forward(self, x, shape):
inputs = [x]
self.unify_devices(inputs)
outputs = [_ReferenceTensor(x)]
callback = lambda A: self.update_arguments(A, shape)
return self.run(inputs, outputs, callback=callback)