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