Converting PyTorch(efficientNet) model to TFLite using ONNX (GPU Binding)

Hello I am trying to convert a PyTorch EfficeintNet-B0 model to TFLite using ONNX. I am able to run the CPU version but the model will not bind to GPU. Is there any way to Bind the model to GPU? A method I tried of binding it to GPU does not yield the right output and the model returns NaNs.