Fix comment to reflect actual logic.

PiperOrigin-RevId: 311483016
Change-Id: Ib66f41fd4c470bdcd516f4f03de7d78fb8ddde1c
This commit is contained in:
Taehee Jeong 2020-05-14 00:31:22 -07:00 committed by TensorFlower Gardener
parent d36ad412f4
commit da78c46560

View File

@ -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_;
};