| 2117 | return [_get_blob_ref(x) for x in self._net.external_output] |
| 2118 | |
| 2119 | def set_input_record(self, input_record): |
| 2120 | from caffe2.python import schema |
| 2121 | assert self._input_record is None or (input_record.has_blobs() and |
| 2122 | set(input_record.field_blobs()) == |
| 2123 | set(self._input_record.field_blobs())), ( |
| 2124 | 'Input schema cannot be reset') |
| 2125 | if not input_record.has_blobs(): |
| 2126 | with NameScope(self.Name()): |
| 2127 | self._input_record = schema.NewRecord(self, input_record) |
| 2128 | else: |
| 2129 | self._input_record = input_record |
| 2130 | |
| 2131 | for blob in self._input_record.field_blobs(): |
| 2132 | if not self.is_external_input(blob): |
| 2133 | self.AddExternalInput(blob) |
| 2134 | return self._input_record |
| 2135 | |
| 2136 | def recover_input_record_by_prefix(self, prefix): |
| 2137 | """ |