(trt_network: trt.INetworkDefinition, layer_name: str,
plugin_info: PluginInfo)
| 85 | |
| 86 | |
| 87 | def set_plugin_info(trt_network: trt.INetworkDefinition, layer_name: str, |
| 88 | plugin_info: PluginInfo): |
| 89 | if not has_extra_attr(trt_network, "plugin_infos"): |
| 90 | set_extra_attr(trt_network, "plugin_infos", {}) |
| 91 | plugin_infos = get_extra_attr(trt_network, "plugin_infos") |
| 92 | plugin_infos[layer_name] = plugin_info |
| 93 | |
| 94 | |
| 95 | def delete_plugin_info(trt_network: trt.INetworkDefinition, layer_name: str): |
no test coverage detected