下载
中文
注册

自定义算子简介

特征检索方案使用TIK算子开发实现特征距离计算逻辑,包含以下的自定义算子。

  1. Flat距离计算算子:得到特征底库数据和待检索的特征向量之间的距离(L2/IP)。
  2. SQ8距离计算算子:得到SQ量化的特征底库数据和待检索的未量化特征向量之间的距离(L2/IP)。
  3. IVFSQ8算子:得到IVFSQ8算法所需要的算子。
  4. INT8Flat距离计算算子:得到INT8量化的特征底库数据和待检索的INT8量化特征向量之间的距离(L2/COS)。
  5. IVFSQT算子:得到IVFSQT算法一二三级所需的距离算子。
  6. FlatAT算子:主要用于在IVF场景,减少train和add的耗时,其中“code_num”等于“nlist”
  7. FlatInt8AT算子:优化在Atlas 推理系列产品下IVFSQT中train、add与update的耗时。
  8. AICPU算子:调度昇腾AI处理器的CPU完成排序等计算,充分利用硬件性能。
  9. BinaryFlat算子:得到二值化算法所需算子。
  10. Mask算子:得到时空库属性过滤算法所需的Mask算子。
  11. IVFSP算子:得到IVFSP算法所需的业务算子、AICPU算子,以及训练生成IVFSP码本时所需的训练算子。