From e0857cc6da71dbbaeb683fbf69fdf01be3a807d5 Mon Sep 17 00:00:00 2001 From: Derek Murray Date: Thu, 5 Mar 2020 16:17:25 -0800 Subject: [PATCH] [tf.data] Avoid creating and destroying unused tensor in RangeDataset iterator. PiperOrigin-RevId: 299221519 Change-Id: I87d11cbafd07be4bfd9218bc575b52a5c29daf0a --- tensorflow/core/kernels/data/range_dataset_op.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/tensorflow/core/kernels/data/range_dataset_op.cc b/tensorflow/core/kernels/data/range_dataset_op.cc index 6decb93d70a..a03fe82d815 100644 --- a/tensorflow/core/kernels/data/range_dataset_op.cc +++ b/tensorflow/core/kernels/data/range_dataset_op.cc @@ -107,7 +107,6 @@ class RangeDatasetOp::Dataset : public DatasetBase { return Status::OK(); } out_tensors->reserve(1); - Tensor result(dataset()->output_dtypes()[0]); switch (dataset()->output_dtypes()[0]) { #define HANDLE_TYPE(type) \ case DataTypeToEnum::value: { \