diff options
author | 2018-03-16 11:45:42 -0700 | |
---|---|---|
committer | 2018-03-16 11:50:09 -0700 | |
commit | fff901507e932188932fe21ae56c55e4aba5ae52 (patch) | |
tree | b0a0be291b0b39f9818f44aead3f53a1f96caa8b /tensorflow/contrib/lite/model.h | |
parent | 53cbb5a50e94a772b595dcadd63af3b79bb884c7 (diff) |
Set number of threads at Java interpreter constructor so that Conv Kernels can be selected properly.
Remove setNumThreads in the Java API as its behavior is ambiguous.
PiperOrigin-RevId: 189370770
Diffstat (limited to 'tensorflow/contrib/lite/model.h')
-rw-r--r-- | tensorflow/contrib/lite/model.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.h b/tensorflow/contrib/lite/model.h index 51a622a28d..0c777760cb 100644 --- a/tensorflow/contrib/lite/model.h +++ b/tensorflow/contrib/lite/model.h @@ -154,6 +154,8 @@ class InterpreterBuilder { InterpreterBuilder(const InterpreterBuilder&) = delete; InterpreterBuilder& operator=(const InterpreterBuilder&) = delete; TfLiteStatus operator()(std::unique_ptr<Interpreter>* interpreter); + TfLiteStatus operator()(std::unique_ptr<Interpreter>* interpreter, + int num_threads); private: TfLiteStatus BuildLocalIndexToRegistrationMapping(); |