Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- // Initialize interpreter with GPU delegate.
- GpuDelegate delegate = new GpuDelegate();
- Interpreter.Options options = (new Interpreter.Options()).addDelegate(delegate);
- Interpreter interpreter = new Interpreter(model, options);
- // Run inference.
- while (true) {
- writeToInputTensor(inputTensor);
- interpreter.run(inputTensor, outputTensor);
- readFromOutputTensor(outputTensor);
- }
- // Clean up.
- delegate.close();
Add Comment
Please, Sign In to add comment