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 df24079 commit 911ab5bCopy full SHA for 911ab5b
core/conversion/converters/impl/constant_pad.cpp
@@ -108,7 +108,7 @@ auto constant_pad_registrations TORCHTRT_UNUSED = RegisterNodeConversionPatterns
108
// fill the right_gather_out with value
109
auto fill_layer = ctx->net->addFill(nvinfer1::Dims{1, {1}}, nvinfer1::FillOperation::kLINSPACE);
110
auto shape_gather_out = ctx->net->addShape(*right_gather_out)->getOutput(0);
111
- fill_layer->setInput(0, *shape_gather_out);
+ fill_layer->setInput(0, *shape_gather_out);
112
fill_layer->setInput(1, *valueTensor);
113
at::Tensor delta_tensor = torch::zeros(inRank, util::TRTDataTypeToScalarType(in->getType()));
114
auto deltaTensor = tensor_to_const(ctx, delta_tensor);
0 commit comments