def _flatten(converter, node, inputs):
    x_in = converter.outputs[inputs[0]]
    shape = x_in.shape.as_list()
    non_batch = 1
    for dim in shape[1:]:
        non_batch *= dim
    return converter.protocol.reshape(x_in, [(- 1), non_batch])
def _flatten(converter, node, inputs):
    x_in = converter.outputs[inputs[0]]
    shape = x_in.shape.as_list()
    non_batch = 1
    for dim in shape[1:]:
        non_batch *= dim
    return converter.protocol.reshape(x_in, [(- 1), non_batch])