Python mulexp Examples

Example #1

File: pooling_nd_kernel.py Project: PhysicsTeacher13/CHAINER

 def aux(out_x, outs):
     head = outs[0]
     tail = outs[1:]
     if tail:
         return 'int {} = i / ({}) % {};'.format(
             out_x, conv_nd_kernel.mulexp(tail), head)
     else:
         return 'int {} = i % {};'.format(out_x, head)

Example #2

File: pooling_nd_kernel.py Project: PhysicsTeacher13/CHAINER

 def aux(x, ds, p):
     head = ds[0]
     tail = ds[1:]
     if tail:
         return 'int {} = i / ({}) % {} + {};'.format(
             x, conv_nd_kernel.mulexp(tail), head, p)
     else:
         return 'int {} = i % {} + {};'.format(x, head, p)

Example #3

File: pooling_nd_kernel.py Project: Fhrozen/chainer

 def aux(out_x, outs):
     head = outs[0]
     tail = outs[1:]
     if tail:
         return 'int {} = i / ({}) % {};'.format(
             out_x, conv_nd_kernel.mulexp(tail), head)
     else:
         return 'int {} = i % {};'.format(out_x, head)

Example #4

File: pooling_nd_kernel.py Project: Fhrozen/chainer

 def aux(x, ds, p):
     head = ds[0]
     tail = ds[1:]
     if tail:
         return 'int {} = i / ({}) % {} + {};'.format(
             x, conv_nd_kernel.mulexp(tail), head, p)
     else:
         return 'int {} = i % {} + {};'.format(x, head, p)

Example #5

File: max_pooling_nd_kernel.py Project: PaulGureghian1/Chainer

 def aux(max_val, out_val, stride_val, pad_val, ksize_vals):
     head = ksize_vals[0]
     tail = ksize_vals[1:]
     if tail:
         command = 'int {} = max(0, {} * {} - {} + index / ({}) % {});'
         return command.format(max_val, out_val, stride_val, pad_val,
                               conv_nd_kernel.mulexp(tail), head)
     else:
         return 'int {} = max(0, {} * {} - {} + index % {});'.format(
             max_val, out_val, stride_val, pad_val, head)

Example #6

File: max_pooling_nd_kernel.py Project: Fhrozen/chainer

 def aux(max_val, out_val, stride_val, pad_val, ksize_vals):
     head = ksize_vals[0]
     tail = ksize_vals[1:]
     if tail:
         command = 'int {} = max(0, {} * {} - {} + index / ({}) % {});'
         return command.format(
             max_val, out_val, stride_val, pad_val,
             conv_nd_kernel.mulexp(tail), head)
     else:
         return 'int {} = max(0, {} * {} - {} + index % {});'.format(
             max_val, out_val, stride_val, pad_val, head)

Example #7

File: pooling_nd_kernel.py Project: PhysicsTeacher13/CHAINER

 def _compile_c0(self):
     # 2D: int c0 = i / (out_0 * out_1);
     return ['int c0 = i / ({});'.format(conv_nd_kernel.mulexp(self.outs))]

Example #8

File: pooling_nd_kernel.py Project: Fhrozen/chainer

 def _compile_c0(self):
     # 2D: int c0 = i / (out_0 * out_1);
     return ['int c0 = i / ({});'.format(conv_nd_kernel.mulexp(self.outs))]