feature/benchmark/com.shifthackz.aisdv1.feature.benchmark/InferenceLikeWorkload/Companion

Companion

private object Companion(source)

Properties

private const val ATTENTION_DIM: Int = 32

ATTENTION_WRITE_WEIGHT

private const val ATTENTION_WRITE_WEIGHT: Float

private const val CHANNELS: Int = 16

CHECKSUM_STRIDE

private const val CHECKSUM_STRIDE: Int = 251

private const val EPSILON: Float = 1.0E-5f

private const val GROUPS: Int = 4

private const val KERNEL_SIZE: Int = 9

private const val LATENT_SIZE: Int = 32

private const val PLANE: Int

RESIDUAL_BLOCKS

private const val RESIDUAL_BLOCKS: Int = 4

RESIDUAL_WEIGHT

private const val RESIDUAL_WEIGHT: Float = 0.22f

private const val TENSOR_SIZE: Int

private const val TOKEN_COUNT: Int = 96