File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
core/conversion/converters/impl Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -140,7 +140,7 @@ auto batch_norm_registrations TORCHTRT_UNUSED =
140
140
scales = args[1 ].unwrapToTensor (at::ones (shape[1 ], options)).cpu ().contiguous ();
141
141
}
142
142
auto bias = at::zeros (shape[1 ], options);
143
- if (!args[2 ].IValue ()->isNone ()){
143
+ if (!args[2 ].IValue ()->isNone ()) {
144
144
bias = args[2 ].unwrapToTensor (at::zeros (shape[1 ], options)).cpu ().contiguous ();
145
145
}
146
146
// track_running_stats=True
@@ -170,7 +170,7 @@ auto batch_norm_registrations TORCHTRT_UNUSED =
170
170
so for some functionalities, users need to install correct \
171
171
cuDNN version by themselves. Please see our support matrix \
172
172
here: https://docs.nvidia.com/deeplearning/tensorrt/support-matrix/index.html." );
173
- // return false;
173
+ // return false;
174
174
}
175
175
176
176
const int relu = 0 ;
You can’t perform that action at this time.
0 commit comments