parameters: (op)
return_statement: return common_shapes.max_pool_shape(op) * 2
docstring: Shape function for MaxPoolWithArgmax op.
docstring_summary: Shape function for MaxPoolWithArgmax op.
docstring_tokens: Shape function for MaxPoolWithArgmax op .
function: def _MaxPoolWithArgMaxShape(op):
    """Shape function for MaxPoolWithArgmax op."""
    return common_shapes.max_pool_shape(op) * 2
function_tokens: def _MaxPoolWithArgMaxShape ( op ) : return common_shapes . max_pool_shape ( op ) * 2