(self)
| 94 | return self |
| 95 | |
| 96 | def _compile(self): |
| 97 | self.all_oprs_map = {} |
| 98 | self.all_vars_map = {} |
| 99 | for opr in self.all_oprs: |
| 100 | if isinstance(opr, (ConstOpBase, Host2DeviceCopy)): |
| 101 | opr.compile(self.graph) |
| 102 | else: |
| 103 | opr.compile() |
| 104 | if opr.name is not None: |
| 105 | opr._opr.name = opr.name |
| 106 | self.all_oprs_map[opr._opr.id] = opr |
| 107 | for o in opr.outputs: |
| 108 | self.all_vars_map[o.var.id] = o |
| 109 | |
| 110 | def optimize_for_inference(self, dest_vars, **kwargs): |
| 111 | r"""Applies optimize_for_inference pass for operator graph. |
no test coverage detected