Share via

DML_TENSOR_FLAGS 枚举 (directml.h)

  • 项目

指定张量说明中的其他选项。 值可以按位 OR 组合在一起。

语法

typedef enum DML_TENSOR_FLAGS {
  DML_TENSOR_FLAG_NONE = 0x0,
  DML_TENSOR_FLAG_OWNED_BY_DML = 0x1
} ;

常量

 
DML_TENSOR_FLAG_NONE
值： 0x0
未指定任何选项。
DML_TENSOR_FLAG_OWNED_BY_DML
值： 0x1
指示张量数据应由 DirectML 拥有和管理。 此标志的效果是 DirectML 在初始化运算符期间创建张量数据的副本，并将其存储在永久性资源中。 这样，DirectML 便可以将张量数据的格式重新设置为其他更有效的格式。 设置此标志可以提高性能，但通常仅适用于其数据在运算符生存期内不会更改的张量 (例如权重张量) 。

此标志只能在输入张量上使用。

在特定张量说明上设置此标志时，相应的张量必须在运算符初始化期间绑定到绑定表，而不是在执行期间。 设置此标志时，尝试在执行期间绑定张量会导致错误。 这与没有 DML_TENSOR_FLAG_OWNED_BY_DML标志) 的行为 (默认行为相反，其中张量应在执行期间绑定，而不是在初始化期间绑定。

要求

   
标头 directml.h

请参阅

DirectML 中的绑定