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 data_feeder import DataFeeder
from core import LoDTensor, CPUPlace, CUDAPlace from core import LoDTensor, CPUPlace, CUDAPlace
from distribute_transpiler import DistributeTranspiler from distribute_transpiler import DistributeTranspiler
from distribute_transpiler_simple import SimpleDistributeTranspiler
import clip import clip
Tensor = LoDTensor Tensor = LoDTensor
__all__ = framework.__all__ + executor.__all__ + [ __all__ = framework.__all__ + executor.__all__ + [
'io', 'initializer', 'layers', 'nets', 'optimizer', 'backward', 'io', 'initializer', 'layers', 'nets', 'optimizer', 'backward',
'regularizer', 'LoDTensor', 'CPUPlace', 'CUDAPlace', 'Tensor', 'ParamAttr' '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 return param_grad_map
class DistributeTranspiler: class SimpleDistributeTranspiler:
def transpile(self, def transpile(self,
optimize_ops, optimize_ops,
params_grads, params_grads,

Loading…
Cancel
Save