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

[tf.data] Reverting portion of cl/322686110, changing back model input time to be zero (i.e. infinitely fast consumer).

Browse Source 
PiperOrigin-RevId: 323628170
Change-Id: Ia6dd4fe48985c431b84c6189e5622e29fde4f8f9
This commit is contained in:
Jiri Simsa 2020-07-28 12:26:00 -07:00 committed by TensorFlower Gardener
parent 32a84465d3
commit f72b707dfc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/data/model_dataset_op.cc
View File

@ -204,7 +204,7 @@ class ModelDatasetOp : public UnaryDatasetOpKernel {
model_input_time = SelfInputTime();
}
model_->Optimize(dataset()->algorithm_, dataset()->cpu_budget_,
dataset()->ram_budget_, model_input_time);
dataset()->ram_budget_, /*model_input_time=*/0);
// Exponentially increase the period of running the optimization
// until a threshold is reached.
if (optimization_period_ms != kOptimizationPeriodThresholdMs) {