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