| 722 | } |
| 723 | |
| 724 | nvinfer1::ILayer * layer_split(const int n_layer_index_, |
| 725 | nvinfer1::ITensor *input_, |
| 726 | nvinfer1::INetworkDefinition* network) |
| 727 | { |
| 728 | auto creator = getPluginRegistry()->getPluginCreator("CHUNK_TRT", "1.0"); |
| 729 | const nvinfer1::PluginFieldCollection* pluginData = creator->getFieldNames(); |
| 730 | nvinfer1::IPluginV2 *pluginObj = creator->createPlugin(("chunk" + std::to_string(n_layer_index_)).c_str(), pluginData); |
| 731 | auto chunk = network->addPluginV2(&input_, 1, *pluginObj); |
| 732 | return chunk; |
| 733 | } |
| 734 | |
| 735 | std::vector<int> parse_int_list(const std::string s_args_) |
| 736 | { |
no test coverage detected