(trt_network: trt.INetworkDefinition,
layer_name: str)
| 75 | |
| 76 | |
| 77 | def get_plugin_info(trt_network: trt.INetworkDefinition, |
| 78 | layer_name: str) -> PluginInfo: |
| 79 | if not has_extra_attr(trt_network, "plugin_infos"): |
| 80 | return None |
| 81 | plugin_infos = get_extra_attr(trt_network, "plugin_infos") |
| 82 | if layer_name not in plugin_infos: |
| 83 | return None |
| 84 | return plugin_infos[layer_name] |
| 85 | |
| 86 | |
| 87 | def set_plugin_info(trt_network: trt.INetworkDefinition, layer_name: str, |
no test coverage detected