tests/cpuTapeLayers/singleheadbench library
Functions
-
generateMatrix(
int rows, int cols) → Tensor< Matrix> -
main(
) → void -
pseudoLoss<
T> (Tensor< T> out) → Tensor<Scalar> -
runBenchmark<
I, O> (String name, Layer< I, O> layer, Tensor<I> input, int iterations) → void