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

[tf.data] Update the input time computation in the InterleaveMany node.

Browse Source 
PiperOrigin-RevId: 315331988
Change-Id: I105dd74bb87092f2d56781cdb8e28d2c39360100
This commit is contained in:
A. Unique TensorFlower 2020-06-08 12:44:45 -07:00 committed by TensorFlower Gardener
parent 73d8cfcf8b
commit 05355c404a
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -59,9 +59,8 @@ class InterleaveMany : public Node {
(*input_times)[long_name()] = old_input_time;
return;
}
double new_input_time =
old_input_time +
SelfProcessingTimeLocked() * static_cast<double>(num_inputs() - 1);
double new_input_time = (old_input_time + SelfProcessingTimeLocked()) *
static_cast<double>(num_inputs() - 1);
(*input_times)[long_name()] = new_input_time;
}