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

[tf.data] Add experimental dataset ops as passthrough ops in prefetch with slack rewrite.

Browse Source 
PiperOrigin-RevId: 251566961
This commit is contained in:
Rachel Lim 2019-06-04 19:58:50 -07:00 committed by TensorFlower Gardener
parent 76171c3cba
commit a9e618682b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/grappler/optimizers/data/slack.cc
View File

@ -51,8 +51,10 @@ bool IsDatasetNodeOfType(const NodeDef& node,
constexpr std::array<const char*, 2> kMultipleInputsDatasetOps = {
"ZipDataset", "ConcatenateDataset"};
constexpr std::array<const char*, 15> kPassThroughOps = {
constexpr std::array<const char*, 17> kPassThroughOps = {
"CacheDataset",
"ExperimentalMaxIntraOpParallelismDataset",
"ExperimentalPrivateThreadPoolDataset",
"FilterDataset",
"Identity",
"MapDataset",