MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / MakeSplitSbpParallel

Function MakeSplitSbpParallel

oneflow/core/job/sbp_parallel.cpp:69–74  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

67}
68
69Maybe<Symbol<SbpParallel>> MakeSplitSbpParallel(int axis) {
70 CHECK_LT_OR_RETURN(axis, kMaxSplitAxis);
71 SbpParallel split_sbp_parallel;
72 split_sbp_parallel.mutable_split_parallel()->set_axis(axis);
73 return SymbolOf(split_sbp_parallel);
74}
75
76Maybe<Symbol<SbpParallel>> MakeBroadcastSbpParallel() {
77 SbpParallel broadcast_sbp;

Callers 1

MakeSplitSbpParallelListFunction · 0.85

Calls 1

SymbolOfFunction · 0.85

Tested by

no test coverage detected