Browse Source

actually calling the model

pull/858/head
Luke Bollam 4 years ago
parent
commit
cee1749b1d
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs

+ 2
- 2
src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs View File

@@ -27,8 +27,8 @@ namespace Tensorflow.Benchmark.Leak
var inputOp = g.OperationByName("inference_input");
var outputOp = g.OperationByName("StatefulPartitionedCall");

var inp = np.zeros(new Shape(new int[] { 1, 96, 2 }));
var ops = g.OperationByName("StatefulPartitionedCall");
var inp = np.zeros(new Shape(new int[] { 1, 2, 96 }), TF_DataType.TF_FLOAT);
sess.run(outputOp.outputs[0], new FeedItem(inputOp.outputs[0], inp));
}
}
}


Loading…
Cancel
Save