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

Method DropMaxLayer

oneflow/core/auto_parallel/sbp_node.cpp:565–567  ·  view source on GitHub ↗

Drop down the maximum layer with the minimum layer form consumer

Source from the content-addressed store, hash-verified

563
564// Drop down the maximum layer with the minimum layer form consumer
565void SbpNode::DropMaxLayer(int32_t upper_bound) {
566 if (upper_bound < max_layer_ || max_layer_ < 0) { max_layer_ = upper_bound; }
567}
568
569// Set max_layer_ = min_layer_ if this node does not have any consumer
570// This is the end of the whole graph

Callers 1

SpreadMaxLayerMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected