Python _initialize_aten_bindings示例

示例#1

import os
import torch
from .version import __version__
from .optim import *
from .ops import *
import _torch_ipex

_torch_ipex._initialize_aten_bindings()

示例#2

文件： test_lazy_reorder.py 项目： pinzhenx/intel-extension-for-pytorch

"""Tests for lazy reorder."""
from __future__ import division
from __future__ import print_function

import os
import math
import time
import random
import unittest
from functools import reduce
import copy
import sys
import torch
import _torch_ipex as ipex
ipex._initialize_aten_bindings()
import intel_pytorch_extension_py

import torch.nn as nn
import torch.backends.cudnn as cudnn
from torch.nn import Parameter
import torch.nn.functional as F
from torch.autograd import gradcheck
from torch.autograd.gradcheck import gradgradcheck
from torch._six import inf, nan

from common_utils import TestCase, iter_indices, TEST_NUMPY, TEST_SCIPY, TEST_MKL, \
    TEST_LIBROSA, run_tests, download_file, skipIfNoLapack, suppress_warnings, \
    IS_WINDOWS, PY3, NO_MULTIPROCESSING_SPAWN, do_test_dtypes, do_test_empty_full, \
    IS_SANDCASTLE, load_tests, brute_pdist, brute_cdist, slowTest, \
    skipCUDANonDefaultStreamIf, skipCUDAMemoryLeakCheckIf

示例#3

文件： __init__.py 项目： JianpingChen066/intel-extension-for-pytorch

import os
import torch
from .version import __version__
from .optim import *
from .ops import *
import _torch_ipex as core

core._initialize_aten_bindings()