Method InferShape

operator/operator/upsample.cpp:27–44

namespace TEngine {
bool Upsample::InferShape(const std::vector<TEngine::TShape>& ishape, std::vector<TEngine::TShape>& oshape, int layout)
{
    TShape shape;
    ;
    std::vector<int> dim;
    dim = ishape[0].GetDim();
    int scale = param_.scale;
    // NCHW output_h = in_h * scale output_w = in_w * scale;
    dim[2] = dim[2] * scale;
    dim[3] = dim[3] * scale;

    shape.SetDim(dim);
    shape.SetDataLayout(TENGINE_LAYOUT_NCHW);

    oshape[0] = shape;

    return true;
}
void Upsample::SetSchema(void)
{
    Input({"input:float32"}).Output({"output:float32"}).SetAttr("scale", 1).SetDoc(R"DOC(Upsample Operator)DOC");