文档
注册

TBE算子UT测试用例定义文件参数解释

表1 TBE算子UT测试用例定义json文件

参数

说明

Operator[xx]

-

可选。

-

OpName

算子的名称。

-

TestName

必选。测试用例的名称。

-

SocVersion

必选。支持的昇腾AI处理器类型。

支持输入“all”,表示支持所有的昇腾AI处理器

-

FunctionType

必选。当前支持的函数类型,支持的参数如下:

  • add_precision_case:为当前算子添加一个精度测试用例。
  • add_case:为当前算子添加一个算子编译的测试,测试算子是否可以支持该规格,编译出.o文件。

Input[xx]

-

必选。

算子的输入。

-

Name

可选。算子输入的名称,string类型。

-

Format

必选。

String类型。

输入tensor数据的排布格式,不允许为空。

常见的数据排布格式如下:

  • NCHW
  • NHWC
  • ND:表示支持任意格式。
  • NC1HWC0:华为自研的5维数据格式。其中,C0与微架构强相关,该值等于cube单元的size,例如16;C1是将C维度按照C0切分:C1=C/C0, 若结果不整除,最后一份数据需要padding到C0。
  • FRACTAL_Z:卷积的权重的格式。
  • FRACTAL_NZ:华为自研的分形格式,在cube单元计算时,输出矩阵的数据格式为NW1H1H0W0。整个矩阵被分为(H1*W1)个分形,按照column major排布,形状如N字形;每个分形内部有(H0*W0)个元素,按照row major排布,形状如z字形。考虑到数据排布格式,将NW1H1H0W0数据格式称为Nz格式。其中,H0,W0表示一个分形的大小,示意图如下所示:

-

OriginFormat

可选。Tensor的原始format。

不带此字段时,默认Tensor的实现format与原始format一致。

-

Type

必选。

String类型。

所有输入输出tensor需支持相同数量的数据类型。

输入数据支持的数据类型。

  • bool
  • int8
  • uint8
  • int16
  • uint16
  • int32
  • int64
  • uint32
  • uint64
  • float16
  • float

-

Shape

必选。

int类型,一维数组。

输入tensor支持的形状,所有输入输出tensor需支持相同数量的形状。

例如:

[8, 3, 256, 256]。

若输入非法的形状会报错,例如:[0]。

需要注意,配置的shape需要和format相匹配。

-

OriginShape

可选。Tensor的原始shape。

不带此字段时,默认Tensor的实现shape与原始shape一致。

Output[xx]

-

必选。

算子输出。

-

Name

算子输入的名称,string类型。

-

Format

必选。

String类型。

输出tensor数据的排布格式,不允许为空。

常见的数据排布格式如下:

  • NCHW
  • NHWC
  • ND:表示支持任意格式。
  • NC1HWC0:华为自研的5维数据格式。其中,C0与微架构强相关,该值等于cube单元的size,例如16;C1是将C维度按照C0切分:C1=C/C0, 若结果不整除,最后一份数据需要padding到C0。
  • FRACTAL_Z:卷积的权重的格式。
  • FRACTAL_NZ:华为自研的分形格式,在cube单元计算时,输出矩阵的数据格式为NW1H1H0W0。整个矩阵被分为(H1*W1)个分形,按照column major排布,形状如N字形;每个分形内部有(H0*W0)个元素,按照row major排布,形状如z字形。考虑到数据排布格式,将NW1H1H0W0数据格式称为Nz格式。其中,H0,W0表示一个分形的大小,示意图如下所示:

-

OriginFormat

可选。Tensor的原始format。

不带此字段时,默认Tensor的实现format与原始format一致。

-

Type

必选。

String类型。

输出数据支持的数据类型:

  • bool
  • int8
  • uint8
  • int16
  • uint16
  • int32
  • int64
  • uint32
  • uint64
  • float16
  • float

-

Shape

必选。

int类型,一维或者二维数组。

输入tensor支持的形状。

例如:

[8, 3, 256, 256]。

若输入非法的形状会报错,例如:[0]。

-

OriginShape

可选。Tensor的原始shape。

不带此字段时,默认Tensor的实现shape与原始shape一致。

Attribute

-

可选。

-

Name

可选。string类型。

-

Type

若配置attr,则为必选属性支持的类型。

string类型。

输出数据支持的数据类型:

  • bool
  • int
  • float
  • string
  • list_bool
  • list_int
  • list_float
  • list_string
  • list_list_int

-

Value

若配置attr,则为必选,且不允许为null。

string类型。

属性值,根据type的不同,属性值不同。

  • bool: true/false
  • int: 10
  • float: 1.0
  • string: “NCHW”
  • list_bool: [false, true]
  • list_int: [1, 224, 224, 3]
  • list_float: [1.0, 0.0]
  • list_string: [“str1”, “str2”]
  • list_list_int: [[1, 3, 5, 7], [2, 4, 6, 8]]
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词