parameters: (op)
return_statement: return [delta_shape]
docstring: Shape function for the ApplyProximalGradientDescent op.
docstring_summary: Shape function for the ApplyProximalGradientDescent op.
docstring_tokens: Shape function for the ApplyProximalGradientDescent op .
function: def _ApplyProximalGradientDescentShape(op):
    """Shape function for the ApplyProximalGradientDescent op."""
    var_shape = op.inputs[0].get_shape()
    _AssertInputIsScalar(op, 1)  # alpha
    _AssertInputIsScalar(op, 2)  # l1
    _AssertInputIsScalar(op, 3)  # l2
    delta_shape = op.inputs[4].get_shape().merge_with(var_shape)
    return [delta_shape]
function_tokens: def _ApplyProximalGradientDescentShape ( op ) : var_shape = op . inputs [ 0 ] . get_shape ( ) _AssertInputIsScalar ( op , 1 ) # alpha _AssertInputIsScalar ( op , 2 ) # l1 _AssertInputIsScalar ( op , 3 ) # l2 delta_shape = op . inputs [ 4 ] . get_shape ( ) . merge_with ( var_shape ) return [ delta_shape ]