• Diego Caballero's avatar
    [MLIR] Add element-wise reduction ops to nGraph dialect (#37) · d34cd05a
    Diego Caballero authored
    This patch introduces tablegen definitions for element-wise reduction
    operations (only definitions, not building/lowering code). This includes
    argmin, argmax, min, max, sum, prod, all and any.
    
    From the design point of view, a single base class seems to be enough to
    cover all the common ground of these operations. A different base class will
    be necessary for avg pool reduction operations.
    d34cd05a
ops.cpp 5.38 KB