OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add TensorShapeAndType (#29848)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29848

design doc: https://docs.google.com/document/d/15luH8R7a0WMiZzoKxu6cI0a1XDW4C0vyaW3-XQ_3G30/edit#heading=h.cyvbc4wtxkn7

Test Plan: buck build

Reviewed By: ipiszy

Differential Revision: D18513718

fbshipit-source-id: c3e3b30b58360b898528422ba9618b1dd3beb0a8
This commit is contained in:
Chunli Fu 2019-11-15 12:56:51 -08:00 committed by Facebook Github Bot
parent 5ab6635de1
commit 7807d44934
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
caffe2/proto/caffe2.proto
View File

@ -158,6 +158,40 @@ message TensorShapes {
repeated TensorShape shapes = 1;
}
// TensorBoundShape is used to save bound shape inference result for a tensor.
// TensorBoundShape.shape is inferred shape for this tensor.
// TensorBoundShape.dimType contains dim_type for every dimension.
// eg: for dimension i, shape.dims[i] is the inferred shape and
// dim_type[i] is corresponding dim_type.
message TensorBoundShape {
optional TensorShape shape = 1;
enum DimType {
UNKNOWN = 0; // unknown
CONSTANT = 1; // constant
// batch, corresponding dimension is batch_size
BATCH = 2;
// batch_of_feature_max,
// corresponding shape is inferred_feature_length * batch_size
BATCH_OF_FEATURE_MAX = 3;
// batch_of_feature_max_default
// corresponding shape is default_feature_length * batch_size
BATCH_OF_FEATURE_MAX_DEFAULT = 4;
// feature_max, corresponding shape is inferred_feature_length
FEATURE_MAX = 5;
// feature_max_default, corresponding shape is default_feature_length
FEATURE_MAX_DEFAULT = 6;
}
repeated DimType dim_type = 2; // dim_type.size() == shape.dims.size()
optional string name = 3;
}
message TensorBoundShapes {
repeated TensorBoundShape shapes = 1;
optional int64 max_batch_size = 2;
optional int64 max_feature_len = 3;
}
// A named argument containing either singular float, integer and string
// values, or repeated float, int and string arrays.
message Argument {