以下是 IDMLCompiledOperator 接口的一些基本信息:
// DirectML.h
interface IDMLCompiledOperator : IUnknown
{
STDMETHOD(GetOutputTensorCount)(
_Out_ UINT32* count) PURE;
STDMETHOD(GetOutputTensor)(
_In_ UINT32 index,
_COM_Outptr_ IDMLCompiledOperatorTensor** tensor) PURE;
// 其他方法...
};
主要方法包括:
1. GetOutputTensorCount: 用于获取该编译操作的输出张量的数量。
2. GetOutputTensor: 用于获取指定索引处的输出张量的接口 (IDMLCompiledOperatorTensor)。
这个接口主要用于获取已编译操作的输出信息,以便在执行时了解操作的输出情况。具体的使用方式和示例可能需要查阅官方文档或示例代码。
在使用这些接口之前,通常需要通过 DirectML 运行时的其他部分编译模型,并获取到编译后的操作,然后通过 IDMLCompiledOperator 接口获取有关操作的信息。
转载请注明出处:http://www.pingtaimeng.com/article/detail/26869/Win32 API/Directml.h/IDMLCompiledOperator