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

Method GetParallelDesc4BnInOp

oneflow/core/operator/operator.cpp:123–128  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

121}
122
123Maybe<const ParallelDesc> Operator::GetParallelDesc4BnInOp(const std::string& bn) const {
124 CHECK_OR_RETURN(bn2parallel_desc_);
125 auto it = bn2parallel_desc_->find(bn);
126 CHECK_OR_RETURN(it != bn2parallel_desc_->end());
127 return it->second;
128}
129
130Maybe<void> Operator::FillBlobParallelDesc(
131 const std::function<Maybe<const ParallelDesc>(const std::string&)>& ParallelDesc4Bn) {

Callers 7

InitializeMemoryMethod · 0.80
InitCopyAndMemoryCostMethod · 0.80
InitializeMemoryMethod · 0.80
GroupBoxingByDstParallelFunction · 0.80
AddAndInferOpMethod · 0.80
CheckPhysicalBlobDescFunction · 0.80

Calls 2

findMethod · 0.80
endMethod · 0.45

Tested by

no test coverage detected