File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ();
You can’t perform that action at this time.
0 commit comments