pytorch/pytorch/third_party/fbgemm/include/fbgemm/ Source Tree

../
ConvUtils.h conv_param_t
Fbgemm.h DoNothing DoSConvOnInpBuffer DoSpmdmOnInpBuffer PackAMatrix PackAWithIm2Col PackAWithQuantRowOffset PackAWithRowOffset PackBMatrix PackMatrix PackWeightMatrixForGConv PackWeightsForConv ReQuantizeForFloat ReQuantizeOutput ReluOutput ScaleOP memCopy
FbgemmBuild.h
FbgemmConvert.h
FbgemmEmbedding.h EmbeddingSpMDMKernelSignature EmbeddingSpMDMRowWiseSparseKernelSignature RowWiseSparseAdaGradFusedSignature SparseAdaGradSignature
FbgemmFP16.h TypeConverter
FbgemmFPCommon.h GemmParams
FbgemmI64.h
FbgemmI8DepthwiseAvx2.h PackedDepthWiseConvMatrix
FbgemmI8DirectconvAvx2.h PackedDirectConvMatrix
FbgemmI8Spmdm.h CompressedSparseColumn
FbgemmPackMatrixB.h PackedGemmMatrixB TypeConverter
FbgemmSparse.h BCSRMatrix CSRMatrix
OutputProcessing-inl.h
PackingTraits-inl.h PackingTraits is_16or32bit
QuantUtils.h
QuantUtilsAvx2.h RequantizationParams TensorQuantizationParams
QuantUtilsAvx512.h
SimdUtils.h simd_info
Types.h
Utils.h BlockingFactors impl_type_t inst_set_t is_8bit layout_t matrix_op_t optimized_conv_t thread_type_t
UtilsAvx2.h QuantizationGranularity block_type_t requantizationForFloatParams_t requantizationParams_t
spmmUtils.h
spmmUtilsAvx2.h trRequantizationParams_t

Browse the source code of pytorch/pytorch/third_party/fbgemm/include/fbgemm/ online