Method InferShape

operator/operator/interp.cpp:28–57

namespace TEngine {

bool Interp::InferShape(const std::vector<TEngine::TShape>& ishape, std::vector<TEngine::TShape>& oshape, int layout)
{

    const TShape& input = ishape[0];
    int in_n = input.GetN();
    int in_c = input.GetC();
    int in_h = input.GetH();
    int in_w = input.GetW();
    

    param_.output_height = in_h * param_.height_scale;
    param_.output_width = in_w * param_.width_scale;


    TShape out_shape;

    std::vector<int> dim(4);

    dim[0] = in_n;
    dim[1] = in_c;
    dim[2] = param_.output_height;
    dim[3] = param_.output_width;

    out_shape.SetDim(dim);
    out_shape.SetDataLayout(input.GetDataLayout());

    oshape[0] = out_shape;

    return true;
}

void Interp::SetSchema(void)
{