Skip to content

Commit 71be7ed

Browse files
committed
Please consider the following formatting changes
1 parent 62d5c83 commit 71be7ed

1 file changed

Lines changed: 23 additions & 7 deletions

File tree

Tools/ML/model.h

Lines changed: 23 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -520,13 +520,29 @@ class OnnxModel
520520
const char* dim = "N";
521521
bool scalar = false;
522522
switch (pin.type) {
523-
case PreprocInput::Type::TrackFloat: et = onnx::TensorProto::FLOAT; break;
524-
case PreprocInput::Type::TrackInt32: et = onnx::TensorProto::INT32; break;
525-
case PreprocInput::Type::TrackUint8: et = onnx::TensorProto::UINT8; break;
526-
case PreprocInput::Type::TrackInt8: et = onnx::TensorProto::INT8; break;
527-
case PreprocInput::Type::TrackBool: et = onnx::TensorProto::BOOL; break;
528-
case PreprocInput::Type::CollisionFloat: et = onnx::TensorProto::FLOAT; dim = "C"; break;
529-
case PreprocInput::Type::ScalarFloat: et = onnx::TensorProto::FLOAT; scalar = true; break;
523+
case PreprocInput::Type::TrackFloat:
524+
et = onnx::TensorProto::FLOAT;
525+
break;
526+
case PreprocInput::Type::TrackInt32:
527+
et = onnx::TensorProto::INT32;
528+
break;
529+
case PreprocInput::Type::TrackUint8:
530+
et = onnx::TensorProto::UINT8;
531+
break;
532+
case PreprocInput::Type::TrackInt8:
533+
et = onnx::TensorProto::INT8;
534+
break;
535+
case PreprocInput::Type::TrackBool:
536+
et = onnx::TensorProto::BOOL;
537+
break;
538+
case PreprocInput::Type::CollisionFloat:
539+
et = onnx::TensorProto::FLOAT;
540+
dim = "C";
541+
break;
542+
case PreprocInput::Type::ScalarFloat:
543+
et = onnx::TensorProto::FLOAT;
544+
scalar = true;
545+
break;
530546
}
531547
tt->set_elem_type(et);
532548
auto* sh = tt->mutable_shape();

0 commit comments

Comments
 (0)