MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / HandleFuseRecvGroup

Function HandleFuseRecvGroup

tensorflow/core/graph/graph_partition.cc:939–964  ·  view source on GitHub ↗

Grouping by 'srcp + dstp + host'

Source from the content-addressed store, hash-verified

937// Grouping by 'srcp + dstp + host'
938//
939Status HandleFuseRecvGroup(const PartitionOptions& opts,
940 const GraphInfo& g_info,
941 const DupFuseRecvTable group,
942 GraphDef* dst_graph,
943 GlobalDupFuseRecvTable* dup_global_fuse_recv) {
944 Status status;
945 DupFuseRecvTable gather_inputs;
946 DupFuseRecvTable no_gather_inputs;
947 SplitByGatherInputs(group, &gather_inputs, &no_gather_inputs);
948
949 if (gather_inputs.size() > 0) {
950 status = DoFuseRecv(opts, g_info, gather_inputs, dst_graph, dup_global_fuse_recv);
951 if (!status.ok()) {
952 return status;
953 }
954 }
955
956 if (no_gather_inputs.size() > 0) {
957 status = DoFuseRecv(opts, g_info, no_gather_inputs, dst_graph, dup_global_fuse_recv);
958 if (!status.ok()) {
959 return status;
960 }
961 }
962
963 return status;
964}
965
966Status HandleFuseRecvGroup(const PartitionOptions& opts,
967 const GraphInfo& g_info,

Callers 1

PartitionWithTensorFuseFunction · 0.85

Calls 5

SplitByGatherInputsFunction · 0.85
DoFuseRecvFunction · 0.85
SplitDupFuseRecvTablesFunction · 0.85
sizeMethod · 0.45
okMethod · 0.45

Tested by

no test coverage detected