3 #include "caffe2/core/operator.h" 8 enum DimType : int8_t { UNKNOWN = 0, CONSTANT = 1, BATCH = 2, SEQ = 3 };
10 ShapeInfo(DimType t, TensorShape&& s) : dim_type(t), shape(std::move(s)) {}
11 ShapeInfo(DimType t,
const TensorShape& s) : dim_type(t), shape(s) {}
14 DimType dim_type{DimType::UNKNOWN};
18 using ShapeInfoMap = std::unordered_map<std::string, ShapeInfo>;
Blob is a general container that hosts a typed pointer.
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...