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

Move class attributes to top.

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