Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Data Fields
caffe2::ShapeInfo Struct Reference

Public Types

enum  DimType : int8_t { UNKNOWN = 0, CONSTANT = 1, BATCH = 2, SEQ = 3 }
 

Public Member Functions

 ShapeInfo (DimType t, TensorShape &&s)
 
 ShapeInfo (DimType t, const TensorShape &s)
 

Data Fields

DimType dim_type {DimType::UNKNOWN}
 
TensorShape shape
 

Detailed Description

Definition at line 7 of file shape_info.h.


The documentation for this struct was generated from the following file: