ONE - On-device Neural Engine
|
#include <TFShapeInferenceHelper.h>
Public Member Functions | |
loco::Padding2D | operator() (const PlaneShape &in) |
Public Member Functions inherited from moco::PlaneInference | |
PlaneShape | operator() (const PlaneShape &in) |
void | padding (const TFPadding &value) |
void | window (const loco::Window< 2 > value) |
void | stride (const loco::Stride< 2 > value) |
Additional Inherited Members | |
Protected Member Functions inherited from moco::PlaneInference | |
void | fill (Parameters &p, const PlaneShape &in) |
PlaneShape | infer (const Parameters &p, const PlaneShape &) |
Definition at line 177 of file TFShapeInferenceHelper.h.
|
inline |
Definition at line 180 of file TFShapeInferenceHelper.h.
References loco::Padding2D::bottom(), moco::PlaneInference::Parameters::effective_window, moco::PlaneInference::fill(), moco::PlaneShape::height, moco::PlaneInference::Parameters::input, loco::Padding2D::left(), loco::Padding2D::right(), moco::PlaneInference::Parameters::stride, loco::Padding2D::top(), loco::Dimension::value(), and moco::PlaneShape::width.