Поделиться через

Перечисление DML_TENSOR_FLAGS (directml.h)

  • Статья

Указывает дополнительные параметры в описании тензора. Значения могут быть побитовой ИЛИ.

Синтаксис

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