TensorRT currently is supporting Caffe; it would definitely be interesting to make a direct Caffe2 compatibility support.