Commit 0aa9ba3

committed

Merge pull request #538 from SKaiNET-developers/fix/535-flatten-shape-inference

Track per-sample shape through DSL spatial layers (#535)

2 parents e6935c3 + 63c4dbf commit 0aa9ba3Copy full SHA for 0aa9ba3

4 files changed

skainet-lang
- skainet-lang-core/src
  - commonMain/kotlin/sk/ainet/lang
    - nn/dsl
      - NetworkBuilder.kt
    - tensor/ops
      - ConvShapeUtils.kt
  - commonTest/kotlin/sk/ainet/lang/nn/dsl
    - CnnShapeInferenceTest.kt
- skainet-lang-models/src/commonMain/kotlin/sk/ainet/lang/model/dnn/cnn
  - MNIST.kt

Comments

(0)