Function build

tensorboard/plugins/graph/tf_graph_common/graph.ts:1018–1277

(
  graphDef: tf_graph_proto.GraphDef,
  params: BuildParams,
  tracker: ProgressTracker
)

  },
};
export function build(
  graphDef: tf_graph_proto.GraphDef,
  params: BuildParams,
  tracker: ProgressTracker
): Promise<SlimGraph> {
  /**
   * A dictionary that maps each in-embedding node name to the node
   * object.
   */
  let inEmbedding: {
    [nodeName: string]: OpNode;
  } = {};
  /**
   * A dictionary that maps each out-embedding node name to the node
   * object.
   */
  let outEmbedding: {
    [nodeName: string]: OpNode;
  } = {};
  /**
   * A dictionary that maps each node name to an array of the node's
   * out-embedding node label objects.
   */
  let outEmbeddings: {
    [inputName: string]: OpNode[];
  } = {};
  let isInEmbeddedPred = getEmbedPredicate(params.inEmbeddingTypes);
  let isOutEmbeddedPred = getEmbedPredicate(params.outEmbeddingTypes);
  let embeddingNodeNames: string[] = [];
  let rawNodes = graphDef.node;
  /**
   * A list of all the non-embedding node names which appear in the processed
   * list of raw nodes. Here we pre-allocate enough room for all the rawNodes,
   * even though there will some number of embeddings. The excess array length
   * is spliced off later.
   *
   * Experimentation shows that around 30% of the array will go unused, and
   * even for very large networks that amounts to less than 10k spaces.
   */
  let nodeNames = new Array<string>(rawNodes.length);
  return tf_graph_util
    .runAsyncTask(
      'Normalizing names',
      30,
      () => {
        let opNodes = new Array<OpNode>(rawNodes.length);
        let index = 0;
        const processRawNode = (rawNode) => {
          let opNode = new OpNodeImpl(rawNode);
          if (isInEmbeddedPred(opNode)) {
            embeddingNodeNames.push(opNode.name);
            inEmbedding[opNode.name] = opNode;
            return opNode;
          }
          if (isOutEmbeddedPred(opNode)) {
            embeddingNodeNames.push(opNode.name);
            outEmbedding[opNode.name] = opNode;
            _.each(opNode.inputs, (input) => {