pytorch/pytorch/third_party/XNNPACK/src/xnnpack/ Source Tree

../
aarch32-assembler.h AddressingMode Assembler Condition ConsecutiveRegisterList CoreRegister CoreRegisterList DRegister DRegisterLane DataSize MacroAssembler MemOperand MemOperandHelper QRegister QRegisterList SRegister SpecialFPRegister VLoadStoreRegList
aarch64-assembler.h AddressingMode AlignInstruction Assembler BranchType Condition DRegister MacroAssembler MemOperand MemOperandHelper PrefetchOp QRegister SRegister ScalarVRegister ScalarVRegisterList VRegister VRegisterLane VRegisterList XRegister
allocator.h
argmaxpool.h
assembler.h AssemblerBase Error Label
avgpool.h
cache.h xnn_cache xnn_cache_bucket xnn_cache_state xnn_cache_type xnn_caches xnn_code_cache xnn_weights_cache
common.h
compute.h argmax_pooling_context average_pooling_context channel_shuffle_context compute_parameters conv2d_context dwconv2d_context dwconv_context elementwise_binary_context floating_point_softmax_context gemm_context global_average_pooling_ncw_context global_average_pooling_nwc_context igemm_context lut_contiguous_context lut_strided_context max_pooling_context pad_context pixelwise_average_pooling_context prelu_context resize_bilinear_chw_context resize_bilinear_context slice_context spmm_context subconv_context subgemm_context transpose_context u8_softmax_context univector_contiguous_context univector_strided_context unpooling_context vmulcaddc_context xnn_parallelization_type
config.h xnn_binary_elementwise_config xnn_binary_elementwise_subconfig xnn_hardware_config xnn_transpose_config xnn_transpose_subconfig xnn_x8_lut_config
conv.h
dwconv.h
fft.h
fill.h
filterbank.h
gavgpool.h
gemm.h
ibilinear.h
igemm.h
indirection.h
intrinsics-polyfill.h
log.h
lut.h
math-stubs.h
math.h
maxpool.h
memory-planner.h xnn_value_allocation_tracker xnn_value_usage
memory.h xnn_code_buffer xnn_weights_buffer
microfnptr.h
microkernel-type.h xnn_microkernel_type
microkernel-utils.h
microparams-init.h
microparams.h jit_gemm_params xnn_bf16_minmax_params xnn_f16_abs_params xnn_f16_chw_params xnn_f16_default_params xnn_f16_elu_params xnn_f16_expminus_params xnn_f16_f32_cvt_params xnn_f16_gavgpool_params xnn_f16_hswish_params xnn_f16_lrelu_params xnn_f16_minmax_params xnn_f16_neg_params xnn_f16_rnd_params xnn_f16_scaleminmax_params xnn_f16_sigmoid_params xnn_f16_sqrt_params xnn_f32_abs_params xnn_f32_chw_params xnn_f32_default_params xnn_f32_elu_params xnn_f32_expminus_params xnn_f32_f16_cvt_params xnn_f32_gavgpool_params xnn_f32_hswish_params xnn_f32_lrelu_params xnn_f32_minmax_params xnn_f32_neg_params xnn_f32_qs8_cvt_params xnn_f32_qu8_cvt_params xnn_f32_relu_params xnn_f32_rnd_params xnn_f32_scaleminmax_params xnn_f32_sigmoid_params xnn_f32_sqrt_params xnn_qc8_conv_minmax_params xnn_qs8_add_minmax_params xnn_qs8_avgpool_minmax_params xnn_qs8_conv_minmax_params xnn_qs8_cvt_params xnn_qs8_f32_cvt_params xnn_qs8_lrelu_params xnn_qs8_mul_minmax_params xnn_qu8_add_minmax_params xnn_qu8_avgpool_minmax_params xnn_qu8_conv_minmax_params xnn_qu8_cvt_params xnn_qu8_f32_cvt_params xnn_qu8_lrelu_params xnn_qu8_mul_minmax_params xnn_s8_minmax_params xnn_u8_minmax_params xnn_x16_transpose_params xnn_x24_transpose_params xnn_x32_transpose_params xnn_x64_transpose_params xnn_x8_transpose_params
mutex.h xnn_mutex
node-type.h xnn_node_type
normalization.h
operator-type.h xnn_operator_type
operator-utils.h
operator.h subconvolution_params xnn_operator xnn_run_state xnn_ukernel xnn_ukernel_conv2d xnn_ukernel_dwconv xnn_ukernel_dwconv2d xnn_ukernel_gemm xnn_ukernel_igemm xnn_ukernel_spmm xnn_ukernel_vbinary xnn_ukernel_vmulcaddc xnn_ukernel_vunary
pack.h xnn_qs8_packing_params xnn_qu8_packing_params xnn_spmm_packing_params
packx.h
pad.h
params.h argmaxpool_parameters avgpool_parameters conv_hwc2chw_parameters dwconv2d_chw_parameters dwconv_fused_ukernels dwconv_parameters fill_parameters gavgpool_cw_parameters gavgpool_parameters gemm_fused_ukernels gemm_parameters ibilinear_chw_parameters ibilinear_parameters maxpool_parameters pad_parameters pavgpool_parameters prelu_parameters raddstoreexpminusmax_parameters spmm_parameters vmulcaddc_parameters vunary_parameters xnn_hmp_gemm_ukernel xnn_hmp_igemm_ukernel xnn_parameters zip_parameters
pavgpool.h
post-operation.h xnn_post_operation xnn_post_operation_type
ppmm.h
prelu.h
raddexpminusmax.h
raddextexp.h
raddstoreexpminusmax.h
requantization-stubs.h
requantization.h
rmax.h
rmaxabs.h
spmm.h
subgraph-validation.h
subgraph.h xnn_allocation_type xnn_blob xnn_compute_type xnn_layout_type xnn_node xnn_operator_data xnn_runtime xnn_shape xnn_subgraph xnn_value xnn_value_type xnn_workspace
transpose.h
unaligned.h
unpool.h
vadd.h
vbinary.h
vcvt.h
vlog.h
vlrelu.h
vlshift.h
vmul.h
vmulcaddc.h
vscaleexpminusmax.h
vscaleextexp.h
vsquareabs.h
vunary.h
window.h
zip.h

Browse the source code of pytorch/pytorch/third_party/XNNPACK/src/xnnpack/ online