Method _find_var_class_kwargs

python/paddle/base/framework.py:6132–6229

(self, new_desc)

        self._grad_var_to_var = None

    def _find_var_class_kwargs(self, new_desc):
        # NOTE: not all variables support shape/dtype/lod_level methods.
        # For example: RAW, STEP_SCOPES, etc.
        def get_var_desc_attr_or_none(var_desc, attr_name, allowed_types):
            if var_desc.type() in allowed_types:
                return getattr(var_desc, attr_name)()
            else:
                return None

        old_desc = self.desc
        all_new_vars = []
        block_num = new_desc.num_blocks()
        for idx in range(block_num):
            if idx > (len(self.blocks) - 1):
                self._create_block()
            new_block_desc = new_desc.block(idx)
            all_new_vars.append([])
            block_new_vars = all_new_vars[-1]
            for new_var_desc in new_block_desc.all_vars():
                if self.blocks[idx].has_var(new_var_desc.name()):
                    old_var = self.blocks[idx].var(new_var_desc.name())
                else:
                    old_var = None

                kwargs = {
                    "type": new_var_desc.type(),
                    "name": new_var_desc.name(),
                    "shape": get_var_desc_attr_or_none(
                        new_var_desc,
                        "shape",
                        [
                            core.VarDesc.VarType.DENSE_TENSOR,
                            core.VarDesc.VarType.SELECTED_ROWS,
                            core.VarDesc.VarType.DENSE_TENSOR_ARRAY,
                        ],
                    ),
                    "dtype": get_var_desc_attr_or_none(
                        new_var_desc,
                        "dtype",
                        [
                            core.VarDesc.VarType.DENSE_TENSOR,
                            core.VarDesc.VarType.SELECTED_ROWS,
                            core.VarDesc.VarType.DENSE_TENSOR_ARRAY,
                        ],
                    ),
                    "lod_level": get_var_desc_attr_or_none(
                        new_var_desc,
                        "lod_level",
                        [
                            core.VarDesc.VarType.DENSE_TENSOR,
                            core.VarDesc.VarType.DENSE_TENSOR_ARRAY,
                        ],
                    ),
                    "error_clip": (
                        old_var.error_clip if old_var is not None else None
                    ),
                    "stop_gradient": (
                        old_var.stop_gradient if old_var is not None else False