| 2060 | self._net.arg.extend([utils.MakeArgument(arg_name, arg_value)]) |
| 2061 | |
| 2062 | def AddExternalInput(self, *inputs): |
| 2063 | assert len(inputs) > 0 |
| 2064 | refs = [] |
| 2065 | input_name_set = set() |
| 2066 | for input in inputs: |
| 2067 | input_name = str(input) |
| 2068 | assert ( |
| 2069 | input_name not in self._external_input_map |
| 2070 | and input_name not in input_name_set |
| 2071 | ), ("Net already contains an input named %s" % input_name) |
| 2072 | input_name_set.add(input_name) |
| 2073 | for input in inputs: |
| 2074 | input_name = str(input) |
| 2075 | self._net.external_input.extend([input_name]) |
| 2076 | self._external_input_map.update([input_name]) |
| 2077 | refs.append(_get_blob_ref(input_name)) |
| 2078 | |
| 2079 | return refs[0] if len(refs) == 1 else refs |
| 2080 | |
| 2081 | def AddExternalOutput(self, *outputs): |
| 2082 | for output in outputs: |