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: