Method on_graph_endpoint_var

src/gopt/impl/tensor_reformat.cpp:190–199

================ EnableTensorCorePass =============== */

/* ================ EnableTensorCorePass =============== */
VarNode* EnableTensorCorePass::on_graph_endpoint_var(
        VarNode* new_var, VarNode* orig_var) const {
    if (!orig_var->shape().eq_shape(new_var->shape())) {
        return RelayoutPlaceholder::make(
                       new_var,
                       ReformatKey{TensorFormats::NCHWc32, TensorFormats::NCHWc4})
                .node();
    }
    return new_var;
}

VarNode* EnableTensorCorePass::trans_to_nchw32(VarNode* new_inp) {
    const TensorShape& shape = new_inp->shape();