parameters: (op)
return_statement: return [op.inputs[0].shape]
docstring: Shape helper function for scale_gradient function below.
docstring_summary: Shape helper function for scale_gradient function below.
docstring_tokens: Shape helper function for scale_gradient function below .
function: def _scale_gradient_shape(op):
    """Shape helper function for scale_gradient function below."""
    return [op.inputs[0].shape]
function_tokens: def _scale_gradient_shape ( op ) : return [ op . inputs [ 0 ] . shape ]