...
def get_frozen_graph(graph_file):
    with tf.io.gfile.GFile(graph_file, "rb") as f:
        graph_def = tf.compat.v1.GraphDef()
        graph_def.ParseFromString(f.read())
    return graph_def
    
trt_graph = get_frozen_graph('./models/mobilenetv2-model-trt.pb')
