Fix comment to reflect actual logic.
PiperOrigin-RevId: 311483016 Change-Id: Ib66f41fd4c470bdcd516f4f03de7d78fb8ddde1c
This commit is contained in:
parent
d36ad412f4
commit
da78c46560
@ -191,11 +191,11 @@ class PartitionGraphIntoIndependentNodeSubsetsImpl {
|
||||
std::vector<NodeSubset>* node_subsets_;
|
||||
std::vector<NodeSubset::Type> node_type_;
|
||||
// Maps from tensor index to the epoch in which it is assigned. Also special
|
||||
// negative values of kEpochNotAssigned if not assigned, kEpochNotReady if it
|
||||
// is an input or constant.
|
||||
// negative values of kEpochNotReady if not assigned, kEpochAlwaysReady if it
|
||||
// is an input to the whole model or a constant that has no dependencies.
|
||||
std::vector<int> tensor_epochs_;
|
||||
// Maps from tensor index to the epoch in which it is assigned. Also special
|
||||
// negative values of kEpochNotAssigned if not assigned.
|
||||
// negative values of kEpochNotReady if not assigned.
|
||||
std::vector<int> node_epochs_;
|
||||
};
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user