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

detectNCHWSlicePattern

fun <T : DType, V> detectNCHWSlicePattern(slices: List<Slice<T, V>>): NCHWSlicePattern(source)

Detects the NCHW slicing pattern for optimization purposes.

Return

the detected pattern type

Parameters

slices

the list of slice descriptors for a 4D NCHW tensor