python - サポートされていない操作 _FusedBatchNormV3 と tensorRT および jetson tx2

Question

私が実行しているJetson TX2で：

• Linux4Tegra R32.2.1
• UFF バージョン 0.6.3
• テンソルRT 5.1.6.1
• クーダ 10
• パイソン 3.6.8

次のエラー メッセージが表示されます。

[TensorRT] ERROR: UffParser: Validator error: sequential/batch_normalization_1/FusedBatchNormV3: Unsupported operation _FusedBatchNormV3

このコードから:

output_nodes = [args.output_node_names]
input_node = args.input_node_name
frozen_graph_pb = args.frozen_graph_pb

uff_model = uff.from_tensorflow(frozen_graph_pb, output_nodes) . #Successfully creates uff model

network = builder.create_network()
G_LOGGER = trt.Logger(trt.Logger.INFO)
builder = trt.Builder(G_LOGGER)
builder.max_batch_size = 10
builder.max_workspace_size = 1 << 30

data_type = trt.DataType.FLOAT

parser = trt.UffParser()
input_verified =parser.register_input(input_node, (1,234,234,3))       #returns true
output_verified = parser.register_output(output_nodes[0])              #returns true
buffer_verified = parser.parse_buffer(uff_model, network, data_type)   #returns false

uff モデルが正常に作成されました。

パーサーは入力と出力を正常に登録しました。

バッファーの解析は、上記のエラーで失敗します。

FusedBatchNormV3 が tensorRT で本当にサポートされていないかどうかを知っている人はいますか?もしそうでない場合は、グラフ外科医モジュールを使用してプルできる既存のプラグインがありますか?