1syntax = "proto3";
2
3package tensorflow.data;
4
5import "tensorflow/core/framework/tensor_shape.proto";
6import "tensorflow/core/framework/types.proto";
7
8// This file contains protocol buffers for working with tf.data Datasets.
9
10// Metadata describing a compressed component of a dataset element.
11message CompressedComponentMetadata {
12  // The dtype of the component tensor.
13  .tensorflow.DataType dtype = 1;
14  // The shape of the component tensor.
15  .tensorflow.TensorShapeProto tensor_shape = 2;
16  // Size of the uncompressed tensor bytes. For tensors serialized as
17  // TensorProtos, this is TensorProto::BytesAllocatedLong(). For raw Tensors,
18  // this is the size of the buffer underlying the Tensor.
19  int64 tensor_size_bytes = 3;
20}
21
22message CompressedElement {
23  // Compressed tensor bytes for all components of the element.
24  bytes data = 1;
25  // Metadata for the components of the element.
26  repeated CompressedComponentMetadata component_metadata = 2;
27}
28