Exemplo n.º 1
0
 def max_pool(data,
              auto_pad=None,
              kernel_shape=None,
              pads=None,
              strides=None):
     padding = _convert_auto_pad(auto_pad, pads)
     return (op.max_pool(data,
                         padding=padding,
                         kernel_shape=kernel_shape,
                         pads=pads,
                         strides=strides), )
Exemplo n.º 2
0
 def global_max_pool(x):
     return (op.max_pool(x,
                         padding=op.AutoPadding.VALID,
                         kernel_shape=x.shape.dims[2:],
                         pads=None,
                         strides=None), )