@@ -126,8 +126,8 @@ void loadAndRunPt2(String modelFolder, String modelSource) throws LoadEngineExce
126
126
// Create the input tensor with the nameand axes given by the rdf.yaml file
127
127
// and add it to the list of input tensors
128
128
Tensor <FloatType > inpTensor = Tensor .build ("input0" , "bcyx" , img1 );
129
- List <Tensor <T >> inputs = new ArrayList <Tensor <T >>();
130
- inputs .add (( Tensor < T >) inpTensor );
129
+ List <Tensor <FloatType >> inputs = new ArrayList <Tensor <FloatType >>();
130
+ inputs .add (inpTensor );
131
131
132
132
// Create the output tensors defined in the rdf.yaml file with their corresponding
133
133
// name and axes and add them to the output list of tensors.
@@ -140,8 +140,8 @@ void loadAndRunPt2(String modelFolder, String modelSource) throws LoadEngineExce
140
140
new FloatType());*/
141
141
final Img < FloatType > img2 = imgFactory .create ( 1 , 2 , 512 , 512 );
142
142
Tensor <FloatType > outTensor = Tensor .build ("output0" , "bcyx" , img2 );
143
- List <Tensor <R >> outputs = new ArrayList <Tensor <R >>();
144
- outputs .add (( Tensor < R >) outTensor );
143
+ List <Tensor <FloatType >> outputs = new ArrayList <Tensor <FloatType >>();
144
+ outputs .add (outTensor );
145
145
146
146
// Run the model on the input tensors. THe output tensors
147
147
// will be rewritten with the result of the execution
@@ -197,8 +197,8 @@ void loadAndRunPt1(String modelFolder, String modelSource) throws LoadEngineExce
197
197
// Create the input tensor with the nameand axes given by the rdf.yaml file
198
198
// and add it to the list of input tensors
199
199
Tensor <FloatType > inpTensor = Tensor .build ("input0" , "bcyx" , img1 );
200
- List <Tensor <T >> inputs = new ArrayList <Tensor <T >>();
201
- inputs .add (( Tensor < T >) inpTensor );
200
+ List <Tensor <FloatType >> inputs = new ArrayList <Tensor <FloatType >>();
201
+ inputs .add (inpTensor );
202
202
203
203
// Create the output tensors defined in the rdf.yaml file with their corresponding
204
204
// name and axes and add them to the output list of tensors.
@@ -211,8 +211,8 @@ void loadAndRunPt1(String modelFolder, String modelSource) throws LoadEngineExce
211
211
new FloatType());*/
212
212
final Img < FloatType > img2 = imgFactory .create ( 1 , 2 , 512 , 512 );
213
213
Tensor <FloatType > outTensor = Tensor .build ("output0" , "bcyx" , img2 );
214
- List <Tensor <R >> outputs = new ArrayList <Tensor <R >>();
215
- outputs .add (( Tensor < R >) outTensor );
214
+ List <Tensor <FloatType >> outputs = new ArrayList <Tensor <FloatType >>();
215
+ outputs .add (outTensor );
216
216
217
217
// Run the model on the input tensors. THe output tensors
218
218
// will be rewritten with the result of the execution
0 commit comments