Update onnx_config.pbtxt
Browse files- onnx_config.pbtxt +7 -1
onnx_config.pbtxt
CHANGED
|
@@ -26,4 +26,10 @@ instance_group [
|
|
| 26 |
kind: KIND_GPU
|
| 27 |
}
|
| 28 |
]
|
| 29 |
-
dynamic_batching { }
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
kind: KIND_GPU
|
| 27 |
}
|
| 28 |
]
|
| 29 |
+
dynamic_batching { }
|
| 30 |
+
optimization { execution_accelerators {
|
| 31 |
+
gpu_execution_accelerator : [ {
|
| 32 |
+
name : "tensorrt"
|
| 33 |
+
parameters { key: "precision_mode" value: "FP32" }
|
| 34 |
+
} ]
|
| 35 |
+
}}
|