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 |