|
|
@ -44,8 +44,8 @@ import transpiler
|
|
|
|
from param_attr import ParamAttr, WeightNormParamAttr
|
|
|
|
from param_attr import ParamAttr, WeightNormParamAttr
|
|
|
|
from data_feeder import DataFeeder
|
|
|
|
from data_feeder import DataFeeder
|
|
|
|
from core import LoDTensor, CPUPlace, CUDAPlace, CUDAPinnedPlace
|
|
|
|
from core import LoDTensor, CPUPlace, CUDAPlace, CUDAPinnedPlace
|
|
|
|
from transpiler import DistributeTranspiler, SimpleDistributeTranspiler, \
|
|
|
|
from transpiler import DistributeTranspiler, InferenceTranspiler, \
|
|
|
|
InferenceTranspiler, memory_optimize, release_memory
|
|
|
|
memory_optimize, release_memory
|
|
|
|
from concurrency import (Go, make_channel, channel_send, channel_recv,
|
|
|
|
from concurrency import (Go, make_channel, channel_send, channel_recv,
|
|
|
|
channel_close, Select)
|
|
|
|
channel_close, Select)
|
|
|
|
from lod_tensor import create_lod_tensor, create_random_int_lodtensor
|
|
|
|
from lod_tensor import create_lod_tensor, create_random_int_lodtensor
|
|
|
|