Add argmode argument to DtypeRunner.run (#350)

Summary:
Pull Request resolved: #350

This will speed up spec generation, since `run_dtypes` will not attempt random tensors first.

Reviewed By: larryliu0820

Differential Revision: D49217150

fbshipit-source-id: 909a09b364ec17571afdfec0361dee8bebbe107a

Author: manuelcandales

exir/dialects/edge/dtype/runner.py

@@ -172,7 +172,10 @@ def run_dtypes(
                 return (False, name, dtypes, ones_args, ones_kwargs)
 
     def run(
-        self, name: str, inputs: Dict[str, List[BaseArg]]
+        self,
+        name: str,
+        inputs: Dict[str, List[BaseArg]],
+        argmode: ArgMode = ArgMode.ONES,
     ) -> List[
         Tuple[
             bool, str, Tuple[Optional[torch.dtype]], List[BaseArg], Dict[str, BaseKwarg]
@@ -181,5 +184,5 @@ def run(
         results = []
         type_tuples = self._get_type_tuples(inputs)
         for element in itertools.product(*type_tuples):
-            results.append(self.run_dtypes(name, inputs, element))
+            results.append(self.run_dtypes(name, inputs, element, argmode))
         return results