提交 487181d6 编写于 作者: L liuqi

Format some python code.

上级 84bf8f31
......@@ -436,10 +436,10 @@ class TFConverter(object):
return False
filter_shape = get_input_tensor(op, 1).shape.as_list()
input_shape = get_input_tensor(op, 0).shape.as_list()
return input_shape[1] == filter_shape[0] and \
input_shape[2] == filter_shape[1] and \
(op.get_attr('padding') == 'VALID' or filter_shape[0] == 1 and
filter_shape[1] == 1)
return input_shape[1] == filter_shape[0] \
and input_shape[2] == filter_shape[1] \
and (op.get_attr('padding') == 'VALID' or filter_shape[0] == 1
and filter_shape[1] == 1)
def convert_global_conv_to_fc(self, op):
op_def = mace_pb2.OperatorDef()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册