fix: Improve input handling for input_signature by gs-olive · Pull Request #1698 · pytorch/TensorRT (original) (raw)
| if (level == 0) { // a single value like A |
|---|
| collection_inputs.resize(1); |
| collection_inputs[0].push_back(cur_input); |
| } else if (level == 1) { // like A in [A, A] or [(B, B), A] |
| collection_inputs[index].push_back(cur_input); |
| } else if (level == 2) { // like A in [(A, A), C] |
| collection_inputs[index].push_back(cur_input); |
| } else { // only support 2 level |
| LOG_ERROR( |
| "Input nesting depth exceeds currently supported depth (3), use 1 level: [A, B], or 2 level: [A, (B, C)]"); |
| } |
| } |
| } |