pytorch/caffe2/perfkernels
2023-06-22 18:30:31 +00:00
..
__init__.py
adagrad_avx2.cc
adagrad.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
adagrad.h
batch_box_cox_avx2.cc
batch_box_cox.cc
batch_box_cox.h
CMakeLists.txt
common_avx.cc
common_avx2.cc
common_avx512.cc
common.h
cvtsh_ss_bugfix.h
embedding_lookup_avx2.cc
embedding_lookup_fused_8bit_rowwise_avx2.cc
embedding_lookup_fused_8bit_rowwise_idx_avx2.cc
embedding_lookup_idx_avx2.cc
embedding_lookup_idx.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
embedding_lookup_idx.h
embedding_lookup.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
embedding_lookup.h
fused_8bit_rowwise_embedding_lookup_idx.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
fused_8bit_rowwise_embedding_lookup_idx.h
fused_8bit_rowwise_embedding_lookup.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
fused_8bit_rowwise_embedding_lookup.h
fused_nbit_rowwise_conversion.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
fused_nbit_rowwise_conversion.h
hp_emblookup_codegen.py
lstm_unit_cpu_avx2.cc
lstm_unit_cpu_common.cc
lstm_unit_cpu_common.h
lstm_unit_cpu-impl.h
lstm_unit_cpu.h
math_cpu_avx2.cc
math_cpu_base.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
math.h
typed_axpy_avx.cc
typed_axpy_avx2.cc
typed_axpy.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +00:00
typed_axpy.h
vectorizer.h