Method Eye

src/opr/impl/tensor_gen.cpp:136–143

MGB_DYN_TYPE_OBJ_FINAL_IMPL(Eye);

Eye::Eye(VarNode* shape, const Param& param, const OperatorNodeConfig& config)
        : Super{shape->owner_graph(), config, "eye", {shape}}, m_param{param} {
    add_input({shape});
    add_output(None)
            ->dtype(DType::from_enum(param.dtype))
            .add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE);
    add_equivalence_component<PODHash<Param>>(&m_param);
}

SymbolVar Eye::make(
        SymbolVar shape, const Param& param, const OperatorNodeConfig& config) {