Skip to content

dot (outer product) + reduce fusion operation #27

@rjpower

Description

@rjpower

This depends on #18; we should have an optimization which fuses dot product with a reduction; this will allow us to write applications like k-means in a natural way:

closest = argmin(dot(pts, clusters) - sum(clusters**2, axis=1))

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions