experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tf.data] Delete the code to increase ram budget.

Browse Source 
PiperOrigin-RevId: 333780280
Change-Id: I69fea02ec9312a997fc15cf5a8664c60a7cb00b5
This commit is contained in:
Jay Shi 2020-09-25 12:03:49 -07:00 committed by TensorFlower Gardener
parent 0dd44657b3
commit ea7169e047
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/framework/model.cc
View File

@ -1390,9 +1390,6 @@ void Model::OptimizeGradientDescent(int64 cpu_budget, int64 ram_budget,
VLOG(2) << "Starting optimization of tunable parameters with GradientDescent";
auto parameters = CollectTunableParameters(snapshot);
auto essential_parameters = CollectEssentialParallelism(snapshot, parameters);
// We add the number of model's buffered bytes because it is excluded from the
// memory budget, but it is included in the maximum number of buffered bytes.
ram_budget += TotalBufferedBytes(snapshot);
for (auto& pair : parameters) {
pair.second->value = pair.second->min;
}
@ -1467,9 +1464,6 @@ void Model::OptimizeHillClimb(int64 cpu_budget, int64 ram_budget,
VLOG(2) << "Starting optimization of tunable parameters with HillClimb";
const double processing_time = TotalProcessingTime(snapshot);
auto parameters = CollectTunableParameters(snapshot);
// We add the number of model's buffered bytes because it is excluded from the
// memory budget, but it is included in the maximum number of buffered bytes.
ram_budget += TotalBufferedBytes(snapshot);
// Buffer size parameter will only be incremented if the output latency
// improvement is greater than this constant.
constexpr double kBufferSizeMinDelta = 1.0L;