DML_TENSOR_FLAGS列舉 (directml.h)

指定 Tensor 描述中的其他選項。 值可以位 OR'd 在一起。

Syntax

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 擁有和管理 Tensor 資料。 此旗標的效果是 DirectML 會在運算子初始化期間建立張量資料的複本,並將它儲存在永續性資源中。 這可讓 DirectML 將張量資料重新格式化成其他更有效率的表單。 設定此旗標可能會提升效能,但通常只適用于資料不會變更運算子存留期的量值 (,例如權數張量) 。

此旗標只能在輸入張量上使用。

當這個旗標在特定 Tensor 描述上設定時,對應的 Tensor 必須在運算子初始化期間系結至系結資料表,而不是在執行期間。 嘗試在執行期間系結 tensor,但此旗標設定會導致錯誤。 這與預設行為相反, (沒有 DML_TENSOR_FLAG_OWNED_BY_DML 旗標) 的行為,其中 tensor 預期會在執行期間系結,而不是在初始化期間系結。

需求

   
標頭 directml.h

另請參閱

在 DirectML 中繫結