harmony 鸿蒙NodeAttrTypes.idl

2023-06-24
浏览 (634)

NodeAttrTypes.idl

概述

该文件定义AI模型算子的参数和功能。

该文文件中所有的结构体仅声明了算子的属性，并不包含执行算子函数的接口，具体介绍如下：

该文件中每一个算子都与NodeType的枚举值一一对应，执行模型推理时，NodeType会在Node的nodeType中存储。
每一个算子都至少有一个“输入”与“输出”，“输入”即为该算子接收的张量，“输出”为经过算子运算之后得到的“张量”；“输入”、“算子”和“输出”之间的关系需要通过Node结构体的inputIndex和outIndex来确认。

Since:

3.2

Version:

1.0

相关模块:

NNRt

汇总

类

名称	描述
Activation	struct 激活类型的算子，所有的激活函数都属于该算子，具体的激活函数类型一句参数来确定。
AddFusion	struct 输入Tensor逐元素相加， 输出x和y的和，数据形状与输入broadcast之后一样，数据类型与较高精度的输入精度一致。
ArgMaxFusion	struct 返回跨轴的tensor前K个索引或者是数值。
AvgPoolFusion	struct 在输入tensor上应用 2D 平均池化。支持int8量化输入。
BatchToSpaceND	struct 将一个4维tensor的batch维度按block_shape切分成小块，并将这些小块拼接到空间维度。
BiasAdd	struct 对给出的输入张量上的各个维度方向上的数据进行偏置。
Cast	struct 根据输出张量的类型对张量数据类型进行转换。
Concat	struct 在指定轴上连接张量，将输入张量按给定的轴连接起来。
Conv2DFusion	struct 对将4维的tensor执行带有偏置的二维卷积运算。
Conv2dTransposeFusion	struct 对一个4维的tensor执行带有偏置的二维反卷积。
DivFusion	struct 将两个tensor执行除法运算。
Eltwise	struct 元素级别操作的算子。
ExpandDims	struct 在给定轴上为tensor添加一个额外的维度。
Fill	struct 根据指定的维度，创建由一个标量填充的tensor。
FullConnection	struct 对输入数据做全连接。
FusedBatchNorm	struct 对一个tensor进行批标准化的运算。
Gather	struct 根据指定的索引和轴返回输入tensor的切片。
LayerNormFusion	struct 对一个tensor从某一axis开始做层归一化。
LessEqual	struct 对输入x1和x2，计算每对元素的x1<=x2的结果。
MatMulFusion	struct 对输入x1和x2，计算x1和x2的内积。
Maximum	struct 对输入x1和x2，计算x1和x2对应元素最大值，x1和x2的输入遵守隐式类型转换规则，使数据类型一致。
MaxPoolFusion	struct 对输入x，计算 2D 最大值池化。
MulFusion	struct 对输入x1和x2，将x1和x2相同的位置的元素相乘得到output。
OneHot	struct 根据indices指定的位置，生成一个由one-hot向量构成的tensor。
PadFusion	struct 在x指定维度的数据前后，添加指定数值进行增广。
PowFusion	struct 求x的y次幂，输入必须是两个tensor或一个tensor和一个标量。
PReLUFusion	struct 计算x和weight的PReLU激活值。
QuantDTypeCast	struct 数据类型转换。
ReduceFusion	struct 减小x张量的维度。
Reshape	struct 根据inputShape调整input的形状。
Resize	struct 按给定的参数对输入的张量进行变形。
Rsqrt	struct 求x的平方根的倒数。
ScaleFusion	struct 给定一个tensor，计算其缩放后的值。
Shape	struct 输出输入tensor的形状。
SliceFusion	struct 在x各维度，在axes维度中，以begin为起点，截取size长度的切片。
Softmax	struct 给定一个tensor，计算其softmax结果。
SpaceToBatchND	struct 将4维张量在空间维度上进行切分成多个小块，然后在batch维度上拼接这些小块。
Split	struct 算子沿 axis 维度将x拆分成多个张量，张量数量由outputNum指定。
Sqrt	struct 给定一个tensor，计算其平方根。
SquaredDifference	struct 计算两个输入的差值并返回差值的平方。SquaredDifference算子支持tensor和tensor相减。
Squeeze	struct 去除axis中，长度为1的维度。支持int8量化输入。
Stack	struct 将一组tensor沿axis维度进行堆叠，堆叠前每个tensor的维数为n，则堆叠后output维数为n+1。
StridedSlice	struct 根据步长和索引对输入张量进行切片提取。
SubFusion	struct 计算两个输入的差值。
TileFusion	struct 以multiples指定的次数拷贝输入张量。
TopKFusion	struct 查找沿axis轴的前K个最大值和对应索引。
Transpose	struct 根据perm对x进行数据重排。
Unsqueeze	struct 根据输入axis的值。增加一个维度。