diff --git a/caffe2/serialize/inline_container.h b/caffe2/serialize/inline_container.h index aa0cb8e0432..6a13d414feb 100644 --- a/caffe2/serialize/inline_container.h +++ b/caffe2/serialize/inline_container.h @@ -104,6 +104,7 @@ class TORCH_API ChunkRecordIterator { // Read at most `chunkSize` into `buf`. Return the number of actual bytes read. size_t next(void* buf); + size_t recordSize() const { return recordSize_; } private: ChunkRecordIterator(