follow comments

add_depthwiseConv_op_gpu
typhoonzero 8 years ago
parent c24da0d3ee
commit 06b326b681

@ -18,13 +18,14 @@ from param_attr import ParamAttr
from data_feeder import DataFeeder
from core import LoDTensor, CPUPlace, CUDAPlace
from distribute_transpiler import DistributeTranspiler
from distribute_transpiler_simple import SimpleDistributeTranspiler
import clip
Tensor = LoDTensor
__all__ = framework.__all__ + executor.__all__ + [
'io', 'initializer', 'layers', 'nets', 'optimizer', 'backward',
'regularizer', 'LoDTensor', 'CPUPlace', 'CUDAPlace', 'Tensor', 'ParamAttr'
'DataFeeder', 'clip', 'DistributeTranspiler'
'DataFeeder', 'clip', 'SimpleDistributeTranspiler', 'DistributeTranspiler'
]

@ -48,7 +48,7 @@ def round_robin(params_grads, pserver_endpoints):
return param_grad_map
class DistributeTranspiler:
class SimpleDistributeTranspiler:
def transpile(self,
optimize_ops,
params_grads,

Loading…
Cancel
Save