mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Part of #123062 - #123062 Pull Request resolved: https://github.com/pytorch/pytorch/pull/123723 Approved by: https://github.com/Skylion007
15 lines
441 B
Python
15 lines
441 B
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
__all__ = ["make_fx", "dispatch_trace", "PythonKeyTracer", "pythonkey_decompose"]
|
|
from torch.fx.experimental.proxy_tensor import (
|
|
decompose,
|
|
dispatch_trace,
|
|
make_fx,
|
|
PythonKeyTracer,
|
|
)
|
|
|
|
pythonkey_decompose = decompose
|