We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 20543c6 commit 58cb53eCopy full SHA for 58cb53e
core/partitioning/shape_analysis.cpp
@@ -61,7 +61,7 @@ void getSegmentsOutputByRunning(
61
jit_inputs_ivalues.push_back(ivalues_maps[input].toBool());
62
} else if (input->type()->kind() == torch::jit::TypeKind::ListType) {
63
jit_inputs_ivalues.push_back(ivalues_maps[input].toList());
64
- } else if (input->type()->kind() == torch::jit::TypeKind::TupleType){
+ } else if (input->type()->kind() == torch::jit::TypeKind::TupleType) {
65
jit_inputs_ivalues.push_back(ivalues_maps[input].toTuple());
66
} else {
67
TRTORCH_THROW_ERROR("Unable to find type for value: " << input->debugName() << " to get the ivalues.\n");
0 commit comments