Python TensorRT-不支持的操作\u填充

我试图通过TensorRT引擎将resnet从.pb（tensorflow）运行到.trt。我将.pb转换为.uff，现在尝试通过以下代码将其加载到引擎：

import tensorrt.legacy as trt
import tensorflow as tf
import pycuda.driver as cuda
import pycuda.autoinit
import numpy as np
import cv2
from tensorrt.legacy.parsers import uffparser
import graphsurgeon as gs

# Build TensorRT engine
uff_model_path = "model/resnet_model_v1.uff"
engine_path = "model/resnet_model_v1.engine"


TRT_LOGGER = trt.Logger(trt.Logger.WARNING)
trt.init_libnvinfer_plugins(TRT_LOGGER, '')

trt_runtime = trt.Runtime(TRT_LOGGER)

with trt.Builder(TRT_LOGGER) as builder, builder.create_network() as network, trt.UffParser() as parser:
  builder.max_workspace_size = 1 << 30
  builder.fp16_mode = True
  builder.max_batch_size = 1
  parser.register_input("input_image", (3, 150, 150))
  parser.register_output("embedding_layer/MatMul")
  parser.parse(uff_model_path, network)
  
  print("Building TensorRT engine, this may take a few minutes...")
  trt_engine = builder.build_cuda_engine(network)

将tensorrt.legacy导入为trt
导入tensorflow作为tf
将pycuda.driver导入为cuda
导入pycuda.autoinit
将numpy作为np导入
进口cv2
从tensorrt.legacy.parsers导入uffparser
将graphsurgeon作为gs导入
#构建TensorRT引擎
uff_model_path=“model/resnet_model_v1.uff”
engine\u path=“model/resnet\u model\u v1.engine”
TRT_LOGGER=TRT.LOGGER（TRT.LOGGER.WARNING）
trt.init_libnvere_插件（trt_记录器“”）
trt\u runtime=trt.runtime（trt\u记录器）
使用trt.Builder（trt_LOGGER）作为生成器，Builder.create_network（）作为网络，trt.UffParser（）作为解析器：
builder.max\u workspace\u size=1我面临同样的问题。我找到了他们讨论这个错误的地方，但不幸的是我无法更改我的网络，所以提到的解决方案对我不起作用。