Method scn_do_execute

src/opr/impl/misc.cpp:51–62

MEGDNN_OPR_INIT1(Argmax, "argmax")

void Argmax::scn_do_execute() {
    if (input(0)->dev_tensor().empty()) {
        mgb_assert(
                input(0)->dev_tensor().layout().shape[param().axis] != 0,
                "Argmax: expected reduction dim %d to be specified for empty input",
                param().axis);
        mgb_assert(output(0)->dev_tensor().empty());
        return;
    }
    mgb_assert(!output(0)->dev_tensor().empty());
    Super::scn_do_execute();
}

MAKE_NODE_PROP_WITH_ZERO_SHAPE_1(Argmax, 0)