diff options
Diffstat (limited to 'tensorflow/core/util/tensor_slice_writer.h')
| -rw-r--r-- | tensorflow/core/util/tensor_slice_writer.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/util/tensor_slice_writer.h b/tensorflow/core/util/tensor_slice_writer.h index 34aa7c0b916..887db9d5c09 100644 --- a/tensorflow/core/util/tensor_slice_writer.h +++ b/tensorflow/core/util/tensor_slice_writer.h @@ -83,7 +83,8 @@ class TensorSliceWriter { const string filename_; const CreateBuilderFunction create_builder_; - const string tmpname_; + string data_filename_; + bool use_temp_file_; // A mapping from the tensor names to their index in meta_.saved_slice_meta() std::unordered_map<string, int> name_to_index_; |