parameters: (op)
return_statement: return [grad_shape]
docstring: Shape function for the ApplyRMSProp op.
docstring_summary: Shape function for the ApplyRMSProp op.
docstring_tokens: Shape function for the ApplyRMSProp op .
function: def _ApplyRMSPropShape(op):
    """Shape function for the ApplyRMSProp op."""
    var_shape = op.inputs[0].get_shape()
    ms_shape = op.inputs[1].get_shape().merge_with(var_shape)
    mom_shape = op.inputs[2].get_shape().merge_with(ms_shape)
    _AssertInputIsScalar(op, 3)  # lr
    _AssertInputIsScalar(op, 4)  # rho
    _AssertInputIsScalar(op, 5)  # momentum
    _AssertInputIsScalar(op, 6)  # epsilon
    grad_shape = op.inputs[7].get_shape().merge_with(mom_shape)
    return [grad_shape]
function_tokens: def _ApplyRMSPropShape ( op ) : var_shape = op . inputs [ 0 ] . get_shape ( ) ms_shape = op . inputs [ 1 ] . get_shape ( ) . merge_with ( var_shape ) mom_shape = op . inputs [ 2 ] . get_shape ( ) . merge_with ( ms_shape ) _AssertInputIsScalar ( op , 3 ) # lr _AssertInputIsScalar ( op , 4 ) # rho _AssertInputIsScalar ( op , 5 ) # momentum _AssertInputIsScalar ( op , 6 ) # epsilon grad_shape = op . inputs [ 7 ] . get_shape ( ) . merge_with ( mom_shape ) return [ grad_shape ]