def _get_double_grads(self, program):
    double_grads = []
    for block in program.blocks:
        for name in block.vars:
            if ('@GRAD' in name):
                var_desc = block.vars[name].desc
                var_base = core.VarBase(var_desc.dtype(), var_desc.shape(), var_desc.name(), var_desc.type(), False)
                double_grads.append(var_base)
    return double_grads
def _get_double_grads(self, program):
    double_grads = []
    for block in program.blocks:
        for name in block.vars:
            if ('@GRAD' in name):
                var_desc = block.vars[name].desc
                var_base = core.VarBase(var_desc.dtype(), var_desc.shape(), var_desc.name(), var_desc.type(), False)
                double_grads.append(var_base)
    return double_grads