File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
core/conversion/converters/impl Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -83,13 +83,16 @@ auto reduce_registrations TRTORCH_UNUSED =
83
83
c10::List<int64_t > calculated_dims;
84
84
auto in_dims = util::toVec (in_tensor->getDimensions ());
85
85
LOG_DEBUG (" InDims " << in_dims); // Some abuse of toDim but just for debug info
86
- LOG_DEBUG (" Dim to reduce(original):" << util::toDims (dims)); // Some abuse of toDim but just for debug info
86
+ LOG_DEBUG (
87
+ " Dim to reduce(original):" << util::toDims (dims)); // Some abuse of toDim but just for debug info
87
88
for (int i = 0 ; i < dims.size (); i++) {
88
89
auto dim_val = dims[i] == -1 ? (in_dims.size () - 1 ) : dims[i];
89
90
calculated_dims.push_back (dim_val);
90
91
}
91
92
92
- LOG_DEBUG (" Dim to reduce(converted):" << util::toDims (calculated_dims)); // Some abuse of toDim but just for debug info
93
+ LOG_DEBUG (
94
+ " Dim to reduce(converted):"
95
+ << util::toDims (calculated_dims)); // Some abuse of toDim but just for debug info
93
96
94
97
uint32_t axis_mask = 0 ;
95
98
for (size_t d = 0 ; d < calculated_dims.size (); d++) {
You can’t perform that action at this time.
0 commit comments