def initialize(self, device, **kwargs):
if (self.input is None or self.input.mem is None or
(self.output is not None and
eq_addr(self.input.mem, self.output.mem))):
raise error.BadFormatError(
"input should be set and should not be equal to output")
super(BackwardLog, self).initialize(device=device, **kwargs)
def initialize(self, device, **kwargs):
if (not self.input or (self.output is not None
and eq_addr(self.input.mem, self.output.mem))):
raise error.BadFormatError(
"input should be set and should not be equal to output")
super(BackwardTanhLog, self).initialize(device=device, **kwargs)
self.output.initialize(self.device)
def numpy_prerun(self, is_raveled, io_usage):
inp = None
out = None
if is_raveled:
if io_usage[0]:
inp = ravel(self.input.mem)
if io_usage[1]:
out = ravel(self.output.mem)
err_input = ravel(self.err_input.mem)
err_output = ravel(self.err_output.mem)
else:
if io_usage[0]:
inp = self.input.mem
if io_usage[1]:
out = self.output.mem
err_input = self.err_input.mem
err_output = self.err_output.mem
if eq_addr(err_input, err_output):
self.err_input.map_write()
else:
self.err_input.map_invalidate()
self.err_output.map_read()
if io_usage[0]:
self.input.map_read()
if io_usage[1]:
self.output.map_read()
return inp, out, err_input, err_output
def numpy_prerun(self, is_raveled, io_usage):
inp = None
out = None
if is_raveled:
if io_usage[0]:
inp = ravel(self.input.mem)
if io_usage[1]:
out = ravel(self.output.mem)
err_input = ravel(self.err_input.mem)
err_output = ravel(self.err_output.mem)
else:
if io_usage[0]:
inp = self.input.mem
if io_usage[1]:
out = self.output.mem
err_input = self.err_input.mem
err_output = self.err_output.mem
if eq_addr(err_input, err_output):
self.err_input.map_write()
else:
self.err_input.map_invalidate()
self.err_output.map_read()
if io_usage[0]:
self.input.map_read()
if io_usage[1]:
self.output.map_read()
return inp, out, err_input, err_output
def numpy_run(self):
if eq_addr(self.err_input.mem, self.err_output.mem):
self.err_output.map_write()
else:
self.err_output.map_read()
self.err_input.map_invalidate()
self.mask.map_read()
numpy.multiply(self.err_output.mem.ravel(), self.mask.mem.ravel(), ravel(self.err_input.mem))
def numpy_run(self):
if eq_addr(self.err_input.mem, self.err_output.mem):
self.err_output.map_write()
else:
self.err_output.map_read()
self.err_input.map_invalidate()
self.mask.map_read()
numpy.multiply(self.err_output.mem.ravel(), self.mask.mem.ravel(),
ravel(self.err_input.mem))
def numpy_prerun(self, make_raveled, copy_in2out):
if make_raveled:
inp = ravel(self.input.mem)
out = ravel(self.output.mem)
else:
inp = self.input.mem
out = self.output.mem
if eq_addr(inp, out):
self.output.map_write()
else:
self.output.map_invalidate()
self.input.map_read()
if copy_in2out:
numpy.copyto(out, inp)
return inp, out
def numpy_prerun(self, make_raveled, copy_in2out):
if make_raveled:
inp = ravel(self.input.mem)
out = ravel(self.output.mem)
else:
inp = self.input.mem
out = self.output.mem
if eq_addr(inp, out):
self.output.map_write()
else:
self.output.map_invalidate()
self.input.map_read()
if copy_in2out:
numpy.copyto(out, inp)
return inp, out
def initialize(self, device, **kwargs):
if (id(self.output) == id(self.input) or
(self.output is not None and self.output.mem is not None and
eq_addr(self.output.mem, self.input.mem))):
raise error.BadFormatError("in_place for this unit is prohibited")
super(ForwardSinCos, self).initialize(device=device, **kwargs)
def initialize(self, device, **kwargs):
if (id(self.output) == id(self.input)
or (self.output is not None and self.output.mem is not None
and eq_addr(self.output.mem, self.input.mem))):
raise error.BadFormatError("in_place for this unit is prohibited")
super(ForwardSinCos, self).initialize(device=device, **kwargs)
