Skip to content

[RFC] Equivalent subgraphs and compiled subgraphs #7301

@justinchuby

Description

@justinchuby

Please scroll to #7301 (comment) for the up-to-date proto def.

# EquivalentSubGraphProto
# The equivalent_subgraph is a field in GraphProto
# We specify the inputs and outputs, such that an alternative route
# can be specified in the computation graph
# This way, multiple equivalent subgraph (potentially overlapping) can be stored in the model, each independent and not interfering
repeated equivalent_subgraph {
    # This can call an op, or a local function. This allows code reuse by now storing the actual function body in the proto message
    repeated string input
    repeated string output
    # No need to specify insertion point
    # + attributes
    string domain
    string op_type
    [...] metadata_props
}

Compiled graph is stored in a Node

Node:
    op_type: Compiled
    inputs
        *values
    outputs
        *values
    attributes
        {other attributes passed to the binary}
        __hash__: str
        __identifier__: str
        __binary__: StringTensor<[]>

Metadata

Metadata

Assignees

Labels

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions