csvance/jetson_tensorrt

PyTorch Generic Support

Closed this issue 6 years ago · 0 comments

csvance commented 6 years ago

Figure out best way to import PyTorch graph
Add support in TensorRT abstraction layer if needed
Implement PyTorch Generic node in pytorch_generic.cpp