The forward of the SINGA model Args: input (Tensors[]): a list of Tensor aux_output (string()): a set of required output name Returns: a OrderedDict of Tensor
(self, *input, aux_output=(), **kwargs)
| 2208 | self.sg_ir.is_graph = True |
| 2209 | |
| 2210 | def forward(self, *input, aux_output=(), **kwargs): |
| 2211 | """ |
| 2212 | The forward of the SINGA model |
| 2213 | Args: |
| 2214 | input (Tensors[]): a list of Tensor |
| 2215 | aux_output (string()): a set of required output name |
| 2216 | |
| 2217 | Returns: |
| 2218 | a OrderedDict of Tensor |
| 2219 | """ |
| 2220 | return self.sg_ir.run(input, aux_output=aux_output, **kwargs) |
| 2221 | |
| 2222 | |
| 2223 | run_node = SingaBackend.run_node |