IDMLCommandRecorder 接口用于记录 DirectML 操作。在机器学习推断过程中,你可能会有一系列的操作(例如张量乘法、卷积等),IDMLCommandRecorder 接口用于捕获这些操作,以便稍后执行。
以下是该接口的一些基本信息:
// DirectML.h
interface IDMLCommandRecorder : IUnknown
{
STDMETHOD(RecordDispatch)(
_In_ IDMLCompiledOperator* compiledOperator,
_In_ const DML_BINDING_DESC* bindings,
_In_ UINT64 bindingsCount) PURE;
// 其他方法...
};
主要方法是 RecordDispatch,它用于记录操作。具体而言,它接收一个已编译的操作 (IDMLCompiledOperator),以及与该操作相关的绑定信息 (DML_BINDING_DESC)。绑定信息描述了操作中涉及的输入和输出张量。
请注意,此接口是 DirectML 中的一部分,可能需要在使用之前检查是否支持该接口,并了解如何正确使用它来记录和执行机器学习推断操作。详细的使用方式和示例可能需要查阅官方文档或示例代码。
转载请注明出处:http://www.zyzy.cn/article/detail/26868/Win32 API/Directml.h/IDMLCommandRecorder