提交 584bb9ba authored 作者: notoraptor's avatar notoraptor

Move class attributes to top.

上级 941a3192
......@@ -1665,7 +1665,8 @@ class GpuDnnBatchNorm(DnnBase):
__props__ = ('mode', 'running_averages', 'inplace_running_mean',
'inplace_running_var', 'inplace_output')
_cop_num_inputs = 7
_cop_num_outputs = 5
check_input = False
params_type = ParamsType(mode=cudnn.cudnnBatchNormMode_t,
inplace_output=bool_t,
......@@ -1710,9 +1711,6 @@ class GpuDnnBatchNorm(DnnBase):
def infer_shape(self, node, shape):
return [shape[0]] + [shape[1]] * (len(node.outputs) - 1)
_cop_num_inputs = 7
_cop_num_outputs = 5
def make_node(self, x, scale, bias, epsilon=1e-4,
running_average_factor=0.1,
running_mean=None, running_var=None):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论