caffe2_perfkernels/
lib.rs

1#![feature(test)]
2extern crate test;
3
4#[macro_use] mod imports;
5use imports::*;
6
7x!{perfkernels_adagrad_avx2}
8x!{perfkernels_adagrad}
9x!{perfkernels_common}
10x!{perfkernels_cvtsh_ss_bugfix}
11x!{perfkernels_embedding_lookup_avx2}
12x!{perfkernels_embedding_lookup_fused_8bit_rowwise_avx2}
13x!{perfkernels_embedding_lookup_fused_8bit_rowwise_idx_avx2}
14x!{perfkernels_embedding_lookup_idx_avx2}
15x!{perfkernels_embedding_lookup_idx}
16x!{perfkernels_embedding_lookup}
17x!{perfkernels_fused_8bit_rowwise_embedding_lookup_idx}
18x!{perfkernels_fused_8bit_rowwise_embedding_lookup}
19x!{perfkernels_fused_nbit_rowwise_conversion}
20x!{perfkernels_lstm_unit_cpu_avx2}
21x!{perfkernels_lstm_unit_cpu_common}
22x!{perfkernels_lstm_unit_cpu}
23x!{perfkernels_math_cpu_avx2}
24x!{perfkernels_math_cpu_base}
25x!{perfkernels_math}
26x!{perfkernels_typed_axpy_avx2}
27x!{perfkernels_typed_axpy_avx}
28x!{perfkernels_typed_axpy}