11.TensorRT中文版开发教程-----TensorRT之条件控制
TensorRT之条件控制点击此处加入NVIDIA开发者计划NVIDIA TensorRT 支持条件 if-then-else 流控制。 TensorRT 条件用于实现网络子图的条件执行。11.1. Defining A Conditionalif-conditional 由条件边界层定义:IConditionLayer表示predicate 并指定条件是应该执行真分支(then-branch)还是假分支(else-branch)。IIfConditionalInputLayer指定两个条件