(net)
| 340 | |
| 341 | |
| 342 | def GetNetName(net): |
| 343 | if isinstance(net, basestring): |
| 344 | return net |
| 345 | if type(net).__name__ == "Net" or type(net).__name__ == "NetWithShapeInference": |
| 346 | return net.Name() |
| 347 | if isinstance(net, caffe2_pb2.NetDef): |
| 348 | return net.name |
| 349 | raise Exception("Not a Net object: {}".format(str(net))) |
| 350 | |
| 351 | |
| 352 | def FeedBlob(name, arr, device_option=None): |
no test coverage detected
searching dependent graphs…