If you have already the tf->onnx it is on the pytorch side.
E.g.:
  

      github.com/pytorch/pytorch
  

  
    
  
	  
  

  
    
      Import ONNX model to Pytorch 179
    

    
      
        opened 
        
          
        
        Jun 12, 2019
      
      


      
        
          
          hossein1387
        
      
    

    
        module: onnx
        feature
        triaged
    
  


  
    ## 🚀 Feature

Importing ONNX models into Pytorch.
## Motivation

Almost all… other frameworks already support this. Importing ONNX models into Pytorch makes Pytorch much more flexible.

## Pitch

In `torch.onnx`, a function should be created to take the ONNX model and outputs a Pytorch model.