Function init_shape

dnn/src/cambricon/softmax/opr_impl.cpp:32–41

}

SmallVector<size_t> init_shape(_megdnn_tensor_in src, const int param_axis) {
    auto dims = src.layout.shape;
    const int rank = src.layout.ndim;
    const int axis = CanonicalAxis(param_axis, rank);
    const size_t dim = dims[axis];
    const size_t N = SizeToAxis(axis, dims);
    const size_t D = SizeOutAxis(axis, dims, rank);

    return {N, dim, D};
}
}  // anonymous namespace

void SoftmaxForwardImpl::exec(