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:

• caffe2/opt/bound_shape_inferencer.h
• caffe2/opt/bound_shape_inferencer.cc