Method exec

dnn/src/naive/softmax/opr_impl.cpp:43–75

}

void SoftmaxForwardImpl::exec(
        _megdnn_tensor_in src, _megdnn_tensor_out dst, _megdnn_workspace workspace) {
#if !MGE_BUILD_WITHOUT_NAIVE_EXEC
    check_exec(src.layout, dst.layout, workspace.size);

    WorkspaceBundle workspace_bundle{
            workspace.raw_ptr, {src.layout.span().dist_byte(), reduce_worksize}};

    TensorLayout tmp_layout;
    reduce_opr->param().mode = Reduce::Mode::MAX;
    reduce_opr->deduce_layout(src.layout, tmp_layout);
    TensorND max_tensor{workspace_bundle.get_workspace(0).raw_ptr, tmp_layout};
    reduce_opr->exec(src, max_tensor, workspace_bundle.get_workspace(1));

    elemwise_opr->param().mode = Elemwise::Mode::SUB;
    elemwise_opr->exec({src, max_tensor}, dst);

    // no broadcast
    elemwise_opr->param().mode = Elemwise::Mode::EXP;
    elemwise_opr->exec({dst}, dst);

    reduce_opr->param().mode = Reduce::Mode::SUM;
    reduce_opr->deduce_layout(src.layout, tmp_layout);

    TensorND deno_tensor{workspace_bundle.get_workspace(0).raw_ptr, tmp_layout};
    reduce_opr->exec(dst, deno_tensor, workspace_bundle.get_workspace(1));

    elemwise_opr->param().mode = Elemwise::Mode::TRUE_DIV;
    elemwise_opr->exec({dst, deno_tensor}, dst);
#else
    __builtin_trap();
#endif
}

//=============================Softmax backward ============================