Home
last modified time | relevance | path

Searched refs:include_device (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/torch/_functorch/_aot_autograd/
Djit_compile_runtime_wrappers.py348 include_device=True,
355 print_output=False, include_stride=True, include_device=True
530 include_device=True,
541 include_device=True,
548 print_output=False, include_stride=True, include_device=True
554 print_output=False, include_stride=True, include_device=True
Ddispatch_and_compile_graph.py211 include_device=True,
218 print_output=False, include_stride=True, include_device=True
Dutils.py357 include_device=True,
378 include_device=True,
/external/pytorch/torch/fx/
Dgraph_module.py265 include_device=False, argument
275 include_device=include_device,
292 include_device=include_device,
870 …def print_readable(self, print_output=True, include_stride=False, include_device=False, colored=Fa… argument
879 include_device,
Dgraph.py378 …verbose: bool = False, include_stride: bool = False, include_device: bool = False, colored: bool =…
388 include_device = include_device or (os.environ.get("FX_GRAPH_SHOW_DEVICE", "0") == "1")
587 device_annotation = f"{meta_val.device}" if include_device else ""
1400 …verbose: bool = False, include_stride: bool = False, include_device: bool = False, colored: bool =…
1462 … verbose=verbose, include_stride=include_stride, include_device=include_device, colored=colored
1467 …verbose: bool = False, include_stride: bool = False, include_device: bool = False, colored: bool =…
1471 … verbose=verbose, include_stride=include_stride, include_device=include_device, colored=colored
/external/pytorch/torch/export/
Dunflatten.py161 include_device=False, argument
169 include_device,
516 include_device=False, argument
524 include_device,
/external/pytorch/torch/_dynamo/
Dcompiled_autograd.py322 include_device=True,
Doutput_graph.py1343 name, gm, include_stride=True, include_device=True, colored=True
1350 print_output=False, include_stride=True, include_device=True
/external/pytorch/test/expect/
DTestFXAPIBackwardCompatibility.test_function_back_compat-fx_backcompat_function_signatures.expect26 …ot_module: str, verbose: bool = False, include_stride: bool = False, include_device: bool = False,…
/external/pytorch/torch/_inductor/
Dcompile_fx.py759 "AFTER POST GRAD", gm, include_stride=True, include_device=True
765 print_output=False, include_stride=True, include_device=True