├── .gitignore ├── AIPUBuilder ├── Optimizer │ ├── __init__.py │ ├── analyzer │ │ ├── __init__.py │ │ ├── cosine.py │ │ └── running_time.py │ ├── config │ │ ├── __init__.py │ │ ├── cfg_fields.py │ │ └── parser.py │ ├── features │ │ ├── __init__.py │ │ ├── autosearch │ │ │ ├── __init__.py │ │ │ └── mixed_precision_naive_search.py │ │ ├── calibration │ │ │ ├── __init__.py │ │ │ ├── calibration.py │ │ │ ├── global_calibration │ │ │ │ ├── __init__.py │ │ │ │ ├── adaquant_zy.py │ │ │ │ ├── adaround.py │ │ │ │ ├── awq_zy.py │ │ │ │ ├── easy_quant.py │ │ │ │ ├── gptq_zy.py │ │ │ │ ├── mvn_correction.py │ │ │ │ ├── smooth_quant_zy.py │ │ │ │ └── svd_based_quant.py │ │ │ └── local_calibration │ │ │ │ ├── __init__.py │ │ │ │ ├── aciq_gauss.py │ │ │ │ ├── aciq_laplace.py │ │ │ │ ├── extrema.py │ │ │ │ ├── in_ir.py │ │ │ │ ├── kld.py │ │ │ │ ├── mean.py │ │ │ │ ├── nstd.py │ │ │ │ ├── percentile.py │ │ │ │ └── weighted_scale_param.py │ │ └── imagetiling │ │ │ ├── __init__.py │ │ │ └── image_tiling.py │ ├── framework │ │ ├── __init__.py │ │ ├── opt_register.py │ │ ├── pycore │ │ │ ├── __init__.py │ │ │ ├── pygraph.py │ │ │ ├── pyir.py │ │ │ ├── pynode.py │ │ │ ├── pytensor.py │ │ │ └── pytype.py │ │ └── qgraph.py │ ├── logger │ │ ├── __init__.py │ │ ├── aipu_logger.py │ │ ├── opt_log_management.py │ │ └── opt_logger.py │ ├── ops │ │ ├── LRN.py │ │ ├── __init__.py │ │ ├── abs.py │ │ ├── accidentalhits.py │ │ ├── acos.py │ │ ├── acosh.py │ │ ├── activation.py │ │ ├── adativepool.py │ │ ├── add.py │ │ ├── affine_grid.py │ │ ├── argminmax.py │ │ ├── asin.py │ │ ├── asinh.py │ │ ├── atan.py │ │ ├── atanh.py │ │ ├── basiclstm.py │ │ ├── batchtodepth.py │ │ ├── batchtospace.py │ │ ├── batchtospaceNd.py │ │ ├── bias_add.py │ │ ├── bitshift.py │ │ ├── bitwise.py │ │ ├── bn.py │ │ ├── bnll.py │ │ ├── boundingbox.py │ │ ├── cacheupdate.py │ │ ├── cast.py │ │ ├── ceil.py │ │ ├── celu.py │ │ ├── channelshuffle.py │ │ ├── clip.py │ │ ├── col2im.py │ │ ├── collapse_repeated.py │ │ ├── compress.py │ │ ├── concat.py │ │ ├── constant.py │ │ ├── constant_of_shape.py │ │ ├── control_op.py │ │ ├── conv.py │ │ ├── conv2d_integer.py │ │ ├── conv3d.py │ │ ├── convtranspose3d.py │ │ ├── convwinograd.py │ │ ├── cosh.py │ │ ├── cosine.py │ │ ├── count.py │ │ ├── crelu.py │ │ ├── crop.py │ │ ├── crop_and_resize.py │ │ ├── ctcgreedydecoder.py │ │ ├── cumulate.py │ │ ├── datastride.py │ │ ├── decodebox.py │ │ ├── deconv.py │ │ ├── depthtospace.py │ │ ├── depthwiseconv.py │ │ ├── dequantize.py │ │ ├── detectionoutput.py │ │ ├── dilation2d.py │ │ ├── div.py │ │ ├── div_mod.py │ │ ├── dummy.py │ │ ├── eltwise.py │ │ ├── elu.py │ │ ├── embedding_lookup_sparse.py │ │ ├── erf.py │ │ ├── erosion2d.py │ │ ├── exp.py │ │ ├── fake_quant_with_minmax_vars.py │ │ ├── fc.py │ │ ├── filter.py │ │ ├── filterbox.py │ │ ├── floor.py │ │ ├── fractionalpool.py │ │ ├── gather.py │ │ ├── gather_elements.py │ │ ├── gather_nd.py │ │ ├── gelu.py │ │ ├── gemm.py │ │ ├── generateproposal.py │ │ ├── get_valid_count.py │ │ ├── globalpooling.py │ │ ├── grid_sample.py │ │ ├── groupnorm.py │ │ ├── gruv1.py │ │ ├── gruv3.py │ │ ├── hardmax.py │ │ ├── hardsigmoid.py │ │ ├── hardswish.py │ │ ├── heatmapMaxkeypoint.py │ │ ├── inp.py │ │ ├── instancenorm.py │ │ ├── interp.py │ │ ├── intopk.py │ │ ├── isinf.py │ │ ├── isnan.py │ │ ├── layernorm.py │ │ ├── leakyrelu.py │ │ ├── log.py │ │ ├── logical.py │ │ ├── logsoftmax.py │ │ ├── lpnormalization.py │ │ ├── matmul.py │ │ ├── matmul_integer.py │ │ ├── maxpooling_withargmax.py │ │ ├── maxroipooling.py │ │ ├── maxunpool.py │ │ ├── meshgrid.py │ │ ├── mish.py │ │ ├── mod.py │ │ ├── moments.py │ │ ├── mul.py │ │ ├── multibox_transform_Loc.py │ │ ├── mvn.py │ │ ├── neg.py │ │ ├── nms.py │ │ ├── nonzero.py │ │ ├── noop.py │ │ ├── normal_moments.py │ │ ├── onehot.py │ │ ├── overlapadd.py │ │ ├── pad.py │ │ ├── permute.py │ │ ├── pooling.py │ │ ├── pooling3D.py │ │ ├── postnms1.py │ │ ├── postnms2.py │ │ ├── pow.py │ │ ├── prelu.py │ │ ├── proposal.py │ │ ├── pyramidroi.py │ │ ├── quantize.py │ │ ├── queryrebatch.py │ │ ├── reciprocal.py │ │ ├── reduce.py │ │ ├── region.py │ │ ├── regionfuse.py │ │ ├── relu.py │ │ ├── relu6.py │ │ ├── relu_family.py │ │ ├── repeat.py │ │ ├── reshape.py │ │ ├── resize.py │ │ ├── reversesequence.py │ │ ├── rgb2yuv.py │ │ ├── rms_norm.py │ │ ├── rnn.py │ │ ├── roialign.py │ │ ├── roipooling.py │ │ ├── rotary_embedding.py │ │ ├── round.py │ │ ├── rsqrt.py │ │ ├── scatter_elements.py │ │ ├── scatter_nd.py │ │ ├── sdp_attetnion.py │ │ ├── segment_reduce.py │ │ ├── select.py │ │ ├── selu.py │ │ ├── shape_operator.py │ │ ├── shrink.py │ │ ├── sigmoid.py │ │ ├── sign.py │ │ ├── silu.py │ │ ├── sine.py │ │ ├── sinh.py │ │ ├── slice_operator.py │ │ ├── slotupdate.py │ │ ├── softmax.py │ │ ├── softplus.py │ │ ├── softsign.py │ │ ├── sort.py │ │ ├── spacetobatch.py │ │ ├── spacetodepth.py │ │ ├── split.py │ │ ├── sqrt.py │ │ ├── square.py │ │ ├── squared_difference.py │ │ ├── squeeze.py │ │ ├── stridedslice.py │ │ ├── sub.py │ │ ├── sufficientStatistics.py │ │ ├── swish.py │ │ ├── tan.py │ │ ├── tanh.py │ │ ├── tf_ops │ │ │ ├── __init__.py │ │ │ └── conv2d.py │ │ ├── thresholdrelu.py │ │ ├── tile.py │ │ ├── topk.py │ │ ├── transpose.py │ │ ├── trunc.py │ │ ├── unidirectional_rnn.py │ │ ├── unique.py │ │ ├── upsamplebyindex.py │ │ ├── where.py │ │ ├── yuv2rgb.py │ │ └── zerofraction.py │ ├── optmaster.py │ ├── passes │ │ ├── __init__.py │ │ ├── absorb_cast_to_clip.py │ │ ├── batch_modifications.py │ │ ├── check_quantization_info_s1.py │ │ ├── control_op_preprocess.py │ │ ├── convert_resize_to_convolution.py │ │ ├── decompose_nonmonotonic_activations_s1.py │ │ ├── detect_inf_mask_nodes.py │ │ ├── eliminate_op.py │ │ ├── global_calibration_prepare.py │ │ ├── insert_op.py │ │ ├── merge_inserted_op.py │ │ ├── merge_matmul_mul_s1.py │ │ ├── merge_scaled_dot_product_attention.py │ │ ├── optimize_x2_wdc.py │ │ ├── passes.py │ │ ├── set_unquantifiable.py │ │ ├── shrink_pow_exponent_s1.py │ │ ├── split_act_perchannel_matmul_s1.py │ │ ├── split_qkv_fc_s1.py │ │ ├── transfer_op_to_reshape_op_s3.py │ │ ├── tune_op_extra_params_s1.py │ │ └── unify_scales_for_multi_inputs_operator.py │ ├── plugins │ │ ├── __init__.py │ │ ├── aipubt_dataset_NumpyZipped.py │ │ ├── aipubt_dataset_OpTestNumpyZipped.py │ │ ├── aipubt_dataset_aishell.py │ │ ├── aipubt_dataset_bevformer.py │ │ ├── aipubt_dataset_bevformer_static.py │ │ ├── aipubt_dataset_cgtdnn.py │ │ ├── aipubt_dataset_coco.py │ │ ├── aipubt_dataset_cocokp.py │ │ ├── aipubt_dataset_fasterrcnnvoc.py │ │ ├── aipubt_dataset_generaldict.py │ │ ├── aipubt_dataset_iwslt.py │ │ ├── aipubt_dataset_librispeech.py │ │ ├── aipubt_dataset_llama2.py │ │ ├── aipubt_dataset_mpii.py │ │ ├── aipubt_dataset_mtcnn.py │ │ ├── aipubt_dataset_nhwcrgb2nhwcbgr.py │ │ ├── aipubt_dataset_numpy.py │ │ ├── aipubt_dataset_numpymultiinput.py │ │ ├── aipubt_dataset_numpymultiinputNCHW.py │ │ ├── aipubt_dataset_numpymultiinputwithoutbatchdim.py │ │ ├── aipubt_dataset_numpynchw2nhwc.py │ │ ├── aipubt_dataset_numpynhwc2nchw.py │ │ ├── aipubt_dataset_numpynhwcrgb2ncbgrhw.py │ │ ├── aipubt_dataset_numpywithdim.py │ │ ├── aipubt_dataset_random.py │ │ ├── aipubt_dataset_sphereface_lfw.py │ │ ├── aipubt_dataset_stable_diffusion_unet.py │ │ ├── aipubt_dataset_tensorfromnumpymultiinput.py │ │ ├── aipubt_dataset_tusimple.py │ │ ├── aipubt_dataset_vocnchw.py │ │ ├── aipubt_dataset_vocnhwc.py │ │ ├── aipubt_dataset_widerface.py │ │ ├── aipubt_metric_CosDistance.py │ │ ├── aipubt_metric_CosDistance_with_seqlen.py │ │ ├── aipubt_metric_Detr_mAP.py │ │ ├── aipubt_metric_EachCosDistance.py │ │ ├── aipubt_metric_FlattenCosDistance.py │ │ ├── aipubt_metric_IWSLT_BLEU.py │ │ ├── aipubt_metric_IWSLT_BLEU_2_gram.py │ │ ├── aipubt_metric_KeywordSpotting.py │ │ ├── aipubt_metric_LMHead.py │ │ ├── aipubt_metric_MaskRcnnCOCOmAP.py │ │ ├── aipubt_metric_MaxAbsError.py │ │ ├── aipubt_metric_MaxAbsError_with_seqlen.py │ │ ├── aipubt_metric_Ocr.py │ │ ├── aipubt_metric_OpTestCosDistance.py │ │ ├── aipubt_metric_RMSE.py │ │ ├── aipubt_metric_SSDmAP.py │ │ ├── aipubt_metric_WER.py │ │ ├── aipubt_metric_YOLOmAP.py │ │ ├── aipubt_metric_bevformer.py │ │ ├── aipubt_metric_centerface.py │ │ ├── aipubt_metric_centernet.py │ │ ├── aipubt_metric_cocokeypoint.py │ │ ├── aipubt_metric_delta1.py │ │ ├── aipubt_metric_f1mesure.py │ │ ├── aipubt_metric_facebox.py │ │ ├── aipubt_metric_fasterrcnnmAP.py │ │ ├── aipubt_metric_fcos_mAP.py │ │ ├── aipubt_metric_imdb.py │ │ ├── aipubt_metric_lightface.py │ │ ├── aipubt_metric_mAP.py │ │ ├── aipubt_metric_mIoU.py │ │ ├── aipubt_metric_mobiledetSSDmAP.py │ │ ├── aipubt_metric_pckh.py │ │ ├── aipubt_metric_poly_lanenet.py │ │ ├── aipubt_metric_psnr.py │ │ ├── aipubt_metric_retinafacebox.py │ │ ├── aipubt_metric_retinanetmAP.py │ │ ├── aipubt_metric_roc.py │ │ ├── aipubt_metric_sphereface.py │ │ ├── aipubt_metric_topk.py │ │ ├── aipubt_metric_widerface.py │ │ ├── aipubt_op_ssd_postprocess.py │ │ ├── aipubt_op_tile.py │ │ ├── aipubt_qconfig_bevformer.py │ │ ├── aipubt_qconfig_minicpm_vpm.py │ │ └── aipubt_qconfig_qwen15_w4a8.py │ ├── qat │ │ ├── __init__.py │ │ ├── qatmain.py │ │ ├── readme.md │ │ ├── readme_CN.md │ │ └── src │ │ │ ├── __init__.py │ │ │ ├── config │ │ │ ├── __init__.py │ │ │ └── config.py │ │ │ ├── fuser │ │ │ ├── __init__.py │ │ │ ├── concat_fuser.py │ │ │ ├── convolution_fuser.py │ │ │ ├── dropout_fuser.py │ │ │ ├── eltwise_fuser.py │ │ │ ├── expand_fuser.py │ │ │ ├── fullyconnected_fuser.py │ │ │ ├── gelu_fuser.py │ │ │ ├── hardsigmoid_fuser.py │ │ │ ├── hardswish_fuser.py │ │ │ ├── layernorm_fuser.py │ │ │ ├── matmul_bn_fuser.py │ │ │ ├── mha_fuser.py │ │ │ ├── multiheadattention_fuser.py │ │ │ ├── pooling_fuser.py │ │ │ ├── reshape_fuser.py │ │ │ └── transpose_fuser.py │ │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── qat_activation.py │ │ │ ├── qat_base_operator.py │ │ │ ├── qat_batchnorm.py │ │ │ ├── qat_concat.py │ │ │ ├── qat_constant.py │ │ │ ├── qat_convolution.py │ │ │ ├── qat_dropout.py │ │ │ ├── qat_eltwise.py │ │ │ ├── qat_expand.py │ │ │ ├── qat_fullyconnected.py │ │ │ ├── qat_gelu.py │ │ │ ├── qat_hardsigmoid.py │ │ │ ├── qat_hardswish.py │ │ │ ├── qat_input.py │ │ │ ├── qat_layernorm.py │ │ │ ├── qat_matmul.py │ │ │ ├── qat_multiheadattention.py │ │ │ ├── qat_pooling.py │ │ │ ├── qat_reshape.py │ │ │ ├── qat_softmax.py │ │ │ ├── qat_split.py │ │ │ └── qat_transpose.py │ │ │ ├── plugin │ │ │ ├── __init__.py │ │ │ ├── aipubt_train_mbv3.py │ │ │ ├── aipubt_train_resnet50.py │ │ │ └── aipubt_train_vitb16.py │ │ │ ├── qatfield.py │ │ │ ├── qatlogger.py │ │ │ ├── qatmaster.py │ │ │ ├── qatregister.py │ │ │ ├── qinfo.py │ │ │ ├── quantizer │ │ │ ├── __init__.py │ │ │ ├── basequantizer.py │ │ │ └── pytorchquantizer.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── cmp.py │ │ │ ├── common_utils.py │ │ │ ├── extra_params.py │ │ │ └── fuser_utils.py │ ├── qtlib_optimize.py │ ├── scripts │ │ ├── git_hooks │ │ │ ├── pre-commit │ │ │ └── pre-commit.d │ │ │ │ ├── format_pyfile │ │ │ │ └── python │ │ └── install_git_hooks.sh │ ├── test │ │ ├── feature_test │ │ │ ├── compare_gt_and_opt_op_forward.py │ │ │ ├── test_betensor_detile.py │ │ │ ├── test_compass_ir_to_torch_module.py │ │ │ └── test_merge_insert_op.py │ │ ├── model_test │ │ │ └── squeezenet │ │ │ │ ├── calibration2.npy │ │ │ │ ├── opt.cfg │ │ │ │ ├── run.sh │ │ │ │ ├── squeezenet_s.bin │ │ │ │ ├── squeezenet_s.txt │ │ │ │ ├── validation10.npy │ │ │ │ └── vlabel10.npy │ │ ├── op_test │ │ │ ├── atan_test.py │ │ │ ├── data.npy │ │ │ ├── ds_reshape_test.py │ │ │ ├── label.npy │ │ │ ├── opt.cfg │ │ │ ├── run.sh │ │ │ ├── single_eltwise_1.bin │ │ │ ├── single_eltwise_1.txt │ │ │ └── softmax_test.py │ │ └── plugin_test │ │ │ ├── aipubt_dataset_my_numpynhwcrgb2ncbgrhw.py │ │ │ ├── aipubt_metric_my_topk.py │ │ │ ├── aipubt_op_my_softmax.py │ │ │ ├── opt.cfg │ │ │ └── run.sh │ ├── tools │ │ ├── __init__.py │ │ ├── generate_plugin_template.py │ │ ├── opt_forward_main.py │ │ ├── optimizer_forward.py │ │ └── optimizer_main.py │ ├── utils │ │ ├── __init__.py │ │ ├── dtype_utils.py │ │ ├── files_utils.py │ │ ├── math_utils.py │ │ ├── passes_utils.py │ │ ├── quant_tool_utils.py │ │ ├── random_utils.py │ │ └── string_utils.py │ └── version.py └── __init__.py ├── LICENSE ├── entry_points_opt.ini ├── images ├── opt_flow.svg └── opt_uml.svg ├── readme.md ├── readme_CN.md └── tutorial.pdf /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/.gitignore -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/analyzer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/analyzer/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/analyzer/cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/analyzer/cosine.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/analyzer/running_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/analyzer/running_time.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/config/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/config/cfg_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/config/cfg_fields.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/config/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/config/parser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/autosearch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/autosearch/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/autosearch/mixed_precision_naive_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/autosearch/mixed_precision_naive_search.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/calibration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/calibration.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/adaquant_zy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/adaquant_zy.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/adaround.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/adaround.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/awq_zy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/awq_zy.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/easy_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/easy_quant.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/gptq_zy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/gptq_zy.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/mvn_correction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/mvn_correction.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/smooth_quant_zy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/smooth_quant_zy.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/global_calibration/svd_based_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/global_calibration/svd_based_quant.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/aciq_gauss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/aciq_gauss.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/aciq_laplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/aciq_laplace.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/extrema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/extrema.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/in_ir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/in_ir.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/kld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/kld.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/mean.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/nstd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/nstd.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/percentile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/percentile.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/calibration/local_calibration/weighted_scale_param.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/calibration/local_calibration/weighted_scale_param.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/imagetiling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/imagetiling/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/features/imagetiling/image_tiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/features/imagetiling/image_tiling.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/opt_register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/opt_register.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/pycore/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/pycore/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/pycore/pygraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/pycore/pygraph.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/pycore/pyir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/pycore/pyir.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/pycore/pynode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/pycore/pynode.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/pycore/pytensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/pycore/pytensor.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/pycore/pytype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/pycore/pytype.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/framework/qgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/framework/qgraph.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/logger/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/logger/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/logger/aipu_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/logger/aipu_logger.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/logger/opt_log_management.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/logger/opt_log_management.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/logger/opt_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/logger/opt_logger.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/LRN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/LRN.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/abs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/abs.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/accidentalhits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/accidentalhits.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/acos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/acos.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/acosh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/acosh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/activation.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/adativepool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/adativepool.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/add.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/affine_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/affine_grid.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/argminmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/argminmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/asin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/asin.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/asinh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/asinh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/atan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/atan.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/atanh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/atanh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/basiclstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/basiclstm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/batchtodepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/batchtodepth.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/batchtospace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/batchtospace.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/batchtospaceNd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/batchtospaceNd.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/bias_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/bias_add.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/bitshift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/bitshift.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/bitwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/bitwise.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/bn.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/bnll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/bnll.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/boundingbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/boundingbox.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/cacheupdate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/cacheupdate.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/cast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/cast.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/ceil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/ceil.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/celu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/celu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/channelshuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/channelshuffle.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/clip.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/col2im.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/col2im.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/collapse_repeated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/collapse_repeated.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/compress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/compress.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/concat.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/constant.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/constant_of_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/constant_of_shape.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/control_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/control_op.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/conv.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/conv2d_integer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/conv2d_integer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/conv3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/conv3d.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/convtranspose3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/convtranspose3d.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/convwinograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/convwinograd.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/cosh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/cosh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/cosine.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/count.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/crelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/crelu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/crop.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/crop_and_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/crop_and_resize.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/ctcgreedydecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/ctcgreedydecoder.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/cumulate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/cumulate.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/datastride.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/datastride.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/decodebox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/decodebox.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/deconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/deconv.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/depthtospace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/depthtospace.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/depthwiseconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/depthwiseconv.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/dequantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/dequantize.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/detectionoutput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/detectionoutput.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/dilation2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/dilation2d.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/div.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/div_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/div_mod.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/dummy.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/eltwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/eltwise.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/elu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/elu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/embedding_lookup_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/embedding_lookup_sparse.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/erf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/erf.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/erosion2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/erosion2d.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/exp.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/fake_quant_with_minmax_vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/fake_quant_with_minmax_vars.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/fc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/filter.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/filterbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/filterbox.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/floor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/floor.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/fractionalpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/fractionalpool.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gather.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gather_elements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gather_elements.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gather_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gather_nd.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gelu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gemm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/generateproposal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/generateproposal.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/get_valid_count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/get_valid_count.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/globalpooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/globalpooling.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/grid_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/grid_sample.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/groupnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/groupnorm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gruv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gruv1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/gruv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/gruv3.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/hardmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/hardmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/hardsigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/hardsigmoid.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/hardswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/hardswish.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/heatmapMaxkeypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/heatmapMaxkeypoint.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/inp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/inp.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/instancenorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/instancenorm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/interp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/interp.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/intopk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/intopk.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/isinf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/isinf.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/isnan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/isnan.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/layernorm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/leakyrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/leakyrelu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/log.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/logical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/logical.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/logsoftmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/logsoftmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/lpnormalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/lpnormalization.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/matmul.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/matmul_integer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/matmul_integer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/maxpooling_withargmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/maxpooling_withargmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/maxroipooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/maxroipooling.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/maxunpool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/maxunpool.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/meshgrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/meshgrid.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/mish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/mish.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/mod.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/moments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/moments.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/mul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/mul.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/multibox_transform_Loc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/multibox_transform_Loc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/mvn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/mvn.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/neg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/neg.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/nms.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/nonzero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/nonzero.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/noop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/noop.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/normal_moments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/normal_moments.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/onehot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/onehot.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/overlapadd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/overlapadd.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/pad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/pad.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/permute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/permute.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/pooling.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/pooling3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/pooling3D.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/postnms1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/postnms1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/postnms2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/postnms2.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/pow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/pow.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/prelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/prelu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/proposal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/proposal.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/pyramidroi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/pyramidroi.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/quantize.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/queryrebatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/queryrebatch.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/reciprocal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/reciprocal.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/reduce.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/region.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/region.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/regionfuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/regionfuse.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/relu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/relu6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/relu6.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/relu_family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/relu_family.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/repeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/repeat.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/reshape.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/resize.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/reversesequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/reversesequence.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/rgb2yuv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/rgb2yuv.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/rms_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/rms_norm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/rnn.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/roialign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/roialign.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/roipooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/roipooling.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/rotary_embedding.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/round.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/round.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/rsqrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/rsqrt.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/scatter_elements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/scatter_elements.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/scatter_nd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/scatter_nd.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sdp_attetnion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sdp_attetnion.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/segment_reduce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/segment_reduce.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/select.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/select.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/selu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/selu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/shape_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/shape_operator.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/shrink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/shrink.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sigmoid.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sign.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/silu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/silu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sine.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sinh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sinh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/slice_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/slice_operator.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/slotupdate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/slotupdate.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/softmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/softplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/softplus.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/softsign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/softsign.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sort.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/spacetobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/spacetobatch.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/spacetodepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/spacetodepth.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/split.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sqrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sqrt.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/square.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/square.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/squared_difference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/squared_difference.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/squeeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/squeeze.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/stridedslice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/stridedslice.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sub.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/sufficientStatistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/sufficientStatistics.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/swish.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/tan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/tan.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/tanh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/tanh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/tf_ops/__init__.py: -------------------------------------------------------------------------------- 1 | # SPDX-License-Identifier: Apache-2.0 2 | # Copyright © 2022-2025 Arm Technology (China) Co. Ltd. 3 | -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/tf_ops/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/tf_ops/conv2d.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/thresholdrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/thresholdrelu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/tile.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/topk.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/transpose.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/trunc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/trunc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/unidirectional_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/unidirectional_rnn.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/unique.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/unique.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/upsamplebyindex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/upsamplebyindex.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/where.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/where.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/yuv2rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/yuv2rgb.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/ops/zerofraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/ops/zerofraction.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/optmaster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/optmaster.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/absorb_cast_to_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/absorb_cast_to_clip.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/batch_modifications.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/batch_modifications.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/check_quantization_info_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/check_quantization_info_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/control_op_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/control_op_preprocess.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/convert_resize_to_convolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/convert_resize_to_convolution.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/decompose_nonmonotonic_activations_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/decompose_nonmonotonic_activations_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/detect_inf_mask_nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/detect_inf_mask_nodes.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/eliminate_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/eliminate_op.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/global_calibration_prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/global_calibration_prepare.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/insert_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/insert_op.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/merge_inserted_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/merge_inserted_op.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/merge_matmul_mul_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/merge_matmul_mul_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/merge_scaled_dot_product_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/merge_scaled_dot_product_attention.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/optimize_x2_wdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/optimize_x2_wdc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/passes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/passes.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/set_unquantifiable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/set_unquantifiable.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/shrink_pow_exponent_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/shrink_pow_exponent_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/split_act_perchannel_matmul_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/split_act_perchannel_matmul_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/split_qkv_fc_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/split_qkv_fc_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/transfer_op_to_reshape_op_s3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/transfer_op_to_reshape_op_s3.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/tune_op_extra_params_s1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/tune_op_extra_params_s1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/passes/unify_scales_for_multi_inputs_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/passes/unify_scales_for_multi_inputs_operator.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_NumpyZipped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_NumpyZipped.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_OpTestNumpyZipped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_OpTestNumpyZipped.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_aishell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_aishell.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_bevformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_bevformer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_bevformer_static.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_bevformer_static.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_cgtdnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_cgtdnn.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_coco.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_cocokp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_cocokp.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_fasterrcnnvoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_fasterrcnnvoc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_generaldict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_generaldict.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_iwslt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_iwslt.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_librispeech.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_librispeech.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_llama2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_llama2.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_mpii.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_mpii.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_mtcnn.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_nhwcrgb2nhwcbgr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_nhwcrgb2nhwcbgr.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpy.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpymultiinput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpymultiinput.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpymultiinputNCHW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpymultiinputNCHW.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpymultiinputwithoutbatchdim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpymultiinputwithoutbatchdim.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpynchw2nhwc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpynchw2nhwc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpynhwc2nchw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpynhwc2nchw.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpynhwcrgb2ncbgrhw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpynhwcrgb2ncbgrhw.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpywithdim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_numpywithdim.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_random.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_sphereface_lfw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_sphereface_lfw.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_stable_diffusion_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_stable_diffusion_unet.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_tensorfromnumpymultiinput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_tensorfromnumpymultiinput.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_tusimple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_tusimple.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_vocnchw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_vocnchw.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_vocnhwc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_vocnhwc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_dataset_widerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_dataset_widerface.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_CosDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_CosDistance.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_CosDistance_with_seqlen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_CosDistance_with_seqlen.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_Detr_mAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_Detr_mAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_EachCosDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_EachCosDistance.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_FlattenCosDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_FlattenCosDistance.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_IWSLT_BLEU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_IWSLT_BLEU.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_IWSLT_BLEU_2_gram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_IWSLT_BLEU_2_gram.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_KeywordSpotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_KeywordSpotting.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_LMHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_LMHead.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_MaskRcnnCOCOmAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_MaskRcnnCOCOmAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_MaxAbsError.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_MaxAbsError.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_MaxAbsError_with_seqlen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_MaxAbsError_with_seqlen.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_Ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_Ocr.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_OpTestCosDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_OpTestCosDistance.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_RMSE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_RMSE.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_SSDmAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_SSDmAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_WER.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_WER.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_YOLOmAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_YOLOmAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_bevformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_bevformer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_centerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_centerface.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_centernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_centernet.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_cocokeypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_cocokeypoint.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_delta1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_delta1.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_f1mesure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_f1mesure.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_facebox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_facebox.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_fasterrcnnmAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_fasterrcnnmAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_fcos_mAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_fcos_mAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_imdb.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_lightface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_lightface.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_mAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_mAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_mIoU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_mIoU.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_mobiledetSSDmAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_mobiledetSSDmAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_pckh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_pckh.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_poly_lanenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_poly_lanenet.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_psnr.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_retinafacebox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_retinafacebox.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_retinanetmAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_retinanetmAP.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_roc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_roc.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_sphereface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_sphereface.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_topk.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_metric_widerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_metric_widerface.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_op_ssd_postprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_op_ssd_postprocess.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_op_tile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_op_tile.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_qconfig_bevformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_qconfig_bevformer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_qconfig_minicpm_vpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_qconfig_minicpm_vpm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/plugins/aipubt_qconfig_qwen15_w4a8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/plugins/aipubt_qconfig_qwen15_w4a8.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/qatmain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/qatmain.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/readme.md -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/readme_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/readme_CN.md -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/config/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/config/config.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/concat_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/concat_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/convolution_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/convolution_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/dropout_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/dropout_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/eltwise_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/eltwise_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/expand_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/expand_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/fullyconnected_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/fullyconnected_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/gelu_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/gelu_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/hardsigmoid_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/hardsigmoid_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/hardswish_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/hardswish_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/layernorm_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/layernorm_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/matmul_bn_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/matmul_bn_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/mha_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/mha_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/multiheadattention_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/multiheadattention_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/pooling_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/pooling_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/reshape_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/reshape_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/fuser/transpose_fuser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/fuser/transpose_fuser.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_activation.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_base_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_base_operator.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_batchnorm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_concat.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_constant.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_convolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_convolution.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_dropout.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_eltwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_eltwise.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_expand.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_fullyconnected.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_fullyconnected.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_gelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_gelu.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_hardsigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_hardsigmoid.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_hardswish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_hardswish.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_input.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_layernorm.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_matmul.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_multiheadattention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_multiheadattention.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_pooling.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_reshape.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_softmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_split.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/ops/qat_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/ops/qat_transpose.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/plugin/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/plugin/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/plugin/aipubt_train_mbv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/plugin/aipubt_train_mbv3.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/plugin/aipubt_train_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/plugin/aipubt_train_resnet50.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/plugin/aipubt_train_vitb16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/plugin/aipubt_train_vitb16.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/qatfield.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/qatfield.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/qatlogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/qatlogger.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/qatmaster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/qatmaster.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/qatregister.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/qatregister.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/qinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/qinfo.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/quantizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/quantizer/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/quantizer/basequantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/quantizer/basequantizer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/quantizer/pytorchquantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/quantizer/pytorchquantizer.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/utils/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/utils/cmp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/utils/cmp.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/utils/common_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/utils/extra_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/utils/extra_params.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qat/src/utils/fuser_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qat/src/utils/fuser_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/qtlib_optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/qtlib_optimize.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/scripts/git_hooks/pre-commit: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/scripts/git_hooks/pre-commit -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/scripts/git_hooks/pre-commit.d/format_pyfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/scripts/git_hooks/pre-commit.d/format_pyfile -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/scripts/git_hooks/pre-commit.d/python: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/scripts/git_hooks/pre-commit.d/python -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/scripts/install_git_hooks.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/scripts/install_git_hooks.sh -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/feature_test/compare_gt_and_opt_op_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/feature_test/compare_gt_and_opt_op_forward.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/feature_test/test_betensor_detile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/feature_test/test_betensor_detile.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/feature_test/test_compass_ir_to_torch_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/feature_test/test_compass_ir_to_torch_module.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/feature_test/test_merge_insert_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/feature_test/test_merge_insert_op.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/calibration2.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/calibration2.npy -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/opt.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/opt.cfg -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/run.sh -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/squeezenet_s.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/squeezenet_s.bin -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/squeezenet_s.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/squeezenet_s.txt -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/validation10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/validation10.npy -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/model_test/squeezenet/vlabel10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/model_test/squeezenet/vlabel10.npy -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/atan_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/atan_test.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/data.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/data.npy -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/ds_reshape_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/ds_reshape_test.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/label.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/label.npy -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/opt.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/opt.cfg -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/run.sh -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/single_eltwise_1.bin: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/single_eltwise_1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/single_eltwise_1.txt -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/op_test/softmax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/op_test/softmax_test.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/plugin_test/aipubt_dataset_my_numpynhwcrgb2ncbgrhw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/plugin_test/aipubt_dataset_my_numpynhwcrgb2ncbgrhw.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/plugin_test/aipubt_metric_my_topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/plugin_test/aipubt_metric_my_topk.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/plugin_test/aipubt_op_my_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/plugin_test/aipubt_op_my_softmax.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/plugin_test/opt.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/plugin_test/opt.cfg -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/test/plugin_test/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/test/plugin_test/run.sh -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/tools/__init__.py: -------------------------------------------------------------------------------- 1 | # SPDX-License-Identifier: Apache-2.0 2 | # Copyright © 2022-2025 Arm Technology (China) Co. Ltd. 3 | -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/tools/generate_plugin_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/tools/generate_plugin_template.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/tools/opt_forward_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/tools/opt_forward_main.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/tools/optimizer_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/tools/optimizer_forward.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/tools/optimizer_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/tools/optimizer_main.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/__init__.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/dtype_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/dtype_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/files_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/files_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/math_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/passes_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/passes_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/quant_tool_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/quant_tool_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/random_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/random_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/utils/string_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/utils/string_utils.py -------------------------------------------------------------------------------- /AIPUBuilder/Optimizer/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/AIPUBuilder/Optimizer/version.py -------------------------------------------------------------------------------- /AIPUBuilder/__init__.py: -------------------------------------------------------------------------------- 1 | # SPDX-License-Identifier: Apache-2.0 2 | # Copyright © 2022-2025 Arm Technology (China) Co. Ltd. 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/LICENSE -------------------------------------------------------------------------------- /entry_points_opt.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/entry_points_opt.ini -------------------------------------------------------------------------------- /images/opt_flow.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/images/opt_flow.svg -------------------------------------------------------------------------------- /images/opt_uml.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/images/opt_uml.svg -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/readme.md -------------------------------------------------------------------------------- /readme_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/readme_CN.md -------------------------------------------------------------------------------- /tutorial.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Arm-China/Compass_Optimizer/HEAD/tutorial.pdf --------------------------------------------------------------------------------