expass.py 文件源码

def end_pass(self, **kwargs):
        super(SSAConversion, self).end_pass(**kwargs)
        for source_tensor_decl, current_exop in iteritems(self.tensor_map):
            if current_exop.output_decls[0].tensor_decl is source_tensor_decl:
                continue
            if not source_tensor_decl.is_output:
                continue
            copy_exop = ExOp(computation_decl=self.computation_decl,
                             create_value=False,
                             op=WriteOp(axes=[]))
            copy_exop.add_write_arg(source_tensor_decl.exop.output_decls[0])
            copy_exop.add_input_decl(current_exop.output_decls[0])
            self.exop_block.add_exop(copy_exop)