EnzymeAD / EnzymeAD/Enzyme-JAX

[All2All] Generalize triple concat

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

Description

```

%16123 = stablehlo.slice %arg31 [0:8, 8:504, 8:248] : (tensor<528x512x256xf64>) -> tensor<8x496x240xf64> loc(#loc2609)
%16118 = stablehlo.negate %16117 : tensor<512x496x240xf64> loc(#loc1468)
%16124 = stablehlo.slice %arg31 [520:528, 8:504, 8:248] : (tensor<528x512x256xf64>) -> tensor<8x496x240xf64> loc(#loc2609)

%16125 = stablehlo.concatenate %16123, %16118, %16124, dim = 0 : (tensor<8x496x240xf64>, tensor<512x496x240xf64>, tensor<8x496x240xf64>) -> tensor<528x496x240xf64> loc(#loc2609)
```

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.