diff options
author | Brennan Saeta <saeta@google.com> | 2018-10-09 11:54:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 11:58:43 -0700 |
commit | 072fcb995a3fd658ee2461b59b159498c710513d (patch) | |
tree | f3def3d3ac6e270ad32e428889a79d662c8bc9cf /tensorflow/core/framework/model.h | |
parent | 12e164d1e7c0b197f06d5d3c2ed26318b89b5e4c (diff) |
[tf.data] NUMA-aware MapAndBatch dataset.
PiperOrigin-RevId: 216395709
Diffstat (limited to 'tensorflow/core/framework/model.h')
-rw-r--r-- | tensorflow/core/framework/model.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/framework/model.h b/tensorflow/core/framework/model.h index eae0fa70e8..9596252664 100644 --- a/tensorflow/core/framework/model.h +++ b/tensorflow/core/framework/model.h @@ -335,7 +335,7 @@ class Model { if (name_ == "Map") { return Type::MAP; } - if (name_ == "MapAndBatch") { + if (name_ == "MapAndBatch" || name_ == "NumaMapAndBatch") { return Type::MAP_AND_BATCH; } if (name_ == "PaddedBatch") { |