Method NonZero

src/opr/impl/misc.cpp:391–400

MGB_DYN_TYPE_OBJ_FINAL_IMPL(NonZero);

NonZero::NonZero(VarNode* data, const Param& param, const OperatorNodeConfig& config)
        : Super(data->owner_graph(), config, "NonZero", {data}) {
    init_megdnn_opr(*this, param);
    add_input({data});
    auto dtype = megdnn_opr()->infer_type(data->dtype());
    output(0)
            ->add_flag(VarNode::Flag::NO_SYS_MEM_ALLOC)
            .add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE)
            .dtype(dtype);
}

NonZero::NodeProp* NonZero::do_make_node_prop() const {
    auto ret = Super::do_make_node_prop();