| import warnings | |
| from torch.onnx import _CAFFE2_ATEN_FALLBACK | |
| if not _CAFFE2_ATEN_FALLBACK: | |
| warnings.warn("Caffe2 support is not fully enabled in this PyTorch build. " | |
| "Please enable Caffe2 by building PyTorch from source with `BUILD_CAFFE2=1` flag.") | |