Method InferShape

operator/operator/gemm.cpp:29–59

namespace TEngine {

bool Gemm::InferShape(const std::vector<TEngine::TShape>& ishape, std::vector<TEngine::TShape>& oshape, int layout)
{
    int m, n;

    const TShape& input = ishape[0];
    const TShape& weight = ishape[1];

    if(param_.transA)
        m = input.Shape(1);
    else
        m = input.Shape(0);

    if(param_.transB)
        n = weight.Shape(0);
    else
        n = weight.Shape(1);

    TShape out_shape;

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

    dim[0] = m;
    dim[1] = n;

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

    oshape[0] = out_shape;

    return true;
}

void Gemm::SetSchema(void)
{