@@ -148,7 +148,7 @@ class TensorInfo(google.protobuf.message.Message):
148148 NAME_FIELD_NUMBER : builtins .int
149149 DESCRIPTION_FIELD_NUMBER : builtins .int
150150 DATA_TYPE_FIELD_NUMBER : builtins .int
151- N_DIM_FIELD_NUMBER : builtins .int
151+ SHAPE_FIELD_NUMBER : builtins .int
152152 ASSOCIATED_FILES_FIELD_NUMBER : builtins .int
153153 EXTRA_FIELD_NUMBER : builtins .int
154154 name : builtins .str
@@ -157,8 +157,10 @@ class TensorInfo(google.protobuf.message.Message):
157157 'description of the data in the array/tensor'
158158 data_type : builtins .str
159159 'data type of the array/tensor, e.g. float32, float64, uint8'
160- n_dim : builtins .int
161- 'number of dimensions in the array/tensor'
160+
161+ @property
162+ def shape (self ) -> google .protobuf .internal .containers .RepeatedScalarFieldContainer [builtins .int ]:
163+ """shape of the array/tensor (-1 for unknown)"""
162164
163165 @property
164166 def associated_files (self ) -> google .protobuf .internal .containers .RepeatedCompositeFieldContainer [global___File ]:
@@ -168,13 +170,13 @@ class TensorInfo(google.protobuf.message.Message):
168170 def extra (self ) -> google .protobuf .struct_pb2 .Struct :
169171 """anything else you want to say"""
170172
171- def __init__ (self , * , name : builtins .str = ..., description : builtins .str = ..., data_type : builtins .str = ..., n_dim : builtins .int = ..., associated_files : collections .abc .Iterable [global___File ] | None = ..., extra : google .protobuf .struct_pb2 .Struct | None = ...) -> None :
173+ def __init__ (self , * , name : builtins .str = ..., description : builtins .str = ..., data_type : builtins .str = ..., shape : collections . abc . Iterable [ builtins .int ] | None = ..., associated_files : collections .abc .Iterable [global___File ] | None = ..., extra : google .protobuf .struct_pb2 .Struct | None = ...) -> None :
172174 ...
173175
174176 def HasField (self , field_name : typing_extensions .Literal ['extra' , b'extra' ]) -> builtins .bool :
175177 ...
176178
177- def ClearField (self , field_name : typing_extensions .Literal ['associated_files' , b'associated_files' , 'data_type' , b'data_type' , 'description' , b'description' , 'extra' , b'extra' , 'n_dim ' , b'n_dim ' , 'name ' , b'name ' ]) -> None :
179+ def ClearField (self , field_name : typing_extensions .Literal ['associated_files' , b'associated_files' , 'data_type' , b'data_type' , 'description' , b'description' , 'extra' , b'extra' , 'name ' , b'name ' , 'shape ' , b'shape ' ]) -> None :
178180 ...
179181global___TensorInfo = TensorInfo
180182
0 commit comments