Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::BoundShapeInferencer Class Reference

Public Member Functions

 BoundShapeInferencer (const BoundShapeSpec &spec)
void InferBoundShapeAndType (const NetDef &net, const std::unordered_map< std::string, ShapeInfo > &info)
const ShapeInfoMap & shape_info () const
std::string PrintShapeInfo () const
 Print out all the shape info.

Detailed Description

Definition at line 30 of file bound_shape_inferencer.h.

The documentation for this class was generated from the following files: