示例#1
0
def _MaxPoolWithArgMaxShape(op):
  """Shape function for MaxPoolWithArgmax op."""
  return common_shapes.max_pool_shape(op) * 2
示例#2
0
def _QuantizedMaxPoolShape(op):
    return [
        common_shapes.max_pool_shape(op)[0],
        tensor_shape.scalar(),
        tensor_shape.scalar()
    ]
示例#3
0
def _MaxPoolWithArgMaxShape(op):
    """Shape function for MaxPoolWithArgmax op."""
    return common_shapes.max_pool_shape(op) * 2
示例#4
0
def _QuantizedMaxPoolShape(op):
  return [common_shapes.max_pool_shape(op)[0], tensor_shape.scalar(),
          tensor_shape.scalar()]