[GitHub] [incubator-tvm] FrozenGene commented on a change in pull request #6901: [TFLite runtime] Allow to set number of threads to TFLite interpreter

[GitBox <gi...@apache.org>]

FrozenGene commented on a change in pull request #6901:
URL: https://github.com/apache/incubator-tvm/pull/6901#discussion_r521721976



##########
File path: python/tvm/contrib/tflite_runtime.py
##########
@@ -109,3 +110,12 @@ def get_output(self, index):
             The output index
         """
         return self._get_output(index)
+
+    def set_num_threads(self, num_threads):
+        """Set the number of threads via kwargs
+        Parameters
+        ----------
+        num_threads : int
+           The number of threads
+        """
+        self._set_num_threads(num_threads)

Review comment:
       how about adding one new line at the end of file to meet the requirement of PEP8?




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org