Searched refs:use_python (Results 1 – 2 of 2) sorted by relevance
61 def _register_jit_decomposition_for_jvp(decomp, use_python=False): argument75 if use_python:325 _register_jit_decomposition_for_jvp(torch.ops.aten.trace.default, use_python=True)
100 parser.add_argument('-p', dest='use_python', action='store_true')104 shell = sys.executable if options.use_python else os.environ.get('SHELL', 'sh')