Skip to content

Commit d2570a6

Browse files
committed
[tmva][sofie] Update TMVA_SOFIE_ONNX.C tutorial to use as input a user providedd ONNX file
1 parent 6075207 commit d2570a6

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tutorials/tmva/TMVA_SOFIE_ONNX.C

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,18 @@
1010

1111
using namespace TMVA::Experimental;
1212

13-
void TMVA_SOFIE_ONNX(){
13+
void TMVA_SOFIE_ONNX(std::string inputFile = ""){
14+
if (inputFile.empty() )
15+
inputFile = std::string(gROOT->GetTutorialsDir()) + "/tmva/Linear_16.onnx";
16+
1417
//Creating parser object to parse ONNX files
15-
SOFIE::RModelParser_ONNX Parser;
16-
SOFIE::RModel model = Parser.Parse(std::string(gROOT->GetTutorialsDir()) + "/tmva/Linear_16.onnx");
18+
SOFIE::RModelParser_ONNX parser;
19+
SOFIE::RModel model = parser.Parse(inputFile, true);
1720

1821
//Generating inference code
1922
model.Generate();
20-
model.OutputGenerated("Linear_16.hxx");
23+
// write the code in a file (by default Linear_16.hxx and Linear_16.dat
24+
model.OutputGenerated();
2125

2226
//Printing required input tensors
2327
model.PrintRequiredInputTensors();

0 commit comments

Comments
 (0)