prim2code.py 28.6 KB
Newer Older
S
SunAhong1993 已提交
1
# -*- coding:UTF-8 -*-
S
SunAhong1993 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14 15
#   Copyright (c) 2020  PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

S
SunAhong1993 已提交
16
NO_OUTPUT_COUNT = 0
S
SunAhong1993 已提交
17 18


S
SunAhong1993 已提交
19 20 21 22 23 24 25 26 27 28 29
def gen_codes(code_list, indent=0):
    indent_blank = "    " * indent
    codes = []
    for code_line in code_list:
        if code_line.strip() == "":
            codes.append('\n')
        else:
            codes.append(indent_blank + code_line + '\n')
    return codes


S
SunAhong1993 已提交
30
def get_value(layer, key, layer_id=None, different_attrs=None):
S
SunAhong1993 已提交
31 32 33 34 35 36
    """ 进行optimizer后可能把inputs的value直接用数值代替(ConstantFuser),
        会把input换成attr,所以需要此处的操作。
    """
    if key in layer.inputs:
        return layer.inputs[key]
    else:
S
SunAhong1993 已提交
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
        if different_attrs is None:
            return str(layer.attrs[key])
        else:
            key_name = "{}_{}".format(layer.outputs[0], key)
            if key_name in different_attrs:
                return key_name
            else:
                if layer_id is None:
                    return str(layer.attrs[key])
                key_name = "{}_{}".format("layer_id/{}".format(layer_id), key)
                if key_name in different_attrs:
                    new_key_name = "attr_{}".format(NO_OUTPUT_COUNT)
                    NO_OUTPUT_COUNT += 1
                    diff_index = different_attrs.index(key_name)
                    different_attrs[diff_index] = new_key_name
                    return new_key_name
                else:
                    return str(layer.attrs[key])


S
SunAhong1993 已提交
57 58 59 60 61 62
def prim_add(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
S
SunAhong1993 已提交
63
    line = "{} = {} + {}".format(layer.outputs[0],
S
SunAhong1993 已提交
64 65
                                 get_value(layer, "x", different_attrs),
                                 get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
66 67 68
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
69 70 71 72 73 74
def prim_add_(layer,
              indent=1,
              init_func=[],
              forward_func=[],
              layer_id=None,
              different_attrs=None):
S
SunAhong1993 已提交
75
    line = "{} = {} + {} * {}".format(layer.outputs[0],
S
SunAhong1993 已提交
76
                                      get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
77
                                      layer.attrs["alpha"],
S
SunAhong1993 已提交
78
                                      get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
79 80 81
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
82 83 84 85 86 87 88
def prim_and(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None,
             is_return_line=False):
S
SunAhong1993 已提交
89
    line = "{} = {} and {}".format(layer.outputs[0],
S
SunAhong1993 已提交
90 91
                                   get_value(layer, "x", different_attrs),
                                   get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
92 93
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
94 95 96
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
97 98 99 100 101 102
def prim_append(layer,
                indent=1,
                init_func=[],
                forward_func=[],
                layer_id=None,
                different_attrs=None):
S
SunAhong1993 已提交
103
    line = "{}.append({})".format(
S
SunAhong1993 已提交
104
        get_value(layer, "list", layer_id, different_attrs),
S
SunAhong1993 已提交
105
        get_value(layer, "element", layer_id, different_attrs))
S
SunAhong1993 已提交
106 107 108
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
109 110 111 112 113 114
def prim_assert(layer,
                indent=1,
                init_func=[],
                forward_func=[],
                layer_id=None,
                different_attrs=None):
S
SunAhong1993 已提交
115 116 117 118 119 120 121 122 123 124
    if layer.attrs["type"] == "eq":
        values = get_value(layer, "key")
        if "value" in layer.attrs:
            values = layer.attrs["value"]
        if isinstance(values, list):
            s = ""
            for v in values:
                s += "{} == {} or ".format(get_value(layer, "key"), v)
            if len(s) > 0:
                s = s[:-4]
S
SunAhong1993 已提交
125
            lc = locals()
S
SunAhong1993 已提交
126 127
            exec("assert_result = {}".format(s))
            assert_result = lc['assert_result']
S
SunAhong1993 已提交
128 129 130
            line = "assert {}, \'The {} must be {}!\'".format(
                s, get_value(layer, "key"), get_value(layer, "value"))
        else:
S
SunAhong1993 已提交
131 132 133
            s = "{} == {}".format(
                get_value(layer, "key"), get_value(layer, "value"))
            lc = locals()
S
SunAhong1993 已提交
134 135 136 137
            exec("assert_result = {}".format(s))
            assert_result = lc['assert_result']
            line = "assert {}, \'The {} must be {}!\'".format(
                s, get_value(layer, "key"), get_value(layer, "value"))
S
SunAhong1993 已提交
138 139
    else:
        raise Exception("Not implement yet!")
S
SunAhong1993 已提交
140 141
    if not assert_result:
        forward_func.extend(gen_codes([line], indent=indent))
S
SunAhong1993 已提交
142 143


S
SunAhong1993 已提交
144 145 146 147 148 149
def prim_check_dim(layer,
                   indent=1,
                   init_func=[],
                   forward_func=[],
                   layer_id=None,
                   different_attrs=None):
S
SunAhong1993 已提交
150
    lines = []
S
SunAhong1993 已提交
151 152
    dim = get_value(layer, "dim", different_attrs)
    lines.append("if {} < 0:".format(dim))
S
SunAhong1993 已提交
153
    lines.append("    {} = {} + {}".format(layer.outputs[
S
SunAhong1993 已提交
154
        0], dim, get_value(layer, "len", different_attrs)))
S
SunAhong1993 已提交
155
    lines.append("else:")
S
SunAhong1993 已提交
156
    lines.append("    {} = {}".format(layer.outputs[0], dim))
S
SunAhong1993 已提交
157 158 159
    forward_func.extend(gen_codes(lines, indent=indent))


S
SunAhong1993 已提交
160 161 162 163 164 165
def prim_constant(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
S
SunAhong1993 已提交
166 167 168 169
    line = "{} = {}".format(layer.outputs[0], layer.attrs["value"])
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
170 171 172 173 174 175 176
def prim_contain(layer,
                 indent=1,
                 init_func=[],
                 forward_func=[],
                 layer_id=None,
                 different_attrs=None,
                 is_return_line=False):
S
SunAhong1993 已提交
177
    line = "{} = {} in {}".format(layer.outputs[0],
S
SunAhong1993 已提交
178 179
                                  get_value(layer, "element", different_attrs),
                                  get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
180 181
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
182 183 184
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
185 186 187 188 189 190
def prim_dict(layer,
              indent=1,
              init_func=[],
              forward_func=[],
              layer_id=None,
              different_attrs=None):
S
SunAhong1993 已提交
191 192
    line = "{} = dict()".format(layer.outputs[0])
    forward_func.extend(gen_codes([line], indent=indent))
S
SunAhong1993 已提交
193 194 195 196 197 198 199 200


def prim_dict_construct(layer,
                        indent=1,
                        init_func=[],
                        forward_func=[],
                        layer_id=None,
                        different_attrs=None):
S
SunAhong1993 已提交
201
    lines = list()
S
SunAhong1993 已提交
202
    line = "{} = dict()".format(layer.outputs[0])
S
SunAhong1993 已提交
203 204
    lines.append(line)
    for i in range(len(layer.inputs)):
S
SunAhong1993 已提交
205 206 207 208
        line = "{}[{}] = {}".format(
            layer.outputs[0],
            get_value(layer, "key{}".format(i), different_attrs),
            get_value(layer, "value{}".format(i), different_attrs))
S
SunAhong1993 已提交
209 210
        lines.append(line)
    forward_func.extend(gen_codes(lines, indent=indent))
S
SunAhong1993 已提交
211 212 213 214 215 216 217 218 219 220


def prim_dict2values(layer,
                     indent=1,
                     init_func=[],
                     forward_func=[],
                     layer_id=None,
                     different_attrs=None):
    line = "{} = list({}.values())".format(
        layer.outputs[0], get_value(layer, "x", different_attrs))
S
SunAhong1993 已提交
221 222 223
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
224 225 226 227 228 229
def prim_div(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
S
SunAhong1993 已提交
230
    line = "{} = {} / {}".format(layer.outputs[0],
S
SunAhong1993 已提交
231
                                 get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
232
                                 get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
233 234 235
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
236 237 238 239 240 241 242
def prim_eq(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
243
    line = "{} = {} == {}".format(layer.outputs[0],
S
SunAhong1993 已提交
244
                                  get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
245
                                  get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
246 247
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
248 249 250
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
251 252 253 254 255 256 257 258
def prim_equal(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None):
    line = "{} = {}".format(layer.outputs[0],
                            get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
259 260 261
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
262 263 264 265 266 267 268 269
def prim_exception(layer,
                   indent=1,
                   init_func=[],
                   forward_func=[],
                   layer_id=None,
                   different_attrs=None):
    line = "raise Exception({})".format(
        get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
270 271 272
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
273 274 275 276 277 278 279 280
def prim_float(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None):
    line = "{} = float({})".format(layer.outputs[0],
                                   get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
281 282 283
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
284 285 286 287 288 289
def prim_floor(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None):
S
SunAhong1993 已提交
290
    line = "{} = math.floor({})".format(layer.outputs[0],
S
SunAhong1993 已提交
291
                                        get_value(layer, "x", different_attrs))
S
SunAhong1993 已提交
292 293 294
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
295 296 297 298 299 300
def prim_floordiv(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
S
SunAhong1993 已提交
301
    line = "{} = {} // {}".format(layer.outputs[0],
S
SunAhong1993 已提交
302
                                  get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
303
                                  get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
304 305 306
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
307 308 309 310 311 312
def prim_getitem(layer,
                 indent=1,
                 init_func=[],
                 forward_func=[],
                 layer_id=None,
                 different_attrs=None):
S
SunAhong1993 已提交
313
    line = "{} = {}[{}]".format(layer.outputs[0],
S
SunAhong1993 已提交
314 315
                                get_value(layer, "list", different_attrs),
                                get_value(layer, "index", different_attrs))
S
SunAhong1993 已提交
316 317 318
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
319 320 321 322 323 324 325
def prim_gt(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
326
    line = "{} = {} > {}".format(layer.outputs[0],
S
SunAhong1993 已提交
327
                                 get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
328
                                 get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
329 330
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
331 332 333
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
334 335 336 337 338 339
def prim_if(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None):
S
SunAhong1993 已提交
340 341 342 343
    try:
        exec_s = None
        for line in forward_func:
            s = line.replace("    ", "")
S
SunAhong1993 已提交
344 345
            if s.startswith("{} = ".format(
                    get_value(layer, "input", different_attrs))):
S
SunAhong1993 已提交
346
                exec_s = s.split(" = ")[1]
S
SunAhong1993 已提交
347
        lc = locals()
S
SunAhong1993 已提交
348 349 350
        if exec_s is not None:
            exec("if_result = {}".format(exec_s))
        else:
S
SunAhong1993 已提交
351 352
            exec("if_result = {}".format(
                get_value(layer, "input", different_attrs)))
S
SunAhong1993 已提交
353 354 355 356 357 358
        if_result = lc['if_result']
        if if_result:
            block = layer.blocks[0]
        else:
            block = layer.blocks[1]
        if len(block.layers) > 0:
S
SunAhong1993 已提交
359 360
            b_init_lines, b_forward_lines = block.gen_dygraph_code(
                indent=indent)
S
SunAhong1993 已提交
361 362 363 364 365 366 367 368 369 370
            init_func.extend(b_init_lines)
            forward_func.extend(b_forward_lines)
    except:
        line = "if {} :".format(get_value(layer, "input", different_attrs))
        forward_func.extend(gen_codes([line], indent=indent))
        block = layer.blocks[0]
        if len(block.layers) == 0:
            line = "pass"
            forward_func.extend(gen_codes([line], indent=indent + 1))
        else:
S
SunAhong1993 已提交
371 372
            b_init_lines, b_forward_lines = block.gen_dygraph_code(
                indent=indent + 1)
S
SunAhong1993 已提交
373 374 375 376 377 378 379 380 381 382 383
            init_func.extend(b_init_lines)
            forward_func.extend(b_forward_lines)
        block = layer.blocks[1]
        if len(block.layers) > 0:
            b_init_lines, b_forward_lines = block.gen_dygraph_code(
                indent=indent + 1)
            if len(b_forward_lines) != 0:
                line = "else:"
                forward_func.extend(gen_codes([line], indent=indent))
            init_func.extend(b_init_lines)
            forward_func.extend(b_forward_lines)
S
SunAhong1993 已提交
384 385


S
SunAhong1993 已提交
386 387 388 389 390 391 392 393
def prim_int(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
    line = "{} = int({})".format(layer.outputs[0],
                                 get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
394 395 396
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
397 398 399 400 401 402 403
def prim_is(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
404
    line = "{} = {} is {}".format(layer.outputs[0],
S
SunAhong1993 已提交
405
                                  get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
406
                                  get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
407 408
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
409 410 411
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
412 413 414 415 416 417 418 419 420 421
def prim_isinstance(layer,
                    indent=1,
                    init_func=[],
                    forward_func=[],
                    layer_id=None,
                    different_attrs=None,
                    is_return_line=False):
    line = "{} = isinstance({}, {})".format(
        layer.outputs[0],
        get_value(layer, "input", different_attrs), layer.attrs["cls"])
S
SunAhong1993 已提交
422 423
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
424 425 426
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
427 428 429 430 431 432 433
def prim_isnot(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None,
               is_return_line=False):
S
SunAhong1993 已提交
434
    line = "{} = {} is not {}".format(layer.outputs[0],
S
SunAhong1993 已提交
435 436
                                      get_value(layer, "x", different_attrs),
                                      get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
437 438
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
439 440 441
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
442 443 444 445 446 447 448
def prim_le(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
449
    line = "{} = {} <= {}".format(layer.outputs[0],
S
SunAhong1993 已提交
450
                                  get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
451
                                  get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
452 453
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
454 455 456
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
457 458 459 460 461 462 463 464
def prim_len(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
    line = "{} = len({})".format(layer.outputs[0],
                                 get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
465 466 467
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
468 469 470 471 472 473
def prim_len2list(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
S
SunAhong1993 已提交
474 475
    lines = []
    lines.append("{} = []".format(layer.outputs[0]))
S
SunAhong1993 已提交
476 477
    lines.append("for i in range({}):".format(
        get_value(layer, "len", different_attrs)))
S
SunAhong1993 已提交
478 479 480 481
    lines.append("    {}.append(i)".format(layer.outputs[0]))
    forward_func.extend(gen_codes(lines, indent=indent))


S
SunAhong1993 已提交
482 483 484 485 486 487 488
def prim_lt(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
489
    line = "{} = {} < {}".format(layer.outputs[0],
S
SunAhong1993 已提交
490
                                 get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
491
                                 get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
492 493
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
494 495 496
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
497 498 499 500 501 502
def prim_list(layer,
              indent=1,
              init_func=[],
              forward_func=[],
              layer_id=None,
              different_attrs=None):
S
SunAhong1993 已提交
503 504 505
    input_len = len(layer.inputs) + len(layer.attrs)
    inputs_list = list()
    for i in range(input_len):
S
SunAhong1993 已提交
506 507
        inputs_list.append(
            get_value(layer, "input{}".format(i), different_attrs))
S
SunAhong1993 已提交
508 509 510 511 512
    inputs_str = ', '.join(inputs_list)
    line = "{} = [{}]".format(layer.outputs[0], inputs_str)
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
513 514 515 516 517 518 519 520
def prim_list_unpack(layer,
                     indent=1,
                     init_func=[],
                     forward_func=[],
                     layer_id=None,
                     different_attrs=None):
    line = "{} = {}".format(", ".join(layer.outputs),
                            get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
521 522 523
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
524 525 526 527 528 529
def prim_loop(layer,
              indent=1,
              init_func=[],
              forward_func=[],
              layer_id=None,
              different_attrs=None):
S
SunAhong1993 已提交
530
    loop_range = get_value(layer, "input", different_attrs)
S
SunAhong1993 已提交
531 532 533 534 535 536 537 538
    line = "for {} in range({}):".format(layer.outputs[1], loop_range)
    forward_func.extend(gen_codes([line], indent=indent))
    block = layer.blocks[0]
    b_init_lines, b_forward_lines = block.gen_dygraph_code(indent=indent + 1)
    init_func.extend(b_init_lines)
    forward_func.extend(b_forward_lines)


S
SunAhong1993 已提交
539 540 541 542 543 544 545 546
def prim_min(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
    line = "{} = min({})".format(layer.outputs[0],
                                 get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
547 548 549
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
550 551 552 553 554 555
def prim_mul(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
S
SunAhong1993 已提交
556
    line = "{} = {} * {}".format(layer.outputs[0],
S
SunAhong1993 已提交
557
                                 get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
558
                                 get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
559 560 561
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
562 563 564 565 566 567 568
def prim_ne(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
569
    line = "{} = {} != {}".format(layer.outputs[0],
S
SunAhong1993 已提交
570
                                  get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
571
                                  get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
572 573
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
574 575 576
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
577 578 579 580 581 582 583 584
def prim_neg(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
    line = "{} = -{}".format(layer.outputs[0],
                             get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
585 586 587
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
588 589 590 591 592 593 594 595 596
def prim_not(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None,
             is_return_line=False):
    line = "{} = not {}".format(layer.outputs[0],
                                get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
597 598
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
599 600 601
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
602 603 604 605 606 607 608
def prim_or(layer,
            indent=1,
            init_func=[],
            forward_func=[],
            layer_id=None,
            different_attrs=None,
            is_return_line=False):
S
SunAhong1993 已提交
609
    line = "{} = {} or {}".format(layer.outputs[0],
S
SunAhong1993 已提交
610
                                  get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
611
                                  get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
612 613
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
614 615 616
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
617 618 619 620 621 622
def prim_replaceitem(layer,
                     indent=1,
                     init_func=[],
                     forward_func=[],
                     layer_id=None,
                     different_attrs=None):
S
SunAhong1993 已提交
623
    line = "{}[{}] = {}".format(
S
SunAhong1993 已提交
624
        get_value(layer, "list", layer_id, different_attrs),
S
SunAhong1993 已提交
625
        get_value(layer, "index", layer_id, different_attrs),
S
SunAhong1993 已提交
626
        get_value(layer, "item", layer_id, different_attrs))
S
SunAhong1993 已提交
627 628 629
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
630 631 632 633 634 635 636 637
def prim_requires_grad(layer,
                       indent=1,
                       init_func=[],
                       forward_func=[],
                       layer_id=None,
                       different_attrs=None):
    line = "{} = not {}.stop_gradient".format(
        layer.outputs[0], get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
638 639 640
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
641 642 643 644 645 646 647 648 649 650 651
def prim_rsub(layer,
              indent=1,
              init_func=[],
              forward_func=[],
              layer_id=None,
              different_attrs=None):
    line = "{} = {} - {} * {}".format(
        layer.outputs[0],
        get_value(layer, "y", different_attrs),
        get_value(layer, "x", different_attrs),
        get_value(layer, "alpha", different_attrs))
S
SunAhong1993 已提交
652 653 654
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
655 656 657 658 659 660 661 662
def prim_select(layer,
                indent=1,
                init_func=[],
                forward_func=[],
                layer_id=None,
                different_attrs=None):
    line = "{} = {}[".format(layer.outputs[0],
                             get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
663 664
    for dim in range(layer.attrs["dim"]):
        line += ":, "
S
SunAhong1993 已提交
665
    line += (get_value(layer, "index", different_attrs) + "]")
S
SunAhong1993 已提交
666 667 668
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
669 670 671 672 673 674 675 676
def prim_set_attr(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
    line = "{} = {}".format(layer.outputs[0],
                            get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
677 678 679
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
680 681 682 683 684 685
def prim_set_item(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
S
SunAhong1993 已提交
686
    line = "{}[{}] = {}".format(
S
SunAhong1993 已提交
687
        get_value(layer, "dict", different_attrs),
S
SunAhong1993 已提交
688 689
        get_value(layer, "key", different_attrs),
        get_value(layer, "value", different_attrs))
S
SunAhong1993 已提交
690 691 692
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
693 694 695 696 697 698 699 700
def prim_shape(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None):
    line = "{} = {}.shape".format(layer.outputs[0],
                                  get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
701 702 703
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742
def prim_shape_dim(layer,
                   indent=1,
                   init_func=[],
                   forward_func=[],
                   layer_id=None,
                   different_attrs=None):
    line = "{} = {}.shape[{}]".format(
        layer.outputs[0],
        get_value(layer, "input", different_attrs),
        get_value(layer, "dim", different_attrs))
    forward_func.extend(gen_codes([line], indent=indent))


def prim_slice(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None):
    line = "{} = {}[{}: {}: {}]".format(
        layer.outputs[0],
        get_value(layer, "input", different_attrs),
        get_value(layer, "start", different_attrs),
        get_value(layer, "end", different_attrs),
        get_value(layer, "step", different_attrs))
    forward_func.extend(gen_codes([line], indent=indent))


def prim_startswith(layer,
                    indent=1,
                    init_func=[],
                    forward_func=[],
                    layer_id=None,
                    different_attrs=None,
                    is_return_line=False):
    line = "{} = {}.startswith({})".format(
        layer.outputs[0],
        get_value(layer, "input", different_attrs),
        get_value(layer, "start_str", different_attrs))
S
fix3  
SunAhong1993 已提交
743 744
    if is_return_line:
        return line.split(" = ")[1]
S
SunAhong1993 已提交
745
    forward_func.extend(gen_codes([line], indent=indent))
S
SunAhong1993 已提交
746 747


S
SunAhong1993 已提交
748 749 750 751 752 753 754 755
def prim_str(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
    line = "{} = str({})".format(layer.outputs[0],
                                 get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
756 757 758
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
759 760 761 762 763 764
def prim_sub(layer,
             indent=1,
             init_func=[],
             forward_func=[],
             layer_id=None,
             different_attrs=None):
S
fix  
SunAhong1993 已提交
765
    if int(float(get_value(layer, "alpha", different_attrs))) == 1:
S
SunAhong1993 已提交
766
        line = "{} = {} - {}".format(layer.outputs[0],
S
SunAhong1993 已提交
767
                                     get_value(layer, "x", different_attrs),
S
SunAhong1993 已提交
768 769
                                     get_value(layer, "y", different_attrs))
    else:
S
SunAhong1993 已提交
770 771 772 773 774
        line = "{} = {} - {} * {}".format(
            layer.outputs[0],
            get_value(layer, "x", different_attrs),
            get_value(layer, "alpha", different_attrs),
            get_value(layer, "y", different_attrs))
S
SunAhong1993 已提交
775 776 777
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
778 779 780 781 782 783
def prim_tuple(layer,
               indent=1,
               init_func=[],
               forward_func=[],
               layer_id=None,
               different_attrs=None):
S
SunAhong1993 已提交
784 785 786
    input_len = len(layer.inputs) + len(layer.attrs)
    inputs_list = list()
    for i in range(input_len):
S
SunAhong1993 已提交
787 788
        inputs_list.append(
            get_value(layer, "input{}".format(i), different_attrs))
S
SunAhong1993 已提交
789 790 791 792 793
    inputs_str = ', '.join(inputs_list)
    line = "{} = ({})".format(layer.outputs[0], inputs_str)
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
794 795 796 797 798 799
def prim_tuple_unpack(layer,
                      indent=1,
                      init_func=[],
                      forward_func=[],
                      layer_id=None,
                      different_attrs=None):
S
SunAhong1993 已提交
800
    outputs_str = ', '.join(layer.outputs)
S
SunAhong1993 已提交
801 802
    line = "{} = {}".format(outputs_str,
                            get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
803 804 805
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
806 807 808 809 810 811 812 813
def prim_type(layer,
              indent=1,
              init_func=[],
              forward_func=[],
              layer_id=None,
              different_attrs=None):
    line = "{} = {}.dtype".format(layer.outputs[0],
                                  get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
814 815 816
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
817 818 819 820 821 822 823 824
def prim_var2list(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
    line = "{} = {}.numpy().tolist()".format(
        layer.outputs[0], get_value(layer, "input", different_attrs))
S
SunAhong1993 已提交
825 826 827
    forward_func.extend(gen_codes([line], indent=indent))


S
SunAhong1993 已提交
828 829 830 831 832 833
def prim_warnings(layer,
                  indent=1,
                  init_func=[],
                  forward_func=[],
                  layer_id=None,
                  different_attrs=None):
S
SunAhong1993 已提交
834 835
    lines = ["import warnings"]
    line = "warnings.warn({}, stacklevel={})".format(
S
SunAhong1993 已提交
836
        get_value(layer, "input", different_attrs), layer.attrs["stacklevel"])
S
SunAhong1993 已提交
837 838
    lines.append(line)
    forward_func.extend(gen_codes(lines, indent=indent))