Function MarkStreamSubGraph

tensorflow/core/graph/stream_subgraph.cc:48–55

} // namesapce

void MarkStreamSubGraph(Graph* g, const MultiStreamOptions& opt) {
  int num_streams = opt.multi_stream_num();
  MultiStreamPartitionPolicy policy = opt.partition_policy();

  if (policy == MultiStreamPartitionPolicy::EMBEDDING_GRAPH_PARTITION) {
    MarkEmbeddingGraph(g, num_streams);
  }
}

void MarkEmbeddingGraph(Graph* g, int num_streams) {
  bool train_graph = false;