EnzymeAD / EnzymeAD/Enzyme-JAX

BatchNorm Inference fusion with Dense or Conv

Open
#533 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
MLIR
Stars
131
Forks
53
Avg merge
1d 10h
Merged PRs (30d)
193

Description

See https://github.com/pytorch/pytorch/blob/main/torch/nn/utils/fusion.py

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.