Function save_graph_to_code

tools/python/encrypt.py:105–151

(j2_env, model_tag, graph_idx, graph, output)

def save_graph_to_code(j2_env, model_tag, graph_idx, graph, output):
    graph_tag = graph.name
    template_name = "tensor_source.jinja2"
    counter = 0
    for tensor in graph.tensors:
        # convert tensor
        source = j2_env.get_template(template_name).render(
            tensor=tensor,
            tensor_id=counter,
            model_tag=model_tag,
            graph_tag=graph_tag,
        )
        cc_path = output + "/" + graph_tag + "_tensor" + str(counter) + ".cc"
        with open(cc_path, "w") as f:
            f.write(source)
        counter += 1

    template_name = "operator.jinja2"
    counter = 0
    op_size = len(graph.op)
    device = 0
    for arg in graph.arg:
        if arg.name == MaceKeyword.mace_runtime_type_str:
            device = arg.i
    for start in range(0, op_size, 10):
        source = j2_env.get_template(template_name).render(
            start=start,
            end=min(start + 10, op_size),
            net=graph,
            model_tag=model_tag,
            graph_tag=graph_tag,
            device=device,
        )
        cc_path = output + "/" + graph_tag + "_op" + str(counter) + ".cc"
        with open(cc_path, "w") as f:
            f.write(source)
        counter += 1

    template_name = "graph.jinja2"
    source = j2_env.get_template(template_name).render(
        net=graph,
        graph_id=graph_idx,
        model_tag=model_tag,
        graph_tag=graph_tag,
    )
    with open(output + "/" + graph_tag + "_graph.cc", "w") as f:
        f.write(source)


def save_model_to_code(namespace, model, params, model_checksum,