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

[tf.data] Avoid creating and destroying unused tensor in RangeDataset iterator.

Browse Source 
PiperOrigin-RevId: 299221519
Change-Id: I87d11cbafd07be4bfd9218bc575b52a5c29daf0a
This commit is contained in:
Derek Murray 2020-03-05 16:17:25 -08:00 committed by TensorFlower Gardener
parent 44506a0e68
commit e0857cc6da
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/data/range_dataset_op.cc
View File

@ -107,7 +107,6 @@ class RangeDatasetOp::Dataset : public DatasetBase {
return Status::OK(); return Status::OK();
} }
out_tensors->reserve(1); out_tensors->reserve(1);
Tensor result(dataset()->output_dtypes()[0]);
switch (dataset()->output_dtypes()[0]) { switch (dataset()->output_dtypes()[0]) {
#define HANDLE_TYPE(type) \ #define HANDLE_TYPE(type) \
case DataTypeToEnum<type>::value: { \ case DataTypeToEnum<type>::value: { \