Update onnx_config.pbtext
Browse files- onnx_config.pbtext +7 -1
onnx_config.pbtext
CHANGED
|
@@ -29,4 +29,10 @@ instance_group [
|
|
| 29 |
count: 1
|
| 30 |
kind: KIND_GPU
|
| 31 |
}
|
| 32 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
count: 1
|
| 30 |
kind: KIND_GPU
|
| 31 |
}
|
| 32 |
+
]
|
| 33 |
+
optimization { execution_accelerators {
|
| 34 |
+
gpu_execution_accelerator : [ {
|
| 35 |
+
name : "tensorrt"
|
| 36 |
+
parameters { key: "precision_mode" value: "FP16" }
|
| 37 |
+
]
|
| 38 |
+
}}
|