skainet-lang-core/sk.ainet.lang.tensor.benchmark

Package-level declarations

Types

data class SlicingBenchmarkConfig(val warmupIterations: Int = 5, val iterations: Int = 20, val tensorShapes: List<IntArray> = listOf( intArrayOf(32, 128, 224, 224), intArrayOf(1024, 768), intArrayOf(8, 8, 512, 512), intArrayOf(100, 1000) ), val accessSamples: Int = 100)

TensorFactory

typealias TensorFactory<T, V> = (ExecutionContext, IntArray) -> Tensor<T, V>

Functions

slicingBenchmarkSuite

fun <T : DType, V> slicingBenchmarkSuite(name: String = "tensor-slicing", config: SlicingBenchmarkConfig = SlicingBenchmarkConfig(), contextFactory: () -> ExecutionContext, tensorFactory: TensorFactory<T, V>): BenchmarkSuite