additional ResNet sizes
@@ -133,6 +133,19 @@ BenchmarkProblems:
- Exact: [ 3136, 2048, 1, 512 ]
- Exact: [ 3025, 256, 64, 64 ]
- Exact: [ 3025, 64, 64, 64 ]
+ # additional ResNet sizes
+ - Exact: [ 1024, 256, 64, 196 ]
+ - Exact: [ 256, 1024, 64, 196 ]
+ - Exact: [ 512, 128, 64, 784 ]
+ - Exact: [ 2048, 512, 64, 49 ]
+ - Exact: [ 256, 64, 64, 3136 ]
+ - Exact: [ 512, 2048, 64, 49 ]
+ - Exact: [ 128, 512, 64, 784 ]
+ - Exact: [ 128, 256, 64, 3136 ]
+ - Exact: [ 64, 256, 64, 3136 ]
+ - Exact: [ 512, 1024, 64, 196 ]
+ - Exact: [ 256, 512, 64, 784 ]
+ - Exact: [ 64, 64, 64, 3136 ]
########################################
# NT - standard
@@ -250,6 +263,19 @@ BenchmarkProblems:
BenchmarkFinalParameters:
- ProblemSizes:
- Range: [ [64, 64, 64, 7000], [64, 64, 64, 7000], [1], [3104] ]
+ # ResNet sizes
+ - Exact: [ 256, 196, 64, 1024 ]
+ - Exact: [ 1024, 196, 64, 256 ]
+ - Exact: [ 128, 784, 64, 512 ]
+ - Exact: [ 512, 49, 64, 2048 ]
+ - Exact: [ 64, 3136, 64, 256 ]
+ - Exact: [ 2048, 49, 64, 512 ]
+ - Exact: [ 512, 784, 64, 128 ]
+ - Exact: [ 256, 3136, 64, 128 ]
+ - Exact: [ 256, 3136, 64, 64 ]
+ - Exact: [ 1024, 196, 64, 512 ]
+ - Exact: [ 512, 784, 64, 256 ]
+ - Exact: [ 64, 3136, 64, 64 ]
########################################
# TN - VectorWidth Correctness
