TransdataOperation

功能

数据格式转换处理。将ACL_FORMAT_ND格式转换为ACL_FORMAT_FRACTAL_NZ格式,或将ACL_FORMAT_FRACTAL_NZ格式转换为ACL_FORMAT_ND格式。

  • 等效描述:ND、ACL_FORMAT_ND
  • 等效描述:NZ、ACL_FORMAT_FRACTAL_NZ、FRACTAL_NZ

定义

struct TransdataParam {
     enum TransdataType : int { 
         UNDEFINED = 0, 
         FRACTAL_NZ_TO_ND, 
         ND_TO_FRACTAL_NZ 
     };
 
     TransdataType transdataType = UNDEFINED;
     SVector<int64_t> outCrops = {0, 0};
};

参数列表

成员名称

类型

默认值

描述

transdataType

TransdataType

UNDEFINED

数据格式转换类型,支持FRACTAL_NZ和ND互相转换。

  • UNDEFINED:默认。
  • FRACTAL_NZ_TO_ND:FRACTAL_NZ转ND。
  • ND_TO_FRACTAL_NZ:ND转FRACTAL_NZ。

outCrops

SVector<int64_t>

{0,0}

仅当FRACTAL_NZ转ND时使用,表示原ND数据格式Shape的最后两维。

ND转NZ输入输出描述

参数

维度

数据类型

格式

描述

x

  1. [batch, m, n]
  2. [m, n]

float16/int8

ND

输入tensor。

y

[batch, n1, m1m0, n0]

float16/int8

NZ

输出tensor,数据类型与输入tensor一致

  • 当输入tensor数据类型为float16时,n0, m0为16。
  • 当输入tensor数据类型为int8时,n0为32,m0为16
  • 当输入tensor为2维时,输出tensor的batch为1

NZ转ND输入输出描述

参数

维度

数据类型

格式

描述

x

[batch, n1, m1m0, n0]

float16

NZ

输入tensor。

y

[batch, m, n]

float16

ND

输出tensor,数据类型与输入tensor一致

功能列表

规格约束