Skip to content

Commit

Permalink
Remove extra input layer export tests with dsl builder
Browse files Browse the repository at this point in the history
  • Loading branch information
juliabeliaeva committed Apr 24, 2022
1 parent 5244383 commit 7d1d372
Showing 1 changed file with 0 additions and 66 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,6 @@ package org.jetbrains.kotlinx.dl.api.inference.keras

import org.jetbrains.kotlinx.dl.api.core.Functional
import org.jetbrains.kotlinx.dl.api.core.Sequential
import org.jetbrains.kotlinx.dl.api.core.dsl.functional
import org.jetbrains.kotlinx.dl.api.core.dsl.sequential
import org.jetbrains.kotlinx.dl.api.core.layer.core.Input
import org.junit.jupiter.api.Test

Expand All @@ -28,68 +26,4 @@ class InputLayerImportExportTest {
LayerImportExportTest.run(Functional.of(Input(128, 128, 3)))
LayerImportExportTest.run(Functional.of(Input(10, 10, 10, 10)))
}

@Test
fun dslBuilderSequential() {
LayerImportExportTest.run(
sequential {
layers {
+Input(4)
}
}
)
LayerImportExportTest.run(
sequential {
layers {
+Input(128, 128)
}
}
)
LayerImportExportTest.run(
sequential {
layers {
+Input(128, 128, 3)
}
}
)
LayerImportExportTest.run(
sequential {
layers {
+Input(10, 10, 10, 10)
}
}
)
}

@Test
fun dslBuilderFunctional() {
LayerImportExportTest.run(
functional {
layers {
+Input(4)
}
}
)
LayerImportExportTest.run(
functional {
layers {
+Input(128, 128)
}
}
)
LayerImportExportTest.run(
functional {
layers {
+Input(128, 128, 3)
}
}
)
LayerImportExportTest.run(
functional {
layers {
+Input(10, 10, 10, 10)
}
}
)
}
}

0 comments on commit 7d1d372

Please sign in to comment.