• Charlie Lin's avatar
    Select module op (#1569) · a63ee2e0
    Charlie Lin authored
    Creates the select_module operator that selects one of the submodules passed to it to run based on the submodule parameters.  The submodule is selected by having the exact same static shapes for the arguments to select_module as the parameters in the submodule
    a63ee2e0
test_select_module_reduce.cpp 3.14 KB