FormatMatchMode

函数功能

设置输入输出tensor的format匹配模式。

函数原型

OpDef &FormatMatchMode(FormatCheckOption option);

参数说明

返回值说明

OpDef算子定义，OpDef请参考OpDef。

约束说明

不调用该接口的情况下，默认将NCHW/NHWC/DHWCN/NCDHW/NCL格式的输入输出转成ND格式进行处理。

调用示例

下面示例中，算子AddCustom输入x只支持format为NCHW，输入y只支持foramt为NHWC，就需要配置FormatMatchMode(FormatCheckOption::STRICT)，如果不配置aclnn框架会转成ND格式传给算子tiling。

 AddCustom(const char* name) : OpDef(name)
{
	this->Input("x")
		.ParamType(REQUIRED)
		.DataType({ge::DT_FLOAT})
		.FormatList({ge::FORMAT_NCHW});
	this->Input("y")
		.ParamType(REQUIRED)
		.DataType({ge::DT_FLOAT})
		.FormatList({ge::FORMAT_NHWC});
	this->Output("z")
		.ParamType(REQUIRED)
		.DataType({ge::DT_FLOAT})
		.FormatList({ge::FORMAT_ND});
	this->AICore().SetTiling(optiling::TilingFunc);
	this->AICore().AddConfig("ascendxxx");
        this->FormatMatchMode(FormatCheckOption::STRICT);
}