Commit 8b97d29

authored

Add optimization of instancenorm (#1330)

* Add optimization of instancenorm * override pytorch instancenorm * update code format * make sure output tensor contiguous under AVX2 * update code format

1 parent 7b3e7ea commit 8b97d29Copy full SHA for 8b97d29

4 files changed

+1488

-0

lines changed

csrc/cpu/aten
- InstanceNorm.cpp
- InstanceNorm.h
- kernels
  - InstanceNormKrnl.cpp
tests/cpu
- test_instance_norm.py

4 files changed

+1488

-0

lines changed

Comments

(0)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit 8b97d29

4 files changed

4 files changed

File tree

4 files changed

4 files changed

0 commit comments